保留 json 的全部信息，按需要解组为结构体

最近发现不少小伙伴都对Golang很感兴趣，所以今天继续给大家介绍Golang相关的知识，本文《保留 json 的全部信息，按需要解组为结构体》主要内容涉及到等等知识点，希望能帮到你！当然如果阅读本文时存在不同想法，可以在评论中表达，但是请勿使用过激的措辞~

我从队列中收到一个 JSON 对象。我想处理该对象的部分内容并在发送到另一个队列之前更新它。

如果我有一个输入

{
  "one": "data1",
  "two": "data2",
  "three": "data3",
  ...
}

我想采用 one 和 two 并在添加新输出并发送类似消息之前计算一些信息

{
  "one": "data1UPDATED",
  "two": "data2",
  "three": "data3",
  ...
  "extra_info": "extra"
}

为了能够更好地处理数据，我想将 JSON 解组为类似的结构

type Message struct {
 One string `json:one`
 Two string `json:two`
}

但不想映射所有字段，因为其中许多字段与此应用程序无关，而其他字段可能会更改。

我尝试过的事情

我尝试将所有字段映射到 json.RawMessage

type Message struct {
 One string `json:"one"`
 Two string `json:"two"`
 ExtraFields json.RawMessage `json:"-"`
}

但是，在编组结构时，不包括 ExtraFields 中包含的字段。

{
  "one": "data1",
  "two": "data2"
}

我还尝试编写一种自定义解组，将完整消息存储在一个结构中，其中有用消息存储在一个元素中，其他信息存储在另一个字段中

type TmpMessage struct{
  Msg Message
  Others json.RawMessage
}

但这变得非常混乱，我希望有一个更简洁的解决方案。

有没有办法做到这一点，或者是不解组到结构并仅使用原始 map[string] 接口{}的唯一选择？

正确答案

您可以编写一个自定义解组器，例如：

https://go.dev/play/p/53r1hqmbuJi

func (m *Message) UnmarshalJSON(data []byte) error {
  tmp := make(map[string]interface{})
  if err := json.Unmarshal(data, &tmp); err != nil {
    return err
  }
  t := reflect.TypeOf(*m)
  for i := 0; i < t.NumField(); i++ {
    f := t.Field(i)
    if f.Name != "ExtraFields" {
      jsonTag := f.Tag.Get("json")
      // this part should probably by a type switch...
      reflect.ValueOf(m).Elem().Field(i).SetString(tmp[jsonTag].(string))
      delete(tmp, jsonTag)
    }
  }
  var err error
  m.ExtraFields, err = json.Marshal(tmp)
  return err
}

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《保留 json 的全部信息，按需要解组为结构体》文章吧，也可关注golang学习网公众号了解相关技术文章。