当在GO中删除YAML时，在列表中键上键值设置默认值的方法是什么?

Question

我在下面有一个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")

2. https://github.com/creasty/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)

3. 手动循环浏览结构列表。尽管这些方法可与字符串一起使用，但它与布尔值不使用，因为它们已经在umarshalling后使用false初始化，并且我们不确定用户是否输入了false。

4. 使用带有布尔值的指针。这是因为非初始化的值等于nil，并且很容易捕获。但是，使用Config Struct时，它将需要取消指针，这不是很方便。另外，可以根据来自Unmarshalling的一个生成新结构。

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

5. 使用hashmap而不是struct。之所以起作用，是因为没有初始化空值，但是，它需要在访问它们或将地图转换为结构之前对映射元素进行检查。

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

有什么想法？

Answer 1

我弄清楚了。我在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一起验证目的。

Answer 2

您可以使用功能来完成工作：

type Connection struct {
    Name      string
    Hosts     []string
    Username  string
    Password  string
    Ssl       *bool
    SslVerify bool
    Version   int
}
// If Ssl is nil, returns true
// otherwise the value
func (c Connection) IsSSL() bool {
    return c.Ssl == nil || *c.Ssl
}
type Config struct {
    Connections []Connection
}

编辑

或更好的是，只需扭转布尔值的逻辑：

type Connection struct {
    Name          string
    Hosts         []string
    Username      string
    Password      string
    SkipSsl       bool
    SkipSslVerify bool
    Version       int
}

以这种方式，除非在config中明确告知不同的SSL，否则当有人读取config。