Go で YAML フィールドを有限の構造体セットに動的に解析する

はじめに

Go で YAML を構造体に解析するのは簡単です。ただし、YAML フィールドが複数の可能な構造体を表すことができる場合、タスクはより複雑になります。この記事では、Go の YAML パッケージを使用した動的アプローチについて説明します。

YAML v2 による動的アンマーシャリング

Yaml v2 の場合は、次のアプローチを使用できます:

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

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

YAML v3 による動的アンマーシャリング

Yaml v3 の場合、アプローチは少し異なります:

type Spec struct {
    Kind string      `yaml:"kind"`
    Spec interface{} `yaml:"-"
}

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

結論

これらの動的なアンマーシャリング手法により、YAML フィールドを構造体の有限セットに柔軟に解析できるようになり、提案された回避策よりも洗練された効率的なソリューションが提供されます。提供されているコード スニペットを自由に調べて、特定の要件に基づいてアプローチを最適化してください。