自定义序列化时间格式
问题场景
有一个User结构体,其中有一个time.Time类型的字段
type User struct {
Name string `json:"name"`
Age int `json:"age"`
Birthday time.Time `json:"birthday"`
}
定义一个jsonutil
func ToStr(v any) string {
b, err := json.Marshal(v)
if err != nil {
return ""
}
return string(b)
}
func ToObj[T any](s string, v *T) *T {
err := json.Unmarshal([]byte(s), v)
if err != nil {
return nil
}
return v
}
如果直接使用默认的json序列化,如下
func Test1(t *testing.T) {
var user = User{
Name: "yimin",
Age: 18,
Birthday: time.Now(),
}
str := ToStr(user)
fmt.Println(str)
}
得到的结果是这样的,时间格式不是很友好
{"name":"yimin","age":18,"birthday":"2024-08-15T15:50:51.3876651+08:00"}
实现接口
让User实现Marshaler接口,自定义序列化逻辑
func (u *User) MarshalJSON() ([]byte, error) {
type UserTemp User // 避免循环引用
type ToMarshal struct {
*UserTemp // 这里必须使用匿名嵌套,序列化才会展开平铺,从而使用string类型Birthday
Birthday string `json:"birthday"`
}
return json.Marshal(ToMarshal{
UserTemp: (*UserTemp)(u),
Birthday: u.Birthday.Format(time.DateTime),
})
}
思路:
- 不要序列化原结构体,而是序列化一个新的结构体
- 新结构体中将原来time.Time类型的字段重新定义为string类型,这样就可以自定义时间格式了
- 同时新结构体还需要保留原结构体所有其他字段
实现:
- 利用json序列化嵌套匿名结构体的特性(展开平铺)
- 这里匿名结构体需要重新定义类型,避免循环引用
- 这样一来,序列化User转化为序列化ToMarshal,且将Birthday改为了string类型(可以自定义格式)
测试
注意此时需要传递user的指针类型
func Test1(t *testing.T) {
var user = User{
Name: "yimin",
Age: 18,
Birthday: time.Now(),
}
str := ToStr(&user)
fmt.Println(str)
}
{"name":"yimin","age":18,"birthday":"2024-08-15 15:59:26"}
那反序列化呢?
func Test2(t *testing.T) {
var str = `{"name":"yimin","age":18,"birthday":"2024-08-15 15:59:26"}`
var user = ToObj[*User](str, &User{})
fmt.Println(user)
}
直接失败了,因为默认不支持这个时间格式
<nil>
解决方法也是一样的,自己实现反序列化逻辑
处理反序列化时间格式
实现接口
func (u *User) UnmarshalJSON(b []byte) error {
type UserTemp User
type ToUnMarshal struct {
*UserTemp
Birthday string `json:"birthday"`
}
var t = ToUnMarshal{
UserTemp: (*UserTemp)(u), // 确保数据反序列化到u,但不会处理Birthday
}
if err := json.Unmarshal(b, &t); err != nil {
return err
}
var err error
// 单独处理birthday
u.Birthday, err = time.Parse(time.DateTime, t.Birthday)
return err
}
思路:
- 大体思路和序列化是一样的,换个Birthday类型是string的结构体反序列化
- 反序列化前需要手动初始化UserTemp为this对象,确保该有的数据都能对应上(除了Birthday)
- 单独处理this对象的Birthday,手动指定格式Parse
测试
func Test2(t *testing.T) {
var str = `{"name":"yimin","age":18,"birthday":"2024-08-15 15:59:26"}`
var user = ToObj[User](str, &User{})
fmt.Printf("%+v\n", user)
}
&{Name:yimin Age:18 Birthday:2024-08-15 15:59:26 +0000 UTC}
进一步简化
- 上面的方法虽然解决了问题,但是对于每一个结构体都要实现方法,比较繁琐
- 我们关注的只是time.Time类型的序列化和反序列化,于是可以自定义时间类型,单独对这个类型实现方法
- 后面需要使用time.Time类型的地方,替换为我们自定义的时间类型
自定义类型
type Time time.Time
const format = "2006-01-02 15:04:05"
// 固定写法,前后加"
func (t Time) MarshalJSON() ([]byte, error) {
b := make([]byte, 0, len(format)+2)
b = append(b, '"')
b = time.Time(t).AppendFormat(b, format)
b = append(b, '"')
return b, nil
}
// 固定写法,格式化字符串前后加上"
// 因为需要修改t,使用指针类型
func (t *Time) UnmarshalJSON(data []byte) error {
formatTime, err := time.ParseInLocation(`"`+format+`"`, string(data), time.Local)
if err != nil {
return err
}
*t = Time(formatTime)
return nil
}
// 格式化输出
func (t Time) String() string {
return time.Time(t).Format(format)
}
time.Time替换为自定义类型
type User struct {
Name string `json:"name"`
Age int `json:"age"`
Birthday Time `json:"birthday"`
}
测试
func Test1(t *testing.T) {
var user = User{
Name: "yimin",
Age: 18,
Birthday: Time(time.Now()),
}
str := ToStr(&user)
fmt.Println(str)
}
func Test2(t *testing.T) {
var str = `{"name":"yimin","age":18,"birthday":"2024-08-15 15:59:26"}`
var user = ToObj[User](str, &User{})
fmt.Printf("%+v\n", user)
}
{"name":"yimin","age":18,"birthday":"2024-08-15 17:23:37"}
&{Name:yimin Age:18 Birthday:2024-08-15 15:59:26}
参考
你需要知道的那些go语言json技巧 - 李文周的博客
go 中,json 转换时间(time.Time)的格式(默认格式为 RFC3339)_golang rfc3339-CSDN博客