mysqldialect: add time location support

uptrace · Sep 10, 2024 · 3d5a68a · 3d5a68a
1 parent 6c83b05
commit 3d5a68a
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/dialect/mysqldialect/dialect.go b/dialect/mysqldialect/dialect.go
@@ -32,6 +32,7 @@ type Dialect struct {
 
 	tables   *schema.Tables
 	features feature.Feature
+	loc      *time.Location
 }
 
 func New() *Dialect {
@@ -50,6 +51,16 @@ func New() *Dialect {
 	return d
 }
 
+func NewWithLocation(loc string) *Dialect {
+	d := New()
+	location, err := time.LoadLocation(loc)
+	if err != nil {
+		panic(fmt.Errorf("mysqldialect can't load provided location %s: %s", loc, err))
+	}
+	d.loc = location
+	return d
+}
+
 func (d *Dialect) Init(db *sql.DB) {
 	var version string
 	if err := db.QueryRow("SELECT version()").Scan(&version); err != nil {
@@ -103,9 +114,13 @@ func (d *Dialect) IdentQuote() byte {
 	return '`'
 }
 
-func (*Dialect) AppendTime(b []byte, tm time.Time) []byte {
+func (d *Dialect) AppendTime(b []byte, tm time.Time) []byte {
 	b = append(b, '\'')
-	b = tm.AppendFormat(b, "2006-01-02 15:04:05.999999")
+	if d.loc != nil {
+		b = tm.In(d.loc).AppendFormat(b, "2006-01-02 15:04:05.999999")
+	} else {
+		b = tm.AppendFormat(b, "2006-01-02 15:04:05.999999")
+	}
 	b = append(b, '\'')
 	return b
 }