Golang工具庫viper的使用教程
viper 是 go 項目中用來讀取配置文件的庫,支持讀取 yaml、toml、json、hcl、env 等格式的配置文件
讀取環(huán)境變量
viper 可以讀取終端的環(huán)境變量
如果你用的是 zsh 在 ~/.zshrc 中設(shè)置,如果是 bash 在 ~/.bashrc 中設(shè)置
寫環(huán)境變量時,等號兩邊不能有空格,否則會報錯
export APP_DEBUG=true
寫好后,記得 source 一下,或者重啟終端
source ~/.zshrc
使用 viper.AutomaticEnv 將環(huán)境變量都讀取到 viper 中
然后就可以使用 viper.GetBool("APP_DEBUG") 來獲取環(huán)境變量了
func GetEnvInfo(env string) bool {
viper.AutomaticEnv()
return viper.GetBool(env)
}
app_debug := GetEnvInfo("APP_DEBUG")
fmt.Println(app_debug) // true讀取本地的配置文件
新建一個 config-debug.yaml 文件,內(nèi)容如下:
host: "127.0.0.1" port: 50051
然后使用 viper 讀取,setConfigFile 填寫的路徑是相對于 main.go 的路徑
使用 viper.ReadInConfig 來讀取配置文件
viper.SetConfigFile("./config-debug.yaml")
if err := viper.ReadInConfig(); err != nil {
panic(err)
}
fmt.Println(viper.GetString("host")) // "127.0.0.1"使用 viper.SetConfigName 來設(shè)置文件名,viper.SetConfigType 來設(shè)置文件類型,可以更靈活的讀取配置文件
viper.SetConfigName設(shè)置文件名,不需要帶后綴viper.SetConfigType設(shè)置文件類型viper.AddConfigPath設(shè)置文件路徑
viper.SetConfigName("config-debug")
viper.SetConfigType("yaml")
viper.AddConfigPath("./")
if err := viper.ReadInConfig(); err != nil {
panic(err)
}
fmt.Println(viper.GetString("host")) // "127.0.0.1"將配置文件中的配置映射到結(jié)構(gòu)體中
每次使用 viper.GetString 來獲取配置文件中的配置,都會寫很多重復(fù)的代碼,可以將配置文件中的配置映射到結(jié)構(gòu)體中
新建結(jié)構(gòu)體,結(jié)構(gòu)體的字段名要和配置文件中的字段名一致,使用 mapstructure 標(biāo)簽來映射
type Server struct {
Host string `mapstructure:"host"`
Port int `mapstructure:"port"`
}使用 viper.Unmarshal 將配置文件中的配置映射到結(jié)構(gòu)體中
var server Server
viper.SetConfigName("config-debug")
viper.SetConfigType("yaml")
viper.AddConfigPath("./")
if err := viper.ReadInConfig(); err != nil {
panic(err)
}
if err := viper.Unmarshal(&server); err != nil {
panic(err)
}
fmt.Println(server.Host) // "127.0.0.1"監(jiān)聽配置信息的變化
如果配置文件發(fā)生變化,viper 支持在運行時監(jiān)聽配置文件的變化
使用 viper.WatchConfig 來監(jiān)聽配置文件的變化,使用 viper.OnConfigChange 來監(jiān)聽配置文件的變化
onConfigChange 傳入的參數(shù)是一個函數(shù),函數(shù)的參數(shù)是 fsnotify.Event,在函數(shù)中重新讀取配置文件,然后將配置文件中的配置映射到結(jié)構(gòu)體中
viper.OnConfigChange(func(e fsnotify.Event) {
_ = viper.ReadInConfig()
_ = viper.Unmarshal(&server)
})
viper.WatchConfig()到此這篇關(guān)于Golang工具庫viper的使用教程的文章就介紹到這了,更多相關(guān)Golang viper內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
詳解prometheus監(jiān)控golang服務(wù)實踐記錄
這篇文章主要介紹了詳解prometheus監(jiān)控golang服務(wù)實踐記錄,本文給大家介紹的非常詳細(xì),對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-11-11
go mutex互斥鎖使用Lock和Unlock方法占有釋放資源
Go號稱是為了高并發(fā)而生的,在高并發(fā)場景下,勢必會涉及到對公共資源的競爭,當(dāng)對應(yīng)場景發(fā)生時,我們經(jīng)常會使用 mutex 的 Lock() 和 Unlock() 方法來占有或釋放資源,雖然調(diào)用簡單,但 mutex 的內(nèi)部卻涉及挺多的,本文來好好研究一下2023-09-09

