Bagaimana untuk Menghuraikan YAML dengan Kekunci Dinamik dalam Go Menggunakan Custom Unmarshaler?

Menghuraikan YAML dengan Dynamic Key in Go

Masalah ini timbul apabila menghuraikan fail YAML dengan kunci yang boleh berbeza-beza bergantung pada versi API yang ditentukan pengguna . Pengguna boleh meninggalkan atau menyusun versi seperti yang diperlukan (cth., V0, V2, V5).

Pendekatan Asal dengan Struktur Persekitaran

Pendekatan awal melibatkan menghurai YAML sebagai struct Persekitaran. Walau bagaimanapun, jenis sebenar data yang dihuraikan ialah map[string]Environment.

<code class="go">type Environment struct {
  SkipHeaderValidation  bool        `yaml:"skip-header-validation"`
  Version               map[string]MajorVersion
}</code>

Untuk menyelesaikan masalah ini, kami memerlukan Unmarshaler tersuai.

Custom Unmarshaler for Environment

Kami mencipta kaedah Unmarshaler tersuai untuk jenis Persekitaran yang menghuraikan kedua-dua kunci statik dan dinamik.

<code class="go">func (e *Environment) UnmarshalYAML(unmarshal func(interface{}) error) error {
    var params struct {
        SkipHeaderValidation bool `yaml:"skip-header-validation"`
    }
    if err := unmarshal(&amp;params); err != nil {
        return err
    }
    var versions map[string]MajorVersion
    if err := unmarshal(&amp;versions); err != nil {
        // Expect error here due to boolean to MajorVersion conversion
        if _, ok := err.(*yaml.TypeError); !ok {
            return err
        }
    }
    e.SkipHeaderValidation = params.SkipHeaderValidation
    e.Versions = versions
    return nil
}</code>

Penghuraian dan Output Ubahsuai

Kami kini boleh mengubah suai kod penghuraian kami untuk mengendalikan kekunci dinamik dan akhirnya mengembalikan struktur data yang diingini.

Fungsi Utama Diubahsuai

<code class="go">func main() {
    var e map[string]Environment
    if err := yaml.Unmarshal([]byte(data), &amp;e); err != nil {
        fmt.Println(err.Error())
    }
    fmt.Printf("%#v\n", e)
}</code>

Output

Output kini akan mencerminkan struktur data yang dijangkakan dengan kunci dinamik dihuraikan dengan sewajarnya.

map[string]main.Environment{
    "development": {
        SkipHeaderValidation: true,
        Versions:             {
            "V2": {
                Current:                "2.0.0",
                MimeTypes:              {"application/vnd.company.jk.identity+json;", "application/vnd.company.jk.user+json;", "application/vnd.company.jk.role+json;", "application/vnd.company.jk.scope+json;", "application/vnd.company.jk.test+json;"},
                SkipVersionValidation:  false,
                SkipMimeTypeValidation: false,
            },
            "V1": {
                Current:                "1.0.0",
                MimeTypes:              {"application/vnd.company.jk.identity+json;", "application/vnd.company.jk.user+json;", "application/vnd.company.jk.role+json;", "application/vnd.company.jk.scope+json;", "application/vnd.company.jk.test+json;"},
                SkipVersionValidation:  true,
                SkipMimeTypeValidation: true,
            },
        },
    },
}

Atas ialah kandungan terperinci Bagaimana untuk Menghuraikan YAML dengan Kekunci Dinamik dalam Go Menggunakan Custom Unmarshaler?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!