问题内容
我从队列中收到一个 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://www.php.cn/link/a73d9b34d6f7c322fa3e34c633b1297d
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
} 
