package main
import (
"encoding/json"
"fmt"
"gorm.io/gorm"
"time"
)
// 定义自定义时间类型
type CustomTime struct {
time.Time
}
// 实现json.Marshaler接口以支持JSON序列化
func (ct CustomTime) MarshalJSON() ([]byte, error) {
return json.Marshal(ct.Format("2006-01-02 15:04:05"))
}
// 实现json.Unmarshaler接口以支持JSON反序列化
func (ct *CustomTime) UnmarshalJSON(data []byte) error {
t, err := time.Parse("2006-01-02 15:04:05", string(data))
if err != nil {
return err
}
ct.Time = t
return nil
}
// 实现GormValuer接口以支持Gorm存储
func (ct CustomTime) GormValue(ctx context.Context, db *gorm.DB) (interface{}, error) {
return ct.Time, nil
}
// 实现GormScanner接口以支持Gorm读取数据库
func (ct *CustomTime) Scan(value interface{}) error {
switch v := value.(type) {
case time.Time:
ct.Time = v
return nil
case nil:
ct.Time = time.Time{}
return nil
default:
return fmt.Errorf("unsupported scan type for CustomTime: %T", value)
}
}
func main() {
// 示例：序列化和反序列化
now := CustomTime{time.Now()}
serialized, _ := json.Marshal(now)
fmt.Println(string(serialized)) // 输出类似 "2023-04-05 12:34:56"
var deserialized CustomTime
json.Unmarshal(serialized, &deserialized)
fmt.Println(deserialized.Format("2006-01-02 15:04:05")) // 输出 "2023-04-05 12:34:56"
// 示例：Gorm存储和读取
// 假设有一个Gorm模型使用CustomTime
type Model struct {
ID uint
CreatedAt CustomTime
}
// 使用Gorm时，CustomTime会自动被识别和处理
}这段代码定义了一个CustomTime结构体，它包含了time.Time作为嵌入字段。它实现了json.Marshaler和json.Unmarshaler接口以支持JSON序列化和反序列化，同时实现了gorm.io/gorm包中的GormValuer和GormScanner接口以支持Gorm ORM的存储和读取。这样，CustomTime可以在JSON和Gorm之间无缝集成，方便进行时间数据的处理。