一个简单例子使用viper读取配置信息。
package main
import (
"fmt"
"strings"
"github.com/spf13/viper"
)
func main() {
//viper.SetConfigFile("config/config.yaml")
viper.AddConfigPath("config")
viper.SetConfigType("yaml")
viper.SetConfigName("config")
if err := viper.ReadInConfig(); err == nil {
fmt.Println("Using config file:", viper.ConfigFileUsed())
}
// For environment variables.
viper.AutomaticEnv()
viper.SetEnvPrefix("TEST")
replacer := strings.NewReplacer(".", "_")
viper.SetEnvKeyReplacer(replacer)
key := "A1.B"
fmt.Printf("v[%s]=[%s]\n", key, viper.GetString(key))
key = "A2.B"
fmt.Printf("v[%s]=[%s]\n", key, viper.GetString(key))
}
函数说明：
viper.SetConfigFile 直接设置配置文件
viper.AddConfigPath/viper.SetConfigType/viper.SetConfigName
通过设置多个路径一次来搜索配置文件，直到搜到一个位置。从函数名可以看出，viper只能支持一个配置文件，一种类型，可以包括多个搜索路径。viper.AutomaticEnv是否去读取环境变量
如果设置了，那么会优先读取环境变量，即使配置文件中已经定义了，环境变量会覆盖配置文件的值。viper.SetEnvPrefix/viper.SetEnvKeyReplacer设置环境变量的读取规则。
首先点号('.')是不能出现在环境变量的，所以必须用viper.SetEnvKeyReplacer把点号('.')映射成下划线('_')，然后viper.SetEnvPrefix可以设置一个环境变量的前缀。
举个例子来说：
- viper.GetString("AA") // 直接读取环境变量AA
- viper.GetString("AA.BB") // 试图读取环境变量AA.BB
但是AA.BB格式是不允许出现在环境变量的，所以可能使用AA_BB环境变量，然后使用viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_")) - viper.GetString("AA.BB")直接读取的环境变量AA_BB，但是有时我们希望不要读取其他进程的，不是为当前进程设置的环境变量，这样我们需要把当前进程的环境变量加一个统一的前缀，例如TEST，所以环境变量TEST_AA_BB的值就是viper.GetString("AA.BB")最终取到的值。
