go实践四 使用viper加载服务器配置config.go

Viper 简介

Viper 是国外大神 spf13 编写的开源配置解决方案，具有如下特性:

• 设置默认值
• 可以读取如下格式的配置文件：JSON、TOML、YAML、HCL
• 监控配置文件改动，并热加载配置文件
• 从环境变量读取配置
• 从远程配置中心读取配置（etcd/consul），并监控变动
• 从命令行 flag 读取配置
• 从缓存中读取配置
• 支持直接设置配置项的值

安装viper:

[root@localhost web]# go get github.com/spf13/viper

在go的根目录下 新建 web/config目录 ,目录参考： /usr/local/go/web/config

在 config目录中，新建 config.yaml （配置）文件，内容如下：

common:
  database:
    name: test
    host: 127.0.0.1

在 config目录中，新建 config.go （获取配置）文件，内容如下：

package config

import (
	"fmt"
	"github.com/spf13/viper"
)

func Init() (interface{},error) {  //模块中供其他包调用的方法，首字母必须大写
	//viper设置 配置
	viper.Set("name","abc")
	fmt.Printf("name的值是%v\n",viper.GetString("name") )

	//读取配置文件配置
	viper.AddConfigPath("config")
	viper.SetConfigName("config")
	error := viper.ReadInConfig()
	/*
	代码解析：
		viper.AddConfigPath("conf")用来指定yaml配置文件的路径
		viper.SetConfigName("config")用来指定配置文件的名称
		viper.ReadInConfig()是解析配置文件的函数，如果配置文件的路径错误获名称错误则解析失败，会报错误
		viper.GetString("database.name")是用来从配置文件中根据层级关系来获取数据
		最后，通过fmt.Println()对数据结果进行输出
	*/
	if(error != nil){
		panic(error)
	}
	c := viper.AllSettings() //获取所有配置
	return c,nil
}

//获取数据库配置信息
func GetDatabaseInfo() map[string]interface{} {  //模块中供其他包调用的方法，首字母必须大写
	return viper.GetStringMap("common.database")
}

//获取环境变量
func GetEnvInfo(env string) string {
	viper.AutomaticEnv()
	return viper.GetString(env)
}

在 web目录中，新建 testviper.go （加载配置）文件，内容如下：

package main

import (
	"fmt"
	"web/config"
)

func main() {
	vipConfig,error := config.Init()                                  //vipConfig是配置
	fmt.Printf("config.init error是%v\n", error)
	//fmt.Printf("config.init vipConfig是%v\n",vipConfig,)
	database := config.GetDatabaseInfo()
	fmt.Printf("直接获取common[database]配置是%v\n", database)
	fmt.Printf("直接获取common[database][host]配置是%v\n", database["host"])

	//因为我们不知道 vipConfig 的下级是什么类型的数据，所以这里使用了interface{}
	//因此所有的类型、任意动态的内容都可以解析成 interface{}。
	for key,val := range vipConfig.(map[string]interface{}){        //循环接口类型，获取配置信息
		fmt.Printf("vipConfig 的key是%v val是%v\n",key,val )

		switch val.(type) {                                          //判断val的类型
		case map[string]interface{}:                                //如果是 interface接口类型
			for ke,va := range val.(map[string]interface{}){        //循环接口类型，获取配置信息
				fmt.Printf("vipConfig 的ke是%v va是%v\n",ke,va )

				switch va.(type) {                                    //判断va的类型
				case map[string]interface{}:                         //如果是 interface接口类型
					for k,v := range va.(map[string]interface{}){   //循环接口类型，获取配置信息
						fmt.Printf("vipConfig 的k是%v v是%v\n",k,v )
					}
				}
			}
		}
	}

	//viper可以获取服务器的环境变量
	GO111MODULE := config.GetEnvInfo("GO111MODULE")
	fmt.Printf("GO111MODULE的值是%v\n",GO111MODULE)

}

 

使用 go run testviper.go 运行该文件即可

结果如下：

[root@localhost web]# go run testviper.go
name的值是abc
config.init error是
直接获取common[database]配置是map[host:[127.0.0.1] name:[test]]
直接获取common[database][host]配置是[127.0.0.1]
vipConfig 的key是common val是map[database:map[host:127.0.0.1 name:test]]
vipConfig 的ke是database va是map[host:127.0.0.1 name:test]
vipConfig 的k是name v是test
vipConfig 的k是host v是127.0.0.1
vipConfig 的key是name val是abc
GOROOT的值是on

viper用法参考：https://www.cnblogs.com/Survivalist/articles/10429498.html