Viper 将环境变量映射到结构成员

知识点掌握了，还需要不断练习才能熟练运用。下面golang学习网给大家带来一个Golang开发实战，手把手教大家学习《Viper 将环境变量映射到结构成员》，在实现功能的过程中也带大家重新温习相关知识点，温故而知新，回头看看说不定又有不一样的感悟！

这是我的 config.yaml 文件

server:
  port: 5000

这是我使用 viper 解组的逻辑

type configurations struct {
    server serverconfig
}

type serverconfig struct {
    port int
}

// loadconfig reads configuration from file or environment variables.
func loadconfig(path string) (config configurations, err error) {
    viper.addconfigpath(path)
    viper.setconfigname("config")
    viper.setconfigtype("yaml")

    viper.automaticenv()

    err = viper.readinconfig()
    if err != nil {
        return
    }

    err = viper.unmarshal(&config)
    return
}

我已经设置了环境变量export port=9000，现​​在我的问题是如何将环境变量port映射到configurations.serverconfig.port。 我想要具有开发默认值的配置文件，并且希望我的应用程序从生产环境变量中读取/覆盖配置，如何使用 viper 实现此目的。

我看到以下选项

1. 设置 server_port，然后使用如下替换器将其解组为结构体

replacer := strings.NewReplacer(".", "_")
viper.SetEnvKeyReplacer(replacer)

1. go viper .yaml 值环境变量覆盖

但是我想将变量 port 映射到结构成员，可能是因为我的托管服务提供商设置了 port env 变量，并且我无法使用第二点，因为我希望使用 yaml 作为默认值。

解决方案

您可以使用 viper.bindenv(string...) error 将特定环境变量绑定到配置。

来自 Working with Environment Variables：

bindenv 接受一个或多个参数。第一个参数是键名，其余参数是绑定到该键的环境变量的名称。如果提供了多个，则它们将按指定的顺序优先。

这里端口的配置键是server.port，环境变量port应该绑定到它。

...
viper.AutomaticEnv()


err = viper.BindEnv("server.port", "PORT")
if err != nil {
    return
}


err = viper.ReadInConfig()
...

完成此更改后，像 port=9999 ./yourbinary 这样的调用将选取 port 并将其写入 server.port。从那里，它将被解组到您的结构中。

今天关于《Viper 将环境变量映射到结构成员》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！