详解如何使用Go的Viper来解析配置信息

go get github.com/spf13/viper

import "github.com/spf13/viper"

viper.SetConfigFile("./config.yaml")
viper.ReadInConfig()
fmt.Println(viper.Get("我是不区分大小写的key"))

// 首先声明一个配置映射对象
type DB struct {
	host string
	port string
}
 
func main() {
 
	viper.SetConfigName("config")
	viper.SetConfigType("yaml")
	viper.AddConfigPath(".")
	viper.ReadInConfig() // 读取配置
	var mysql DB
 
	viper.Unmarshal(&mysql) 将读取到的配置序列化为对象
 
	fmt.Println(mysql.host)
	fmt.Println(mysql.port)
}

// 设置监听回调
viper.OnConfigChange(func(e fsnotify.Event) {
	fmt.Println("检测到配置文件已更改:", e.Name)
})
 
// 开启监听
viper.WatchConfig()

viper.Set("name", "张三")
 
//为name设置一个username的别名
viper.RegisterAlias("username", "name")
 
//通过username可以读取到name的值
fmt.Println(viper.Get("username"))
 
//修改name的配置值，username的值也发生改变
viper.Set("name", "李四")
 
// 输出结果为李四
fmt.Println(viper.Get("username"))
 

// 读取不到
fmt.Println(viper.Get("path"))
 
//开始读取环境变量
viper.AutomaticEnv()
 
//会从环境变量读取到该值
fmt.Println(viper.Get("path"))

//将p绑定到环境变量PATH,注意这里第二个参数是环境变量
viper.BindEnv("p", "PATH")
 
// 通过p读取PATH的值
fmt.Println(viper.Get("p"))

go get github.com/spf13/viper

app: # 应用基本配置
  env: local # 环境名称
  port: 8080 # 服务监听端口号
  app_name: online-practice-system # 应用名称
  app_url: http://localhost # 应用域名

// Configuration为管理全部配置的父结点
type Configuration struct {
	App      App      `mapstructure:"app" yaml:"app"`
}
// 如果是多层配置，就需要定义多层结构体
type App struct {
    // 在tag标签中加入yaml:"env"`,声明配置对应关系
    Env     string `mapstructure:"env" yaml:"env"`
    Port    string `mapstructure:"port" yaml:"port"`
    AppName string `mapstructure:"app_name" yaml:"app_name"`
    AppUrl  string `mapstructure:"app_url" yaml:"app_url"`
}

package bootstrap
import (
	"fmt"
	"github.com/fsnotify/fsnotify"
	"github.com/spf13/viper"
	"online-practice-system/global"
	"os"
)
// viper读取配置文件初始化
func InitializeConfig() *viper.Viper {
	// 设置配置文件路径
	config := "config.yaml"
	// 生产环境可以通过设置环境变量来改变配置文件路径
	if configEnv := os.Getenv("VIPER_CONFIG"); configEnv != "" {
		config = configEnv
	}
	// 初始化 viper
	v := viper.New()
	v.SetConfigFile(config)
	v.SetConfigType("yaml")
	if err := v.ReadInConfig(); err != nil {
		panic(fmt.Errorf("read config failed: %s \n", err))
	}
	// 监听配置文件
	v.WatchConfig()
	v.OnConfigChange(func(in fsnotify.Event) {
		fmt.Println("config fileDriver changed:", in.Name)
		// 重载配置,这里可以进行重启服务器,或者其他操作
		if err := v.Unmarshal(&global.App.Config); err != nil {
			fmt.Println(err)
		}
	})
	// 将配置赋值给全局变量
	if err := v.Unmarshal(&global.App.Config); err != nil {
		fmt.Println(err)
	}
	return v
}

func main() {
    //  初始化yaml配置文件
    InitializeConfig()
    ///
}