Question

以下是我的配置文件，采用toml格式...

[[hosts]]
name = "host1"
username = "user1"
password = "password1"

[[hosts]]
name = "host2"
username = "user2"
password = "password2"

...这是我加载它的代码：

import (
    "fmt"
    "github.com/spf13/viper"
    "strings"
)

type Config struct {
    Hosts []Host
}

type Host struct {
    Name      string `mapstructure:"name"`
    Username  string `mapstructure:"username"`
    Password  string `mapstructure:"password"`
}

func main() {
    viper.AddConfigPath(".")
    viper.AddConfigPath("./config")
    viper.SetConfigName("app")

    if err := viper.ReadInConfig(); err != nil {
        return nil, fmt.Errorf("error reading config file, %v", err)
    }

    config := new(Config)
    if err := viper.Unmarshal(config); err != nil {
        return nil, fmt.Errorf("error parsing config file, %v", err)
    }

    var username, password string

    for i, h := range config.Hosts {
        if len(h.Name) == 0 {
            return nil, fmt.Errorf("name not defined for host %d", i)
        }

        if username = os.Getenv(strings.ToUpper(h.Name) + "_" + "USERNAME"); len(username) > 0 {
            config.Hosts[i].Username = username
        } else if len(h.Username) == 0 {
            return nil, fmt.Errorf("username not defined for %s", e.Name)
        }

        if password = os.Getenv(strings.ToUpper(e.Name) + "_" + "PASSWORD"); len(password) > 0 {
            config.Hosts[i].Password = password
        } else if len(h.Password) == 0 {
            return nil, fmt.Errorf("password not defined for %s", e.Name)
        }

        fmt.Printf("Hostname: %s", h.name)
        fmt.Printf("Username: %s", h.Username)
        fmt.Printf("Password: %s", h.Password)
    }
}

例如，我首先检查环境变量HOST1_USERNAME1，HOST1_PASSWORD1，HOST2_USERNAME2和HOST2_PASSWORD2是否存在......如果存在，那么我设置配置项他们的值，否则我尝试从配置文件中获取值。

我知道viper提供方法AutomaticEnv来执行类似的操作......但它是否适用于像我这样的集合（AutomaticEnv应在环境变量绑定后调用）？

根据我上面的代码，是否可以使用viper提供的机制并删除os.GetEnv？

感谢。

更新

以下是我更新的代码。在我定义的环境变量HOST1_USERNAME和HOST1_PASSWORD中，将配置文件中的相应设置设置为空字符串。

这是我的新配置文件：

[host1] username = "" password = "" [host2] username = "user2" password = "password2"

这是我的代码：

package config import ( "fmt" "github.com/spf13/viper" "strings" "sync" ) type Config struct { Hosts []Host } type Host struct { Name string Username string Password string } var config *Config func (c *Config) Load() (*Config, error) { if config == nil { viper.AddConfigPath(".") viper.AddConfigPath("./config") viper.SetConfigName("myapp") viper.AutomaticEnv() viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_")) if err := viper.ReadInConfig(); err != nil { return nil, fmt.Errorf("error reading config file, %v", err) } allSettings := viper.AllSettings() hosts := make([]Host, 0, len(allSettings)) for key, value := range allSettings { val := value.(map[string]interface{}) if val["username"] == nil { return nil, fmt.Errorf("username not defined for host %s", key) } if val["password"] == nil { return nil, fmt.Errorf("password not defined for host %s", key) } hosts = append(hosts, Host{ Name: key, Unsername: val["username"].(string), Password: val["password"].(string), }) } config = &Config{hosts} } return config, nil }

我现在工作（感谢Chrono Kitsune），我希望它有所帮助， J3D

Answer 1

来自viper.Viper：

来源的优先顺序如下：


覆盖

标志

ENV。变量

配置文件

key / value store

默认

在确定环境变量的名称时可能会遇到问题。您基本上需要一种方法将hosts[0].Username绑定到环境变量HOST1_USERNAME。但是，目前在viper中没有办法做到这一点。实际上，viper.Get("hosts[0].username")会返回nil，这意味着数组索引显然不能与viper.BindEnv一起使用。您还需要将此功能用于可以定义的主机数，这意味着如果您列出了20台主机，则需要拨打viper.BindEnv 40或60次，具体取决于主机名称是否可以被环境变量覆盖。要解决此限制，您需要动态地将主机作为独立表而不是表数组来使用：

[host1]
username = "user1"
password = "password1"

[host2]
username = "user2"
password = "password2"

然后，您可以使用viper.SetEnvKeyReplacer和strings.Replacer来处理环境变量问题：

// host1.username => HOST1_USERNAME
// host2.password => HOST2_PASSWORD
// etc.
viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))

请注意，在撰写本文时some bugs exist when it comes to the resolution order。此问题会影响viper.Unmarshal和viper.Get：环境变量应覆盖配置文件值，但目前仍在使用配置文件值。奇怪的是，viper.AllSettings工作正常。如果没有，则无法执行以下操作，以上述格式使用主机：

// Manually collect hosts for storing in config.
func collectHosts() []Host {
    hosts := make([]Host, 0, 10)
    for key, value := range viper.AllSettings() {
        // viper.GetStringMapString(key)
        // won't work properly with environment vars, etc. until
        //   https://github.com/spf13/viper/issues/309
        // is resolved.
        v := value.(map[string]interface{})
        hosts = append(hosts, Host{
            Name:     key,
            Username: v["username"].(string),
            Password: v["password"].(string),
        })
    }
    return hosts
}

总结一下：

应该从第一个提供的值中获取值：覆盖，标志，环境变量，配置文件，键/值存储，默认值。不幸的是，这并不总是遵循的顺序（因为错误）。
您需要更改配置格式并使用字符串替换器来充分利用viper.AutomaticEnv的便利性。

spf13 Viper：如何将数组中的配置项绑定到环境变量

1 个答案: