使用环境变量替换 Viper 配置的疑问

有志者，事竟成！如果你在学习Golang，那么本文《使用环境变量替换 Viper 配置的疑问》，就很适合你！文章讲解的知识点主要包括，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

我们使用 viper 来读取和解析我们的配置文件，所有这些都没有任何问题。

但是，我们无法使用环境变量覆盖某些配置值。这些是配置绑定到结构或结构数组的特定用例。

这是我们的 config.yaml 中的示例：

app:
    verifiers:
      - name: "test1"
        url: "http://test1.url"
        cache: "5000ms"
      - name: "test2"
        url: "http://test2.url"
        cache: "10000ms"

它绑定到以下结构（golang）：

type app struct {
    appconfig   config      `yaml:"app" mapstructure:"app"`
}

type config struct {
    verifiers []verifierconfig `json:"verifiers"         yaml:"verifiers"         mapstructure:"verifiers"`
}

type verifierconfig struct {
    name     string            `json:"name"                   yaml:"name"                   mapstructure:"name"`
    url      string            `json:"url,omitempty"          yaml:"url,omitempty"          mapstructure:"url"`
    cache    jsontime.duration `json:"cache"                  yaml:"cache"                  mapstructure:"cache"`
}

我们无法使用环境变量覆盖验证程序的值。

以下是我们使用过的 viper 选项：

viper.AutomaticEnv()
viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))

有人遇到过类似的问题或者可以确认 viper 不支持这样的用例吗？

任何指示将不胜感激。

谢谢

解决方案

viper 无法读取配置文件中的环境变量。您必须在使用值的 .go 文件中替换它们。以下是我如何实现目标的示例：

要求 viper 考虑带有“env”前缀的系统环境变量，这意味着 viper 期望“env_mongo_user”和“env_mongo_password”作为系统环境变量

viper.automaticenv()
viper.setenvprefix(viper.getstring("env"))
viper.setenvkeyreplacer(strings.newreplacer(".", "_"))

在 yml 文件中：

mongodb:
 url: "mongodb+srv://${username}:${password}@xxxxxxx.mongodb.net/"
 database: "test"

在 mongodb 连接文件中：（在建立数据库连接之前，将 ${username} 和 ${password} 替换为各自的环境变量，但 viper 将读取这些没有前缀的变量）

const BasePath = "mongodb"
mongoDbUrl := viper.GetString(fmt.Sprintf("%s.%s", BasePath, "url"))
username := viper.GetString("MONGO_USER")
password := viper.GetString("MONGO_PASSWORD")
replacer := strings.NewReplacer("${username}", username, "${password}", password)
mongoDbUrl = replacer.Replace(mongoDbUrl)

理论要掌握，实操不能落！以上关于《使用环境变量替换 Viper 配置的疑问》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！