在 Golang 中从 JSON 解析字符串到枚举

Question

我有一个枚举定义为

type MyEnum int

const(
   FirstEnum MyEnum = iota
)

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

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 来自定义类型，如提到的 here。

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

Playground