json - 如何在 Go 中自动向 JSON 添加类型字段？

Question

如何在 Go 中为每个序列化的 JSON 自动添加一个 type 字段？

例如，在这段代码中:

type Literal struct {
  Value interface{} `json:'value'`
  Raw   string      `json:'raw'`
}

type BinaryExpression struct {
  Operator string  `json:"operator"`
  Right    Literal `json:"right"`
  Left     Literal `json:"left"`
}

os.Stdout.Write(json.Marshal(&BinaryExpression{ ... }))

而不是生成类似的东西:

{
   "operator": "*",
   "left": {
      "value": 6,
      "raw": "6"
   },
   "right": {
      "value": 7,
      "raw": "7"
   }
}

我想生成这个:

{
   "type": "BinaryExpression",
   "operator": "*",
   "left": {
      "type": "Literal",
      "value": 6,
      "raw": "6"
   },
   "right": {
      "type": "Literal",
      "value": 7,
      "raw": "7"
   }
}

Answer 1

您可以为您的结构重写 MarshalJSON 函数，以将类型注入(inject)到辅助结构中，然后返回该结构。

package main

import (
    "encoding/json"
    "os"
)

type Literal struct {
    Value interface{} `json:'value'`
    Raw   string      `json:'raw'`
}

func (l *Literal) MarshalJSON() ([]byte, error) {
    type Alias Literal
    return json.Marshal(&struct {
        Type string `json:"type"`
        *Alias
    }{
        Type:  "Literal",
        Alias: (*Alias)(l),
    })
}

type BinaryExpression struct {
    Operator string  `json:"operator"`
    Right    Literal `json:"right"`
    Left     Literal `json:"left"`
}

func (b *BinaryExpression) MarshalJSON() ([]byte, error) {
    type Alias BinaryExpression
    return json.Marshal(&struct {
        Type string `json:"type"`
        *Alias
    }{
        Type:  "BinaryExpression",
        Alias: (*Alias)(b),
    })
}

func main() {
    _ = json.NewEncoder(os.Stdout).Encode(
        &BinaryExpression{
            Operator: "*",
            Right: Literal{
                Value: 6,
                Raw:   "6",
            },
            Left: Literal{
                Value: 7,
                Raw:   "7",
            },
        })
}