如何测试 Golang 中已初始化的 YAML 的默认值？

在 Golang 中使用 YAML 包时，如何测试已初始化但未在 YAML 文件中指定的默认值是一个常见问题。作者希望当 `verify_ssl` 字段不存在时将其设置为 `true`，而不是默认的 `false`。通过将 `verify_ssl` 字段更改为使用指针并添加 `omitempty` 标记，作者可以取消引用结构中的字段以获取 YAML 解组的真实值。如果字段不存在，则字段为 `nil`，作者可以设置默认值。

我正在编写守护程序的配置处理程序并利用 yaml 包来执行此操作。导入我的文件的工作方式如下：

package daemon

import (
    "ioutil"
    "log"

    "gopkg.in/yaml.v2"
)

type daemonconfig struct {
    bindaddress string `yaml:"bind_address"`
    bindport    int    `yaml:"bind_port"`
    verifyssl   bool   `yaml:"verify_ssl"`
}

我可以轻松地从 yaml 文件中解组数据，如下所示：

func (config *daemonconfig) getconf() *daemonconfig {
    yamlfile, err := ioutil.readfile("config.yaml")
    if err != nil {
        log.fatal("unable to open config.yaml:", err)
    }
    err = yaml.unmarshal(yamlfile, config)
    if err != nil {
        log.fatal("failed to unmarshall config.yaml:", err)
    }
    config, err = setdefaults(config)

    return config
}

我的问题是关于我的自定义 setdefaults 函数。如果未提供字段，例如 bind_port 或 bind_address，我只想将它们设置为默认值：

func setdefaults(config *daemonconfig) (*daemonconfig, error) {
    if len(config.bindaddress) <= 0 {
        config.bindaddress = "0.0.0.0"
    }
    if config.bindport == 0 {
        config.bindport = 9999
    }

    return config, nil
}

您会注意到我没有将默认值设置为 verify_ssl；当 yaml 解组此字段并且无法找到该字段时，它将 bool 初始化为 false，这与我想要的默认行为恰恰相反。我希望用户明确将 ssl 验证设置为关闭，而不是在未指定的情况下默认关闭它。如果我有一个完全空的 config.yaml，并且期望设置默认值，它总是将 verify_ssl 设置为 false（此日志来自应用程序中的其他位置）：

2019/02/19 03:56:08 Currently loaded config: {0.0.0.0 9999 false}

如果正在解组的 yaml 字段中不存在该字段，我将如何检查该行是否存在？我可以手动读取文件并首先检查该参数，但我想知道是否有一种更优雅的方式来处理我所拥有的；否则，我将仅使用 ioutil 和字符串检查来完成此操作。谢谢！

解决方案

感谢 Thomas 和 mh-cbon，我发现我可以使用指针来实现此目的。如果我更改 verifyssl 字段以使用指针并添加 omity：

type daemonconfig struct {
    bindaddress string `yaml:"bind_address"`
    bindport    int    `yaml:"bind_port"`
    verifyssl   *bool  `yaml:"verify_ssl",omitempty`
}

我可以简单地取消引用结构中的字段来获取 yaml 解组的真实值。设置默认值非常简单：

func setdefaults(config *daemonconfig) (*daemonconfig, error) {
    if len(config.bindaddress) <= 0 {
        config.bindaddress = "0.0.0.0"
    }
    if config.bindport == 0 {
        config.bindport = 9999
    }
    if config.verifyssl == nil {
        ssl := true
        config.verifyssl = &ssl
    }

    return config, nil
}

如果它加载到文件中，config.verifyssl 将包含 bool 的地址，因此它不会是 nil，如果它确实根本不在文件中，那么该字段只是 nil 并且我可以设置它或者通过指针获取它。

编辑： 请注意，如果您的 verifyssl 指针为零，则不能像这样分配它：

*config.VerifySSL = true

这是取消引用 nil 指针。我更新了解决方案以包含更易于管理的内容。

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~