Go中如何实现JSON自定义反序列化以处理User字段多类型场景？

问题场景 #

JSON响应中的user字段存在两种格式：

• 直接为字符串：{"user": "usersName"}
• 为包含字符串类型id、布尔类型isActive的对象：{"user": {"id": "usersId", "isActive": true}}

需要实现自定义JSON反序列化逻辑，同时兼容这两种格式，并正确解析isActive字段的值。

解决方案 #

1. 调整结构体定义 #

扩展User结构体，新增IsActive字段存储对象格式中的布尔值，将原user字段改为语义更清晰的ID：

package main

import "encoding/json"

type Result struct {
    Data []Item `json:"data"`
}

type Item struct {
    User User `json:"user"`
}

type User struct {
    ID       string `json:"id"`
    IsActive bool   `json:"isActive"`
}

2. 实现自定义反序列化方法 #

修改UnmarshalJSON方法，安全处理两种格式，避免类型断言引发panic：

func (u *User) UnmarshalJSON(data []byte) error {
    // 先尝试解析为字符串格式
    var userStr string
    if err := json.Unmarshal(data, &userStr); err == nil {
        u.ID = userStr
        u.IsActive = false // 字符串格式下可设置默认值，按需调整
        return nil
    }

    // 字符串解析失败，尝试解析为对象格式
    type tempUser struct {
        ID       string `json:"id"`
        IsActive bool   `json:"isActive"`
    }
    var userObj tempUser
    if err := json.Unmarshal(data, &userObj); err != nil {
        return err
    }
    u.ID = userObj.ID
    u.IsActive = userObj.IsActive
    return nil
}

3. 保持序列化逻辑（可选） #

如果需要将User序列化为原格式（字符串或对象），可修改MarshalJSON方法：

func (u User) MarshalJSON() ([]byte, error) {
    // 按需决定序列化格式：比如IsActive为默认值时序列化为字符串
    if !u.IsActive && u.ID != "" {
        return json.Marshal(u.ID)
    }
    // 否则序列化为对象
    return json.Marshal(struct {
        ID       string `json:"id"`
        IsActive bool   `json:"isActive"`
    }{
        ID:       u.ID,
        IsActive: u.IsActive,
    })
}

关键说明 #

• 安全解析优先级：先尝试解析为字符串，失败再切换到对象解析，比直接断言interface{}更稳妥，避免类型不匹配导致程序崩溃。
• 默认值处理：字符串格式下IsActive可设置业务合理的默认值，比如false。
• 错误传递：每个json.Unmarshal调用都需检查错误，确保反序列化失败时返回有效错误信息。

内容的提问来源于stack exchange，提问作者Coldchain9