Go json unmarshal interface{} field bind to struct

wind5o · · 1051 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

写代码时碰到这么一个需求,某个字段根据不同条件对应不同子结构体,通过interface返给前端,同时前端上传时也要通过这个字段将数据传给后端。

struct -> json这个比较好办,给interface赋值不同的子结构体即可。json -> struct时有点难搞,需要做下特殊处理。默认情况json字符串解析到interface field会是map[string]interface{}。

先上代码:

type Foo struct {
    Type   string      `json:"type"`
    Object interface{} `json:"object"`
}

type A struct {
    A string `json:"a"`
}

type B struct {
    B string `json:"b"`
}

func (f *Foo) UnmarshalJSON(data []byte) error {
    type cloneType Foo

    rawMsg := json.RawMessage{}
    f.Object = &rawMsg

    if err := json.Unmarshal(data, (*cloneType)(f)); err != nil {
        return err
    }

    switch f.Type {
    case "a":
        params := new(A)
        if err := json.Unmarshal(rawMsg, params); err != nil {
            return err
        }
        f.Object = params
    case "b":
        params := new(B)
        if err := json.Unmarshal(rawMsg, params); err != nil {
            return err
        }
        f.Object = params
    default:
        return errors.New("nonsupport type")
    }

    return nil
}

test:

func TestUnmarshal(t *testing.T) {
    foo := &Foo{
        Type:   "a",
        Object: A{A: "rua"},
    }

    bts, err := json.Marshal(foo)
    require.Nil(t, err)
    t.Logf("jsonStr = %s", bts)

    newFoo := new(Foo)
    err = json.Unmarshal(bts, newFoo)
    require.Nil(t, err)
    t.Logf("struct = %+v, object = %+v", newFoo, newFoo.Object)
}

上述代码通过UnmarshalJSON来自定义json unmarshal,先将object赋值为json.RawMessage,再判断type将RawMessage解析到对应结构体,实现delay unmarshal。

这里有一个问题就是在UnmarshalJSON中对相同结构体使用json.Unmarshal会无限递归调用导致StackOverflow,所以新建了一个cloneType。

其实还有一种方法,就是先解析到map,再json.Marshal将map编码为jsonStr,再判断type解析到对应结构体。这种方法多了两个步骤,性能应该不如上面的方法。

参考资料:
https://www.jianshu.com/p/c06...

https://stackoverflow.com/que...


有疑问加站长微信联系(非本文作者)

本文来自:Segmentfault

感谢作者:wind5o

查看原文:Go json unmarshal interface{} field bind to struct

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

1051 次点击  
加入收藏 微博
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传