snapshot of https://github.com/ent/ent.git (master, 2026-09-15; blobs > 131072B omitted)

dialect/gremlin/encoding/graphson/time.go

@@ -0,0 +1,34 @@
+// Copyright 2019-present Facebook Inc. All rights reserved.
+// This source code is licensed under the Apache 2.0 license found
+// in the LICENSE file in the root directory of this source tree.
+
+package graphson
+
+import (
+	"time"
+	"unsafe"
+
+	jsoniter "github.com/json-iterator/go"
+)
+
+func init() {
+	RegisterTypeEncoder("time.Time", typeEncoder{timeCodec{}, Timestamp})
+	RegisterTypeDecoder("time.Time", typeDecoder{timeCodec{}, Types{Timestamp, Date}})
+}
+
+type timeCodec struct{}
+
+func (timeCodec) IsEmpty(ptr unsafe.Pointer) bool {
+	ts := *((*time.Time)(ptr))
+	return ts.IsZero()
+}
+
+func (timeCodec) Encode(ptr unsafe.Pointer, stream *jsoniter.Stream) {
+	ts := *((*time.Time)(ptr))
+	stream.WriteInt64(ts.UnixNano() / time.Millisecond.Nanoseconds())
+}
+
+func (timeCodec) Decode(ptr unsafe.Pointer, iter *jsoniter.Iterator) {
+	ns := iter.ReadInt64() * time.Millisecond.Nanoseconds()
+	*((*time.Time)(ptr)) = time.Unix(0, ns)
+}