Go 中解组 Yaml：为列表中的键设置默认值

亲爱的编程学习爱好者，如果你点开了这篇文章，说明你对《Go 中解组 Yaml：为列表中的键设置默认值》很感兴趣。本篇文章就来给大家详细解析一下，主要介绍一下，希望所有认真读完的童鞋们，都有实质性的提高。

我有一个如下所示的 yaml。

connections:
  - name: demo
    hosts:
      - demo.example.com:9200
    username: admin
    password: password
    ssl: true
    ssl_verify: true
    version: 7
  - name: test
    hosts:
      - "test.example.com:9200"
    username: admin
    password: password

正如您所看到的，ssl 和 ssl_verify 未在列表的第二项中设置。我希望它们默认为 true，但是，它没有发生。我尝试了不同的解决方案。

1. viper 默认值 - 不起作用。

viper.setdefault("connections[].ssl", "true")

1. https://github.com/cresty/defaults - 不起作用。

type config struct {
    connections []struct {
        name      string
        hosts     []string
        username  string
        password  string
        ssl       bool `default:"true"`
        sslverify bool `default:"true"`
        version   int
    }
}

...

err := defaults.set(config)

1. 手动循环结构列表。虽然这些方法适用于字符串，但它不适用于布尔值，因为它们在解组后已使用 false 进行初始化，并且我们不确定 false 是否由用户输入。

2. 使用带有布尔值的指针。这是因为未初始化的值等于 nil 并且它们很容易捕获。但是，使用 config struct 时需要取消引用指针，这不是很方便。或者，可以根据解组产生的结构生成一个新的结构。

type Config struct {
    Connections []struct {
        Name      string
        Hosts     []string
        Username  string
        Password  string
        Ssl       *bool
        SslVerify *bool
        Version   int
    }
}

1. 使用 hashmap 而不是 struct。这是可行的，因为空值未初始化，但是，在访问映射元素或将映射转换为结构之前，需要对它们进行检查。

解决方案 4 和 5 可能会起作用，但我想知道是否有比这更好的方法。

有什么想法吗？

解决方案

我明白了。我在 unmarshalyaml 回调中使用 github.com/creasty/defaults。

type Config struct {
    Connections []Connection
}

type Connection struct {
    Name      string
    Hosts     []string
    Username  string
    Password  string
    Ssl       bool `default:"true"`
    SslVerify bool `default:"true" yaml:"ssl_verify"`
    Version   int  `version:"7"`
}

func (s *Connection) UnmarshalYAML(unmarshal func(interface{}) error) error {
    defaults.Set(s)

    type plain Connection
    if err := unmarshal((*plain)(s)); err != nil {
        return err
    }

    return nil
}

此外，该解决方案与 https://github.com/dealancer/validate 配合良好，可用于验证目的。

到这里，我们也就讲完了《Go 中解组 Yaml：为列表中的键设置默认值》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！