Merge pull request #131 from Oskang09/feat/timezone

feat: timezone
si3nloong · Feb 15, 2022 · 12f87ca · 12f87ca
2 parents f4f506a + 6c47aec
commit 12f87ca
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/sql/codec/decoder.go b/sql/codec/decoder.go
@@ -205,6 +205,28 @@ func (dec DefaultDecoders) DecodeCivilDate(it interface{}, v reflect.Value) erro
 	return nil
 }
 
+// DecodeTimeLocation :
+func (dec DefaultDecoders) DecodeTimeLocation(it interface{}, v reflect.Value) error {
+	var x time.Location
+	switch vi := it.(type) {
+	case string:
+		tz, err := time.LoadLocation(vi)
+		if err != nil {
+			return err
+		}
+		x = *tz
+	case []byte:
+		tz, err := time.LoadLocation(string(vi))
+		if err != nil {
+			return err
+		}
+		x = *tz
+	case nil:
+	}
+	v.Set(reflect.ValueOf(x))
+	return nil
+}
+
 // DecodeCivilTime :
 func (dec DefaultDecoders) DecodeCivilTime(it interface{}, v reflect.Value) error {
 	var (

diff --git a/sql/codec/registry.go b/sql/codec/registry.go
@@ -42,6 +42,7 @@ func buildDefaultRegistry() Codecer {
 	rg.RegisterTypeCodec(reflect.TypeOf(language.Tag{}), enc.EncodeStringer, dec.DecodeLanguage)
 	rg.RegisterTypeCodec(reflect.TypeOf(currency.Unit{}), enc.EncodeStringer, dec.DecodeCurrency)
 	rg.RegisterTypeCodec(reflect.TypeOf(time.Time{}), enc.EncodeTime, dec.DecodeDateTime)
+  rg.RegisterTypeCodec(reflect.TypeOf(time.Location{}), enc.EncodeStringer, dec.DecodeTimeLocation)
 	rg.RegisterTypeCodec(reflect.TypeOf(civil.Date{}), enc.EncodeStringer, dec.DecodeCivilDate)
 	rg.RegisterTypeCodec(reflect.TypeOf(civil.Time{}), enc.EncodeStringer, dec.DecodeCivilTime)
 	rg.RegisterTypeCodec(reflect.TypeOf(sql.RawBytes{}), enc.EncodeRawBytes, dec.DecodeRawBytes)