|
|
- // Copyright 2015 go-swagger maintainers
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
-
- package strfmt
-
- import (
- "database/sql/driver"
- "encoding/binary"
- "encoding/json"
- "fmt"
- "regexp"
- "strings"
- "time"
-
- "go.mongodb.org/mongo-driver/bson"
-
- "go.mongodb.org/mongo-driver/bson/bsontype"
- )
-
- func init() {
- dt := DateTime{}
- Default.Add("datetime", &dt, IsDateTime)
- }
-
- // IsDateTime returns true when the string is a valid date-time
- func IsDateTime(str string) bool {
- if len(str) < 4 {
- return false
- }
- s := strings.Split(strings.ToLower(str), "t")
- if len(s) < 2 || !IsDate(s[0]) {
- return false
- }
-
- matches := rxDateTime.FindAllStringSubmatch(s[1], -1)
- if len(matches) == 0 || len(matches[0]) == 0 {
- return false
- }
- m := matches[0]
- res := m[1] <= "23" && m[2] <= "59" && m[3] <= "59"
- return res
- }
-
- const (
- // RFC3339Millis represents a ISO8601 format to millis instead of to nanos
- RFC3339Millis = "2006-01-02T15:04:05.000Z07:00"
- // RFC3339Micro represents a ISO8601 format to micro instead of to nano
- RFC3339Micro = "2006-01-02T15:04:05.000000Z07:00"
- // ISO8601LocalTime represents a ISO8601 format to ISO8601 in local time (no timezone)
- ISO8601LocalTime = "2006-01-02T15:04:05"
- // ISO8601TimeWithReducedPrecision represents a ISO8601 format with reduced precision (dropped secs)
- ISO8601TimeWithReducedPrecision = "2006-01-02T15:04Z"
- // ISO8601TimeWithReducedPrecision represents a ISO8601 format with reduced precision and no timezone (dropped seconds + no timezone)
- ISO8601TimeWithReducedPrecisionLocaltime = "2006-01-02T15:04"
- // DateTimePattern pattern to match for the date-time format from http://tools.ietf.org/html/rfc3339#section-5.6
- DateTimePattern = `^([0-9]{2}):([0-9]{2}):([0-9]{2})(.[0-9]+)?(z|([+-][0-9]{2}:[0-9]{2}))$`
- )
-
- var (
- dateTimeFormats = []string{RFC3339Micro, RFC3339Millis, time.RFC3339, time.RFC3339Nano, ISO8601LocalTime, ISO8601TimeWithReducedPrecision, ISO8601TimeWithReducedPrecisionLocaltime}
- rxDateTime = regexp.MustCompile(DateTimePattern)
- // MarshalFormat sets the time resolution format used for marshaling time (set to milliseconds)
- MarshalFormat = RFC3339Millis
- )
-
- // ParseDateTime parses a string that represents an ISO8601 time or a unix epoch
- func ParseDateTime(data string) (DateTime, error) {
- if data == "" {
- return NewDateTime(), nil
- }
- var lastError error
- for _, layout := range dateTimeFormats {
- dd, err := time.Parse(layout, data)
- if err != nil {
- lastError = err
- continue
- }
- return DateTime(dd), nil
- }
- return DateTime{}, lastError
- }
-
- // DateTime is a time but it serializes to ISO8601 format with millis
- // It knows how to read 3 different variations of a RFC3339 date time.
- // Most APIs we encounter want either millisecond or second precision times.
- // This just tries to make it worry-free.
- //
- // swagger:strfmt date-time
- type DateTime time.Time
-
- // NewDateTime is a representation of zero value for DateTime type
- func NewDateTime() DateTime {
- return DateTime(time.Unix(0, 0).UTC())
- }
-
- // String converts this time to a string
- func (t DateTime) String() string {
- return time.Time(t).Format(MarshalFormat)
- }
-
- // MarshalText implements the text marshaller interface
- func (t DateTime) MarshalText() ([]byte, error) {
- return []byte(t.String()), nil
- }
-
- // UnmarshalText implements the text unmarshaller interface
- func (t *DateTime) UnmarshalText(text []byte) error {
- tt, err := ParseDateTime(string(text))
- if err != nil {
- return err
- }
- *t = tt
- return nil
- }
-
- // Scan scans a DateTime value from database driver type.
- func (t *DateTime) Scan(raw interface{}) error {
- // TODO: case int64: and case float64: ?
- switch v := raw.(type) {
- case []byte:
- return t.UnmarshalText(v)
- case string:
- return t.UnmarshalText([]byte(v))
- case time.Time:
- *t = DateTime(v)
- case nil:
- *t = DateTime{}
- default:
- return fmt.Errorf("cannot sql.Scan() strfmt.DateTime from: %#v", v)
- }
-
- return nil
- }
-
- // Value converts DateTime to a primitive value ready to written to a database.
- func (t DateTime) Value() (driver.Value, error) {
- return driver.Value(t.String()), nil
- }
-
- // MarshalJSON returns the DateTime as JSON
- func (t DateTime) MarshalJSON() ([]byte, error) {
- return json.Marshal(time.Time(t).Format(MarshalFormat))
- }
-
- // UnmarshalJSON sets the DateTime from JSON
- func (t *DateTime) UnmarshalJSON(data []byte) error {
- if string(data) == jsonNull {
- return nil
- }
-
- var tstr string
- if err := json.Unmarshal(data, &tstr); err != nil {
- return err
- }
- tt, err := ParseDateTime(tstr)
- if err != nil {
- return err
- }
- *t = tt
- return nil
- }
-
- // MarshalBSON renders the DateTime as a BSON document
- func (t DateTime) MarshalBSON() ([]byte, error) {
- return bson.Marshal(bson.M{"data": t})
- }
-
- // UnmarshalBSON reads the DateTime from a BSON document
- func (t *DateTime) UnmarshalBSON(data []byte) error {
- var obj struct {
- Data DateTime
- }
-
- if err := bson.Unmarshal(data, &obj); err != nil {
- return err
- }
-
- *t = obj.Data
-
- return nil
- }
-
- // MarshalBSONValue is an interface implemented by types that can marshal themselves
- // into a BSON document represented as bytes. The bytes returned must be a valid
- // BSON document if the error is nil.
- // Marshals a DateTime as a bsontype.DateTime, an int64 representing
- // milliseconds since epoch.
- func (t DateTime) MarshalBSONValue() (bsontype.Type, []byte, error) {
- // UnixNano cannot be used, the result of calling UnixNano on the zero
- // Time is undefined.
- i64 := time.Time(t).Unix() * 1000
- buf := make([]byte, 8)
- binary.LittleEndian.PutUint64(buf, uint64(i64))
-
- return bsontype.DateTime, buf, nil
- }
-
- // UnmarshalBSONValue is an interface implemented by types that can unmarshal a
- // BSON value representation of themselves. The BSON bytes and type can be
- // assumed to be valid. UnmarshalBSONValue must copy the BSON value bytes if it
- // wishes to retain the data after returning.
- func (t *DateTime) UnmarshalBSONValue(tpe bsontype.Type, data []byte) error {
- i64 := int64(binary.LittleEndian.Uint64(data))
- // TODO: Use bsonprim.DateTime.Time() method
- *t = DateTime(time.Unix(i64/1000, i64%1000*1000000))
-
- return nil
- }
-
- // DeepCopyInto copies the receiver and writes its value into out.
- func (t *DateTime) DeepCopyInto(out *DateTime) {
- *out = *t
- }
-
- // DeepCopy copies the receiver into a new DateTime.
- func (t *DateTime) DeepCopy() *DateTime {
- if t == nil {
- return nil
- }
- out := new(DateTime)
- t.DeepCopyInto(out)
- return out
- }
-
- // GobEncode implements the gob.GobEncoder interface.
- func (t DateTime) GobEncode() ([]byte, error) {
- return t.MarshalBinary()
- }
-
- // GobDecode implements the gob.GobDecoder interface.
- func (t *DateTime) GobDecode(data []byte) error {
- return t.UnmarshalBinary(data)
- }
-
- // MarshalBinary implements the encoding.BinaryMarshaler interface.
- func (t DateTime) MarshalBinary() ([]byte, error) {
- return time.Time(t).MarshalBinary()
- }
-
- // UnmarshalBinary implements the encoding.BinaryUnmarshaler interface.
- func (t *DateTime) UnmarshalBinary(data []byte) error {
- var original time.Time
-
- err := original.UnmarshalBinary(data)
- if err != nil {
- return err
- }
-
- *t = DateTime(original)
-
- return nil
- }
|