如何为 Go 中从标量派生的类型实现 UnmarshalJSON?

Sco*_*ter 3 json go

我有一个简单的类型,它实现了 Go 中子类型整数常量到字符串的转换,反之亦然。我希望能够自动将 JSON 中的字符串解组为这种类型的值。我不能,因为 UnmarshalJSON 没有给我返回或修改标量值的方法。它需要一个结构,其成员由 UnmarshalJSON 设置。“,string”方法不适用于内置标量类型以外的其他类型。有没有办法为派生标量类型正确实现 UnmarshalJSON?

这是我所追求的一个例子。我希望它打印“Hello Ralph”四次,但它打印“Hello Bob”四次,因为 PersonID 没有被更改。

package main

import (
    "encoding/json"
    "fmt"
)

type PersonID int

const (
    Bob PersonID = iota
    Jane
    Ralph
    Nobody = -1
)

var nameMap = map[string]PersonID{
    "Bob":    Bob,
    "Jane":   Jane,
    "Ralph":  Ralph,
    "Nobody": Nobody,
}

var idMap = map[PersonID]string{
    Bob:    "Bob",
    Jane:   "Jane",
    Ralph:  "Ralph",
    Nobody: "Nobody",
}

func (intValue PersonID) Name() string {
    return idMap[intValue]
}

func Lookup(name string) PersonID {
    return nameMap[name]
}

func (intValue PersonID) UnmarshalJSON(data []byte) error {
    // The following line is not correct
    intValue = Lookup(string(data))
    return nil
}

type MyType struct {
    Person   PersonID `json: "person"`
    Count    int      `json: "count"`
    Greeting string   `json: "greeting"`
}

func main() {
    var m MyType
    if err := json.Unmarshal([]byte(`{"person": "Ralph", "count": 4, "greeting": "Hello"}`), &m); err != nil {
        fmt.Println(err)
    } else {
        for i := 0; i < m.Count; i++ {
            fmt.Println(m.Greeting, m.Person.Name())
        }
    }
}
Run Code Online (Sandbox Code Playgroud)

Cer*_*món 6

对 unmarshal 方法使用指针接收器。如果使用值接收器,则当方法返回时,对接收器的更改将会丢失。

unmarshal 方法的参数是 JSON 文本。解组 JSON 文本以获取删除所有 JSON 引用的纯字符串。

func (intValue *PersonID) UnmarshalJSON(data []byte) error {
  var s string
  if err := json.Unmarshal(data, &s); err != nil {
    return err
  }
  *intValue = Lookup(s)
  return nil
}
Run Code Online (Sandbox Code Playgroud)

JSON 标签与示例 JSON 之间不匹配。我更改了 JSON 以匹配标签,但您可以以其他方式更改它。

if err := json.Unmarshal([]byte(`{"person": "Ralph", "count": 4, "greeting": "Hello"}`), &m); err != nil {
Run Code Online (Sandbox Code Playgroud)

playground example