如何在 Go 中将 YAML 字段动态解析为有限结构集？-Golang-PHP中文网

如何在 Go 中将 YAML 字段动态解析为有限结构集？

Barbara Streisand

发布： 2024-10-30 23:11:30

原创

273 人浏览过

How to Dynamically Parse a YAML Field into a Finite Set of Structs in Go?

在 Go 中将 YAML 字段动态解析为有限结构体

简介

在 Go 中将 YAML 解析为结构体非常简单。但是，当 YAML 字段可以表示多个可能的结构时，任务就会变得更加复杂。本文探讨了使用 Go 的 YAML 包的动态方法。

使用 YAML v2 进行动态解组

对于 Yaml v2，可以使用以下方法：

<code class="go">type yamlNode struct {
    unmarshal func(interface{}) error
}

func (n *yamlNode) UnmarshalYAML(unmarshal func(interface{}) error) error {
    n.unmarshal = unmarshal
    return nil
}

type Spec struct {
    Kind string      `yaml:"kind"`
    Spec interface{} `yaml:"-"
}</code>

登录后复制

<code class="go">func (s *Spec) UnmarshalYAML(unmarshal func(interface{}) error) error {
    type S Spec
    type T struct {
        S `yaml:",inline"`
        Spec yamlNode `yaml:"spec"`
    }

    obj := &T{}
    if err := unmarshal(obj); err != nil {
        return err
    }
    *s = Spec(obj.S)

    switch s.Kind {
    case "foo":
        s.Spec = new(Foo)
    case "bar":
        s.Spec = new(Bar)
    default:
        panic("kind unknown")
    }
    return obj.Spec.unmarshal(s.Spec)
}</code>

登录后复制

使用 YAML v3 进行动态解组

对于 Yaml v3，方法略有不同：

<code class="go">type Spec struct {
    Kind string      `yaml:"kind"`
    Spec interface{} `yaml:"-"
}</code>

登录后复制

<code class="go">func (s *Spec) UnmarshalYAML(n *yaml.Node) error {
    type S Spec
    type T struct {
        *S `yaml:",inline"`
        Spec yaml.Node `yaml:"spec"`
    }

    obj := &T{S: (*S)(s)}
    if err := n.Decode(obj); err != nil {
        return err
    }

    switch s.Kind {
    case "foo":
        s.Spec = new(Foo)
    case "bar":
        s.Spec = new(Bar)
    default:
        panic("kind unknown")
    }
    return obj.Spec.Decode(s.Spec)
}</code>

登录后复制