60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
|
package timetools
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strconv"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"gopkg.in/mgo.v2/bson"
|
||
|
|
)
|
||
|
|
|
||
|
|
// We use RFC2822 format for timestamps everywhere ('Thu, 13 Oct 2011 18:02:00 GMT'), but
|
||
|
|
// by default Go's JSON package uses another format when decoding/encoding timestamps.
|
||
|
|
// That's why we declare our own timestamp type and implement its marshal/unmarshal methods
|
||
|
|
// so JSON package decodes/encodes it properly.
|
||
|
|
type RFC2822Time time.Time
|
||
|
|
|
||
|
|
func NewRFC2822Time(timestamp int64) RFC2822Time {
|
||
|
|
return RFC2822Time(time.Unix(timestamp, 0).UTC())
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t RFC2822Time) Unix() int64 {
|
||
|
|
return time.Time(t).Unix()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t RFC2822Time) IsZero() bool {
|
||
|
|
return time.Time(t).IsZero()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t RFC2822Time) MarshalJSON() ([]byte, error) {
|
||
|
|
return []byte(strconv.Quote(time.Time(t).Format(time.RFC1123))), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t *RFC2822Time) UnmarshalJSON(s []byte) error {
|
||
|
|
q, err := strconv.Unquote(string(s))
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if *(*time.Time)(t), err = time.Parse(time.RFC1123, q); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t RFC2822Time) GetBSON() (interface{}, error) {
|
||
|
|
return time.Time(t), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t *RFC2822Time) SetBSON(raw bson.Raw) error {
|
||
|
|
var result time.Time
|
||
|
|
err := raw.Unmarshal(&result)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
*t = RFC2822Time(result)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t RFC2822Time) String() string {
|
||
|
|
return time.Time(t).Format(time.RFC1123)
|
||
|
|
}
|