在開發Go應用程式時，處理配置是一個常見的需求。配置可能來自於配置文件、環境變數、命令行參數等等。Viper是一個強大的庫，可以幫助我們處理這些配置。 什麼是Viper？ Viper是一個應用程式配置解決方案，用於Go應用程式。它支持JSON、TOML、YAML、HCL、envfile和Java p ...
在開發Go應用程式時，處理配置是一個常見的需求。配置可能來自於配置文件、環境變數、命令行參數等等。Viper是一個強大的庫，可以幫助我們處理這些配置。
什麼是Viper？
Viper是一個應用程式配置解決方案，用於Go應用程式。它支持JSON、TOML、YAML、HCL、envfile和Java properties配置文件格式。它還支持環境變數、命令行標誌、遠程配置系統（如etcd或Consul），並且可以直接監控配置文件的變化。
如何使用Viper？
讀取配置文件
首先，我們需要告訴Viper去哪裡找配置文件，以及如何讀取它們。這可以通過以下代碼實現：
viper.SetConfigName("config") // 配置文件名（不帶尾碼）
viper.SetConfigType("yaml") // 如果配置文件名沒有擴展名，需要設置此項
viper.AddConfigPath("/etc/appname/") // 查找配置文件所在的路徑
viper.AddConfigPath("$HOME/.appname") // 多次調用以添加多個搜索路徑
err := viper.ReadInConfig() // 查找並讀取配置文件
if err != nil { // 處理讀取配置文件的錯誤
panic(fmt.Errorf("Fatal error config file: %s \n", err))
}
獲取配置值
一旦Viper讀取了配置文件，我們就可以使用Get函數來獲取配置值：
port := viper.GetInt("port")
databaseDriver := viper.GetString("database.driver")
設置預設值
Viper也允許我們設置配置的預設值。這在配置項未在配置文件中定義，但我們又需要一個預設值的情況下非常有用：
viper.SetDefault("ContentDir", "content")
viper.SetDefault("LayoutDir", "layouts")
使用環境變數
Viper可以讀取環境變數：
viper.AutomaticEnv()
databaseUrl := viper.Get("DATABASE_URL")
使用命令行標誌
Viper也可以與標準庫的flag包一起使用，以處理命令行標誌：
var cfgFile string
func init() {
flag.StringVar(&cfgFile, "c", "", "config file")
flag.Parse()
}
func main() {
if cfgFile != "" { // 如果指定了配置文件，則解析指定的配置文件
viper.SetConfigFile(cfgFile)
err := viper.ReadInConfig()
if err != nil {
log.Fatalf("Error reading config file, %s", err)
}
}
// ...
}
監控配置變化並重新載入配置
Viper甚至可以監控配置文件的變化，併在文件變化時重新載入配置：
viper.WatchConfig()
viper.OnConfigChange(func(e fsnotify.Event) {
fmt.Println("Config file changed:", e.Name)
})
最後
Viper是一個強大的庫，可以幫助我們處理Go應用程式的配置。它支持多種配置源，包括配置文件、環境變數、命令行參數等等，使得處理配置變得簡單而直觀。
聲明：本作品採用署名-非商業性使用-相同方式共用 4.0 國際 (CC BY-NC-SA 4.0)進行許可，使用時請註明出處。
Author: mengbin
blog: mengbin
Github: mengbin92
cnblogs: 戀水無意
