1
0
mirror of https://github.com/json-iterator/go.git synced 2024-12-09 08:55:35 +02:00
json-iterator/extra/time_as_int64_codec.go

35 lines
1012 B
Go
Raw Normal View History

2017-06-20 11:43:47 +02:00
package extra
import (
"github.com/json-iterator/go"
"time"
2017-06-20 18:26:18 +02:00
"unsafe"
2017-06-20 11:43:47 +02:00
)
// keep epoch milliseconds
func RegisterTimeAsInt64Codec(precision time.Duration) {
jsoniter.RegisterTypeEncoder("time.Time", &timeAsInt64Codec{precision})
2017-06-20 11:52:41 +02:00
jsoniter.RegisterTypeDecoder("time.Time", &timeAsInt64Codec{precision})
2017-06-20 11:43:47 +02:00
}
type timeAsInt64Codec struct {
precision time.Duration
2017-06-20 11:43:47 +02:00
}
2017-06-20 18:26:18 +02:00
2017-06-20 11:52:41 +02:00
func (codec *timeAsInt64Codec) Decode(ptr unsafe.Pointer, iter *jsoniter.Iterator) {
nanoseconds := iter.ReadInt64() * codec.precision.Nanoseconds()
*((*time.Time)(ptr)) = time.Unix(0, nanoseconds)
}
2017-06-20 11:43:47 +02:00
func (codec *timeAsInt64Codec) IsEmpty(ptr unsafe.Pointer) bool {
ts := *((*time.Time)(ptr))
return ts.UnixNano() == 0
}
func (codec *timeAsInt64Codec) Encode(ptr unsafe.Pointer, stream *jsoniter.Stream) {
ts := *((*time.Time)(ptr))
stream.WriteInt64(ts.UnixNano() / codec.precision.Nanoseconds())
2017-06-20 11:43:47 +02:00
}
func (codec *timeAsInt64Codec) EncodeInterface(val interface{}, stream *jsoniter.Stream) {
jsoniter.WriteToStream(val, stream, codec)
}