在 Golang 中将 JSON 中的字符串解析为枚举

Question

我有一个枚举定义为

type MyEnum int

const(
   FirstEnum MyEnum = iota
)

然后，我有一个包含键值对的 json "Key": "FirstEnum"。我就是这样解组的。

var data map[string]interface{}
err := json.Unmarshal([]byte(json), &data)
x := data["key"].(MyEnum)

但是，当我运行此命令时，我收到错误：

panic: interface conversion: interface {} is string, not ps.Protocol [recovered]
        panic: interface conversion: interface {} is string, not ps.Protocol

有没有办法让它像 Go 中枚举的字符串表示形式到枚举类型的正常转换一样工作？

Answer 1

我想出了一些类似的方法（至少适用于我目前的情况）：

用于string类似enum常量：

type MyEnum string

const(
   FirstEnum MyEnum = "FirstEnum"
)

现在，使用解码 json 来自定义类型，如此处所述。

data := MyJsonStruct{}
err := json.Unmarshal([]byte(json), &data)

MyJsonStruct看起来像：

type MyJsonStruct struct {
    Key MyEnum
}

然后，您可以使 MyJsonStruct 实现该Unmarshaler接口，以便您可以验证给定的值。

func (s *MyJsonStruct) UnmarshalJSON(data []byte) error {
    // Define a secondary type so that we don't end up with a recursive call to json.Unmarshal
    type Aux MyJsonStruct;
    var a *Aux = (*Aux)(s);
    err := json.Unmarshal(data, &a)
    if err != nil {
        return err
    }

    // Validate the valid enum values
    switch s.Key {
    case FirstEnum, SecondEnum:
        return nil
    default:
        s.Key = ""
        return errors.New("invalid value for Key")
    }
}

操场