我从队列接收一个JSON对象。我想处理该对象的一部分并在发送到另一个队列之前对其进行更新。

如果我有输入

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

我想在添加新输出并发送诸如

{
  "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
}

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

有没有办法做到这一点，或者是唯一的选择，而不是与struct一起使用，而只是与RAW map[string]interface{}?一起工作

分析解答

您可以编写一个自定义的Unmarshaler，类似：

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
}