Bagaimanakah Saya Boleh Menghuraikan Medan YAML secara Dinamik kepada Struktur Tertentu dalam Go?

Menghuraikan Medan YAML secara Dinamik kepada Struktur Pratakrif dalam Go

Untuk fail YAML di mana medan tertentu boleh diwakili oleh sebarang nilai daripada set struct yang telah ditetapkan, satu yang sesuai pendekatan adalah dengan menggunakan kaedah UnmarshalYAML pakej YAML. Ini membolehkan penciptaan logik unmarshaling tersuai untuk jenis tertentu.

YAML v2

Menggunakan YAML v2, kod berikut mencapai gelagat yang diingini:

<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

Untuk YAML v3, gunakan kod berikut:

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

Pendekatan ini menyediakan cara yang mudah dan elegan untuk menyahmarshal data YAML ke dalam set struct yang telah ditetapkan sambil mengelakkan langkah penghuraian tambahan atau penggunaan memori yang berlebihan.

Atas ialah kandungan terperinci Bagaimanakah Saya Boleh Menghuraikan Medan YAML secara Dinamik kepada Struktur Tertentu dalam Go?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!