feat: Support conversion of Unix timestamps in timestamptz (#570)

hermanschaaf · web-flow · commit 6b948ab392c5 · 2023-01-04T12:54:27.000Z
This adds support for Unix timestamps to the Timestamptz type. I think it's reasonable to expect that an integer type is to be interpreted as unix time, and a custom resolver should be used if this is not the case.
diff --git a/schema/timestamptz.go b/schema/timestamptz.go
@@ -76,6 +76,12 @@ func (dst *Timestamptz) Set(src any) error {
 	}
 
 	switch value := src.(type) {
+	case int:
+		*dst = Timestamptz{Time: time.Unix(int64(value), 0), Status: Present}
+	case int64:
+		*dst = Timestamptz{Time: time.Unix(value, 0), Status: Present}
+	case uint64:
+		*dst = Timestamptz{Time: time.Unix(int64(value), 0), Status: Present}
 	case time.Time:
 		*dst = Timestamptz{Time: value, Status: Present}
 	case *time.Time:
diff --git a/schema/timestamptz_test.go b/schema/timestamptz_test.go
@@ -25,6 +25,9 @@ func TestTimestamptzSet(t *testing.T) {
 		{source: time.Date(2000, 1, 1, 0, 0, 0, 0, time.Local), result: Timestamptz{Time: time.Date(2000, 1, 1, 0, 0, 0, 0, time.Local), Status: Present}},
 		{source: time.Date(2000, 1, 1, 0, 0, 1, 0, time.Local), result: Timestamptz{Time: time.Date(2000, 1, 1, 0, 0, 1, 0, time.Local), Status: Present}},
 		{source: time.Date(2200, 1, 1, 0, 0, 0, 0, time.Local), result: Timestamptz{Time: time.Date(2200, 1, 1, 0, 0, 0, 0, time.Local), Status: Present}},
+		{source: int(time.Date(2000, 1, 1, 0, 0, 0, 0, time.Local).Unix()), result: Timestamptz{Time: time.Date(2000, 1, 1, 0, 0, 0, 0, time.Local), Status: Present}},
+		{source: uint64(time.Date(2000, 1, 1, 0, 0, 0, 0, time.Local).Unix()), result: Timestamptz{Time: time.Date(2000, 1, 1, 0, 0, 0, 0, time.Local), Status: Present}},
+		{source: time.Date(2000, 1, 1, 0, 0, 0, 0, time.Local).Unix(), result: Timestamptz{Time: time.Date(2000, 1, 1, 0, 0, 0, 0, time.Local), Status: Present}},
 		{source: _time(time.Date(1970, 1, 1, 0, 0, 0, 0, time.Local)), result: Timestamptz{Time: time.Date(1970, 1, 1, 0, 0, 0, 0, time.Local), Status: Present}},
 		{source: Infinity, result: Timestamptz{InfinityModifier: Infinity, Status: Present}},
 		{source: NegativeInfinity, result: Timestamptz{InfinityModifier: NegativeInfinity, Status: Present}},
