Как установить значения по умолчанию для ключей в списках при отмене вызова Yaml в Go?

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. Ручной цикл по списку структур. Хотя этот метод работает со строками, он не работает с логическими значениями, потому что они уже инициализированы с false после демаршаллинга, и мы точно не знаем, введен ли пользователем false или нет.

4. Использование указателей с логическими значениями. Это работает, поскольку неинициализированные значения равны nil, и их легко поймать. Однако, это потребует разыменования указателей при использовании структуры config, что не очень удобно. Кроме того, новая структура может быть сгенерирована на основе той, которая получена в результате демаршаллинга.

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

Вы можете использовать функцию для выполнения работы:

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
}

EDIT

Или, что еще лучше, просто переверните логику ваших логических значений:

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

Таким образом, SSL будет использоваться, если в конфигурации явно не указано иное, что выделяется, когда кто-то читает конфигурацию.

Answer 2

Я понял это.Я использую github.com/creasty/defaults в обратном вызове UnmarshalYAML.

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 для целей проверки.