Timothy Messier
0f765e7149
When converting a timestamp.Timestamp for use in SQL, this now correctly converts the NegativeInfinity and PositiveInfinity constants into the corresponding `-infinity` and `infinity` SQL constants.
56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package timestamp
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"errors"
|
|
"math"
|
|
"time"
|
|
|
|
"google.golang.org/protobuf/types/known/timestamppb"
|
|
)
|
|
|
|
var (
|
|
NegativeInfinityTS = time.Date(math.MinInt32, time.January, 1, 0, 0, 0, 0, time.UTC)
|
|
PositiveInfinityTS = time.Date(math.MaxInt32, time.December, 31, 23, 59, 59, 1e9-1, time.UTC)
|
|
)
|
|
|
|
// Scan implements sql.Scanner for protobuf Timestamp.
|
|
func (ts *Timestamp) Scan(value any) error {
|
|
switch t := value.(type) {
|
|
case time.Time:
|
|
ts.Timestamp = timestamppb.New(t) // google proto version
|
|
case string:
|
|
switch value {
|
|
case "-infinity":
|
|
ts.Timestamp = timestamppb.New(NegativeInfinityTS)
|
|
case "infinity":
|
|
ts.Timestamp = timestamppb.New(PositiveInfinityTS)
|
|
}
|
|
default:
|
|
return errors.New("Not a protobuf Timestamp")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Scan implements driver.Valuer for protobuf Timestamp.
|
|
func (ts *Timestamp) Value() (driver.Value, error) {
|
|
switch {
|
|
case ts == nil:
|
|
return nil, nil
|
|
case ts.AsTime().Equal(NegativeInfinityTS):
|
|
return "-infinity", nil
|
|
case ts.AsTime().Equal(PositiveInfinityTS):
|
|
return "infinity", nil
|
|
default:
|
|
return ts.Timestamp.AsTime(), nil
|
|
}
|
|
}
|
|
|
|
// GormDataType gorm common data type (required)
|
|
func (ts *Timestamp) GormDataType() string {
|
|
return "timestamp"
|
|
}
|