如何在 Go 中使用动态键解析 YAML？

使用 Golang 在 YAML 中进行动态密钥解析

简介

YAML（YAML 不是）标记语言）是一种广泛使用的数据序列化格式，允许灵活的数据表示。然而，在 Go 中解析具有动态键（即可能变化的键）的 YAML 文件会带来挑战，因为该语言严重依赖静态类型结构。

问题描述

您有一个 YAML 文件，其中密钥代表不同的版本（例如 V1、V2、V3）。用户可以指定他们支持哪些版本，并且密钥可能不按顺序或跳过版本。您想要将此 YAML 文件解析为 Go 结构体，但该结构体的字段必须与动态键匹配。

不正确的方法

最初，您尝试解析 YAML文件放入环境结构中，期望键存在于根级别。然而，根元素的实际类型是字符串（代表环境名称）到环境结构的映射。

解决方案

要使用动态键解析 YAML，您需要需要在结构上使用自定义 UnmarshalYAML 方法来解析键并构造正确的结构。在这种情况下，Environment 结构需要 UnmarshalYAML 方法将根解组为映射，然后解析各个环境。

代码示例

这里是一个示例环境结构的自定义 UnmarshalYAML 方法：

<code class="go">func (e *Environment) UnmarshalYAML(unmarshal func(interface{}) error) error {
    var params struct {
        SkipHeaderValidation bool `yaml:"skip-header-validation"`
    }
    if err := unmarshal(&params); err != nil {
        return err
    }
    var versions map[string]MajorVersion
    if err := unmarshal(&versions); err != nil {
        // Here we expect an error because a boolean cannot be converted to a MajorVersion
        if _, ok := err.(*yaml.TypeError); !ok {
            return err
        }
    }
    e.SkipHeaderValidation = params.SkipHeaderValidation
    e.Versions = versions
    return nil
}</code>

此方法首先解组布尔字段 SkipHeaderValidation。然后，它尝试将其余数据解组到版本名称到 MajorVersion 结构的映射中。但是，由于未编组的数据是布尔值和 MajorVersion 结构的混合，因此将引发 yaml.TypeError 。该代码会忽略此错误，并将未编组的版本映射分配给环境结构的 Versions 字段。

示例用法

使用自定义 UnmarshalYAML 方法，您可以现在正确解组 YAML 数据：

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

这将产生以下输出：

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

这演示了如何成功将带有动态键的 YAML 文件解析为 Go 结构。

以上是如何在 Go 中使用动态键解析 YAML？的详细内容。更多信息请关注PHP中文网其他相关文章！