Go-Gateway-3:参数配置化

最新推荐文章于 2024-03-28 09:52:22 发布

無炆_

最新推荐文章于 2024-03-28 09:52:22 发布

阅读量174

点赞数

分类专栏： Go

本文链接：https://blog.csdn.net/u013205724/article/details/112758787

版权

Go 专栏收录该内容

10 篇文章 1 订阅

订阅专栏

使用 viper 读取配置

go get -u github.com/spf13/viper

配置读取和监控

// 读取配置
func (c *Config) InitConfig() error {
    if c.Name != "" {
        viper.SetConfigFile(c.Name)
    } else {
        viper.AddConfigPath("conf")
        viper.SetConfigName("config")
    }
    viper.SetConfigType("yaml")

    // 从环境变量总读取
    viper.AutomaticEnv()
    viper.SetEnvPrefix("web")
    viper.SetEnvKeyReplacer(strings.NewReplacer("_", "."))

    return viper.ReadInConfig()
}

// 监控配置改动
func (c *Config) WatchConfig(change chan int) {
    viper.WatchConfig()
    viper.OnConfigChange(func(e fsnotify.Event) {
        log.Printf("配置已经被改变: %s", e.Name)
        change <- 1
    })
}