74 lines
1.4 KiB
Go
74 lines
1.4 KiB
Go
package types
|
||
|
||
import (
|
||
"database/sql/driver"
|
||
"encoding/json"
|
||
"time"
|
||
)
|
||
|
||
type JSONTime time.Time
|
||
|
||
const TimeFormat = "2006-01-02 15:04:05"
|
||
|
||
// timeFormatWithFrac SQLite 等可能带小数秒
|
||
const timeFormatWithFrac = "2006-01-02 15:04:05.999999999"
|
||
|
||
func (t JSONTime) MarshalJSON() ([]byte, error) {
|
||
return json.Marshal(time.Time(t).Format(TimeFormat))
|
||
}
|
||
|
||
func (t *JSONTime) UnmarshalJSON(data []byte) error {
|
||
var s string
|
||
if err := json.Unmarshal(data, &s); err != nil {
|
||
return err
|
||
}
|
||
parsed, err := time.Parse(TimeFormat, s)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
*t = JSONTime(parsed)
|
||
return nil
|
||
}
|
||
|
||
// Value 实现 driver.Valuer，供 GORM 写入数据库
|
||
func (t JSONTime) Value() (driver.Value, error) {
|
||
return time.Time(t), nil
|
||
}
|
||
|
||
// Scan 实现 sql.Scanner，供 GORM 从数据库读取
|
||
func (t *JSONTime) Scan(value interface{}) error {
|
||
if value == nil {
|
||
*t = JSONTime(time.Time{})
|
||
return nil
|
||
}
|
||
switch v := value.(type) {
|
||
case time.Time:
|
||
*t = JSONTime(v)
|
||
return nil
|
||
case []byte:
|
||
tm, err := parseTime(string(v))
|
||
if err != nil {
|
||
return err
|
||
}
|
||
*t = JSONTime(tm)
|
||
return nil
|
||
case string:
|
||
tm, err := parseTime(v)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
*t = JSONTime(tm)
|
||
return nil
|
||
default:
|
||
*t = JSONTime(time.Time{})
|
||
return nil
|
||
}
|
||
}
|
||
|
||
func parseTime(s string) (time.Time, error) {
|
||
if t, err := time.Parse(timeFormatWithFrac, s); err == nil {
|
||
return t, nil
|
||
}
|
||
return time.Parse(TimeFormat, s)
|
||
}
|