Viper 是 Go 语言中一个非常流行的配置管理库，旨在使配置文件的读取、解析和管理变得更加简单。它支持多种配置文件格式（如 JSON、YAML、TOML、HCL、INI 等）以及环境变量、命令行标志等不同来源的配置，并且支持配置文件的热加载（即动态更新配置而无需重启应用）。
你可以使用 go get 安装 Viper：
Life is like a journey
go get github.com/spf13/viper
Viper 支持加载不同格式的配置文件，下面是一个加载 YAML 配置文件的示例：
package main
import (
"fmt"
"github.com/spf13/viper"
)
func main() {
// 设置配置文件类型
viper.SetConfigType("yaml")
// 设置配置文件路径
viper.AddConfigPath(".") // 以当前目录为路径
viper.SetConfigName("config") // 配置文件名，不需要包含扩展名
// 读取配置文件
if err := viper.ReadInConfig(); err != nil {
fmt.Println("Error reading config file", err)
return
}
// 获取配置项
appName := viper.GetString("app.name")
port := viper.GetInt("app.port")
fmt.Printf("App Name: %s, Port: %d\n", appName, port)
}
上面的代码中，config.yaml 可能是这样的一个文件：
app:
name: MyApp
port: 8080
Viper 支持多种格式的配置文件，除了 yaml，还可以是 JSON、TOML、INI 等格式。例如，要加载一个 JSON 配置文件，可以这样设置：
viper.SetConfigType("json")
viper.AddConfigPath(".")
viper.SetConfigName("config")
viper.ReadInConfig()
Viper 允许你将环境变量与配置项进行绑定。这使得你可以轻松地从环境变量中读取配置，而不需要额外的配置文件。
package main
import (
"fmt"
"github.com/spf13/viper"
"os"
)
func main() {
// 设置默认值
viper.SetDefault("app.port", 8080)
// 绑定环境变量
viper.BindEnv("app.port", "APP_PORT")
// 从环境变量读取配置
port := viper.GetInt("app.port")
if port == 0 {
port = 8080
}
// 输出环境变量的配置
fmt.Printf("App Port: %d\n", port)
// 设置环境变量
os.Setenv("APP_PORT", "9090")
port = viper.GetInt("app.port")
fmt.Printf("Updated App Port from Env: %d\n", port)
}
如果你设置了 APP_PORT=9090，它会自动读取并覆盖 viper.GetInt("app.port") 中的值。
Viper 可以与 flag 包结合使用，自动将命令行参数与配置项进行绑定，使得命令行参数优先级更高。
package main
import (
"flag"
"fmt"
"github.com/spf13/viper"
)
func main() {
// 设置默认值
viper.SetDefault("app.port", 8080)
// 解析命令行参数
portFlag := flag.Int("port", 8080, "App Port")
flag.Parse()
// 将命令行标志与 viper 绑定
viper.BindPFlag("app.port", flag.Lookup("port"))
// 获取最终配置
port := viper.GetInt("app.port")
fmt.Printf("App Port: %d\n", port)
}
你可以通过命令行传递参数来覆盖配置文件中的设置。例如：
go run main.go --port 9090
Viper 支持嵌套的配置结构，可以直接读取嵌套的字段。
app:
name: MyApp
settings:
theme: dark
language: en
读取嵌套配置：
theme := viper.GetString("app.settings.theme")
language := viper.GetString("app.settings.language")
fmt.Printf("Theme: %s, Language: %s\n", theme, language)
Viper 支持配置文件的热加载功能，可以在程序运行时动态更新配置，而不需要重启应用。
package main
import (
"fmt"
"github.com/spf13/viper"
"time"
)
func main() {
viper.SetConfigType("yaml")
viper.AddConfigPath(".")
viper.SetConfigName("config")
// 读取配置文件
err := viper.ReadInConfig()
if err != nil {
fmt.Println("Error reading config file", err)
return
}
// 监听文件变化并重新加载
viper.WatchConfig()
// 打印配置项
for {
appName := viper.GetString("app.name")
fmt.Printf("App Name: %s\n", appName)
time.Sleep(2 * time.Second) // 每2秒检查一次
}
}
viper.WatchConfig() 会监听配置文件的变化，并在文件内容发生改变时自动重新加载。
Viper 是一个强大的配置管理工具，它提供了多种方式来管理配置文件，包括：
Viper 使得 Go 项目的配置管理更加简单且灵活，适合用于生产环境中需要灵活配置管理的应用程序。