将未编组的 yaml interface{} 转换为实际结构

Question

我有以下问题：我以 yaml 形式接收信息，它可以有 2 种配置 - 单个字符：Character:  Name: Stackoverflow  Type: Awesome或者在多个字符的切片中，带有朋友切片的可选字段：Character:- Name: Stackoverflow  Type: Awesome  Friends: [Ben, John]- Name: Facebook  Type: Ok因为我不知道哪个配置会到达，所以我尝试将 yaml 解组为 interface{}，然后将其转换为其中一个结构，但没有成功：type CharacterConfig struct{Name    string      `yaml:"Name"`Type    string      `yaml:"Type"`Friends []string    `yaml:"Friends,omitempty"`var d interface{}err := yaml.Unmarshal([]byte(yamlData), &d)if err != nil {    panic(err)}res, ok := d.(CharacterConfig)if ok {    fmt.Println("CharacterConfig are ok")}res, ok := d.([]CharacterConfig)if ok {    fmt.Println("[]CharacterConfig are ok")}但是我没有收到任何打印...当我调试时，我可以看到解组到接口的工作，但没有铸件。我知道我可以直接解组到结构中，但我不明白我所做的一切没有奏效。

Answer 1

解组到interface{}不会神奇地猜测您想要自定义CharacterConfig结构中的结果。将YAML 解组为interface{}将用于map[interface{}]interface{}表示对象，并[]interface{}以递归方式表示列表/数组。

所以上面的类型断言产生ok = false了，因为存储在其中的值是interfaces{}. 您可以键入 assert 这些类型，并且断言将成功，但您不会更接近将这些值作为您的结构。

另请注意，您的输入 YAML 需要另一个包装层：

type Character struct {

    CC CharacterConfig `yaml:"Character"`

}

type CharacterList struct {

    CCs []CharacterConfig `yaml:"Character"`

}

所以一个简单的方法是首先尝试解组为 type 的值Character，如果成功，你就完成了。如果没有，请尝试再次解组，这次进入CharacterList.

这是执行此操作的示例代码：

func unmarshal(data []byte) {

    var c Character

    err := yaml.Unmarshal(data, &c)

    if err == nil {

        fmt.Printf("It's Character: %+v\n", c)

        return

    }

    var cl CharacterList

    err = yaml.Unmarshal(data, &cl)

    if err != nil {

        panic(err)

    }

    fmt.Printf("It's CharacterList: %+v\n", cl)

}

像这样测试它：

func main() {

    unmarshal([]byte(src))

    unmarshal([]byte(src2))

}

const src = `Character:

  Name: Stackoverflow

  Type: Awesome`

const src2 = `Character:

- Name: Stackoverflow

  Type: Awesome

  Friends: [Ben, John]

- Name: Facebook

  Type: Ok`

输出将是（在Go Playground上尝试）：

It's Character: {CC:{Name:Stackoverflow Type:Awesome Friends:[]}}

It's CharacterList: {CCs:[{Name:Stackoverflow Type:Awesome Friends:[Ben John]} {Name:Facebook Type:Ok Friends:[]}]}

Answer 2

与您所做的类似，但没有界面：

type CharacterConfig struct{

    Name    string      `yaml:"Name"`

    Type    string      `yaml:"Type"`

    Friends []string    `yaml:"Friends,omitempty"`

}

var a CharacterConfig

var b []CharacterConfig

err := yaml.Unmarshal([]byte(yamlData), &a)

if err == nil {

    // a is valid format

}

err = yaml.Unmarshal([]byte(yamlData), &b)

if err == nil {

    // b is valid format

}