Go-viper读取配置文件

Posted 2022-12-02 聪聪不匆匆

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Go-viper读取配置文件相关的知识，希望对你有一定的参考价值。

hello，大家好，我是聪聪。

1. 介绍

Viper是适用于Go应用程序的完整配置解决方案。它被设计用于在应用程序中工作，并且可以处理所有类型的配置需求和格式。它支持以下特性：

设置默认值
从JSON、TOML、YAML、HCL、envfile和Java properties格式的配置文件读取配置信息
实时监控和重新读取配置文件（可选）
从环境变量中读取
从远程配置系统（etcd或Consul）读取并监控配置变化
从命令行参数读取配置
从buffer读取配置
显式配置值

Viper能够帮助我们执行以下操作：

查找、加载和反序列化JSON、TOML、YAML、HCL、INI、envfile和Java properties格式的配置文件。
提供一种机制为你的不同配置选项设置默认值，进行灵活配置和发布配置。
提供一种机制来通过命令行参数覆盖指定选项的值。
提供别名系统，以便在不破坏现有代码的情况下轻松重命名参数。
当用户提供了与默认值相同的命令行或配置文件时，可以很容易地分辨出它们之间的区别。

Viper能够按照下列优先级进行获取配置：

显示调用Set设置值
命令行参数（flag）
环境变量
配置文件
key/value存储
默认值

2. 安装

go get github.com/spf13/viper

3. Viher使用介绍

3.1 设置默认值

Viper支持设置默认值，当我们无法从配置文件、环境变量、远程配置、命令行获取配置时，默认配置到是很有用的。

func main() 
  //设置各类key对应默认值
  //key 为string类型
  //value可接受interface 类型参数
	viper.SetDefault("rootPath", "rootPath default value")
	viper.SetDefault("defaultPort", 8000)
	viper.SetDefault("defaultConfig", map[string]string
		"HttpMethod": "HttpMethod default value",
		"HttpUrl":    "HttpUrl default value",
	)
  //从viper中读取配置

	fmt.Println("rootPath value:", viper.Get("rootPath"))
	fmt.Println("defaultPort value:", viper.Get("defaultPort"))
	fmt.Println("defaultConfig:", viper.Get("defaultConfig"))
	fmt.Println("defaultConfig:", viper.GetStringMapString("defaultConfig"))

从Viper中读取配置key对应配置信息。打印信息如下：

rootPath value: rootPath default value
defaultPort value: 8000
defaultConfig: map[httpMethod:GET httpUrl:ccoder.cc]
defaultConfig: map[httpMethod:GET httpUrl:ccoder.cc]

3.2 读取配置文件

Viper读取配置文件时，需要指定告诉配置文件路径（或者读取配置方式）。同时Viper支持从JSON、TOML、YAML、HCL、envfile和Java properties格式的配置文件读取配置信息。同时支持搜索多个路径。

下面是一个从配置文件config.json 读取配置的案例。只用指定配置文件config预期出现的路径，无需指定特定的路径。

//在当前工作目录 . 中读取一个config的配置文件 未指定文件后缀.
func printFile() 
  //配置文件名称(无后缀扩展名)，viper会自行解析文件后缀名.
  //若配置文件无后缀时则需要通过SetConfigType配置后水名
  viper.SetConfigName("config")
  //可以重复添加的搜索路径
	viper.AddConfigPath(".")

  //加载配置文件 捕获一下error:文件不存在情况。
	err := viper.ReadInConfig()
	if err != nil 
		if _, ok := err.(viper.ConfigFileNotFoundError); ok 
			log.Println("config not exists")
		 else 
			log.Println("read config error")
		
		log.Fatal(err)
	

  //读取配置文件内容
	fmt.Println("读取port: ", viper.GetInt("port"))
	fmt.Println("读取mysql.url: ", viper.GetString(`mysql.url`))
	fmt.Println("读取mysql.username: ", viper.GetString(`mysql.username`))
	fmt.Println("读取mysql.password: ", viper.GetString(`mysql.password`))
	fmt.Println("读取redis: ", viper.GetStringSlice("redis"))
	fmt.Println("读取mongo: ", viper.GetStringMap("mongo"))
	fmt.Println("读取elasticsearch: ", viper.GetStringMap("elasticsearch"))

代码解析：

viper.SetConfigName ("config") 设置配置文件名为 config, 不需要配置文件扩展名，配置文件的类型 viper 会自动根据扩展名自动匹配。若配置文件无后缀扩展名时，可通过SetConfigType配置解析文件类型。
viper.AddConfigPath (".") 设置配置文件搜索的目录，. 表示和当前编译好的二进制文件在同一个目录。
- 可以添加多个配置文件目录，如在第一个目录中找到就不不继续到其他目录中查找.
viper.ReadInConfig () 加载配置文件内容。
viper.Getxxxx 用来读取配置内容。

3.3 写入配置文件

运行时可以从配置中读取系列配置为应用所用，当然运行时我们会做一系列初始化操作将配置写入到配置文件中；那么此时这种场景，下列API便是非常适用的：

viper.WriteConfig() : 将viper当前配置写入到文件中，会读取当前viper中的配置文件名称
SafeWriteConfig : 将当前的viper配置写入预定义的路径。如果没有预定义的路径，则报错。如果存在，将不会覆盖当前的配置文件。
- 必须存在configPath .否则：missing configuration for 'configPath'
WriteConfigAs : 将当前的viper配置写入给定的文件路径。若文件已存在，将覆盖给定的文件。
SafeWriteConfigAs :将当前的viper配置写入给定的文件路径。若文件已存在，不会覆盖给定的文件。

viper.WriteConfig()
viper.SafeWriteConfig()
viper.WriteConfigAs("new_config.yaml")
viper.SafeWriteConfigAs("new_config.yaml")

3.4 监控配置文件变更

Viper支持监听和重新读取配置文件。

import "github.com/fsnotify/fsnotify"

使用方式：

viper.WatchConfig()
viper.OnConfigChange(func(e fsnotify.Event) 
// 通过fsnotify.Event 事件监听配置文件更新回调
fmt.Println("Config file changed:", e.Name)
)

3.5 环境变量读取

从环境变量中读取配置主要用到以下函数：

vper.AutomaticEnv() AutomaticEnv使Viper检查环境变量是否与任何现有键匹配。在下列函数中匹配前缀prefix 的变量
vper.SetEnvPrefix() SetEnvPrefix定义了一个环境变量将使用的前缀。

func printConfigEnvs() 
	prefix := "prefix"
	envs := map[string]string
		"Level":          "INFO",
		"Mode":           "DEV",
		"MYSQL_USERNAME": "root",
		"MYSQL_PASSWORD": "xxx",
	
	for k, v := range envs 
		err := os.Setenv(fmt.Sprintf("%s_%s", prefix, k), v)
		if err != nil 
			return
		
	
	v := viper.New()
	v.SetEnvPrefix(prefix)
	v.AutomaticEnv()

	for k := range envs 
		fmt.Printf("%s = %s \\n", k, v.GetString(k))

3.6 从远程读取配置

在Viper中启用远程支持，需要在代码中匿名导入viper/remote这个包。

import _ "github.com/spf13/viper/remote"

Viper将读取从Key/Value存储（例如etcd或Consul）中的路径检索到的配置字符串（如JSON、TOML、YAML、HCL、envfile和Java properties格式）。

这些值的优先级高于默认值，但是会被从磁盘、flag或环境变量检索到的配置值覆盖。

也就是说Viper加载配置值的优先级为：磁盘上的配置文件>命令行标志位>环境变量>远程Key/Value存储>默认值。

先本地启动consul 添加一个配置文件
```
consul agent -dev
```
在consul中添加一个名为config的yaml配置文件。

读取配置文件

func printConfigRemote() 
	v := viper.New()
	err := v.AddRemoteProvider("consul", "localhost:8500", "config")
	if err != nil 
		return
	
	v.SetConfigType("YAML")
	if err = v.ReadRemoteConfig(); err != nil 
		if _, ok := err.(viper.ConfigFileNotFoundError); ok 
			log.Println("config not exists")
		 else 
			log.Println("read config error")
		
		log.Fatal(err)
	

	fmt.Println("port: ", v.GetInt("port"))
	fmt.Println("mysql.url: ", v.GetString(`mysql.url`))
	fmt.Println("mysql.username: ", v.GetString(`mysql.username`))
	fmt.Println("mysql.password: ", v.GetString(`mysql.password`))

打印读取配置信息

port:  3306
mysql.url:  url
mysql.username:  username
mysql.password:  password

3.7 从io.Reader读取配置

Viper预先定义了许多配置源，如文件、环境变量、标志和远程K/V存储，但你不受其约束。你还可以实现自己所需的配置源并将其提供给viper。

下面定义了一段JSON配置，将其从Reader中读取进行viper设置。

func printConfigIoReader() 
	v := viper.New()
	v.SetConfigType("json")

	var jsonConfig = []byte(`
		"port" : 3306,
		"username" : "root",
		"password" : "123456"
	`)
	//创建 io.Reader
	//ReadConfig读取配置文件 若配置文件中key不存在则设置nil值
	err := v.ReadConfig(bytes.NewReader(jsonConfig))
	if err != nil 
		fmt.Println(err)
	
	fmt.Println("port: ", v.GetInt("port"))  //3306
	fmt.Println("username: ", v.GetString("username"))  //root
	fmt.Println("password: ", v.GetString("password"))  //123456

3.8 读取配置到结构体

除了上面的用法外，我们还可以在项目中定义与配置文件对应的结构体，viper加载完配置信息后使用结构体变量保存配置信息。

定义一个结构体和全局变量

type Config struct 
	Port    string
	Version string


var Conf = new(Config)

加载配置到全局变量中

func printConfigStruct() 
	v := viper.New()
	v.SetConfigFile("./config/config.yaml")
	if err := v.ReadInConfig(); err != nil 
		fmt.Printf("fail to load the config,%s\\n", err)
		return
	
	if err := v.Unmarshal(Conf); err != nil 
		fmt.Printf("Serialization configuration failed.%s\\n", err)
		return
	

	fmt.Println("port:", Conf.Port)  // 3306
	fmt.Println("version:", Conf.Version) //1.1.1

4. 使用示例

4.1 是用gin获取配置rest接口

package main

import (
	"fmt"
	"github.com/fsnotify/fsnotify"
	"github.com/gin-gonic/gin"
	"github.com/spf13/viper"
	"log"
	"net/http"
)

type Config struct 
	Port    int
	Version string
	Context map[string]interface


var Conf = new(Config)

func main() 
	v := viper.New()
	v.SetConfigFile("./gin/config.yaml")
	if err := v.ReadInConfig(); err != nil 
		if _, ok := err.(viper.ConfigFileNotFoundError); ok 
			fmt.Println("config not exists")
		 else 
			fmt.Println("config read failed")
		
		log.Fatal(err)
	

	if err := v.Unmarshal(Conf); err != nil 
		fmt.Printf("Serialization configuration failed.%s\\n", err)
		return
	

	//监听文件变化
	v.WatchConfig()
	v.OnConfigChange(func(in fsnotify.Event) 
		fmt.Println("The config file has been modified.")
		if err := v.Unmarshal(Conf); err != nil 
			panic(fmt.Errorf("Failed to reload the configuration file.%s\\n", err))
		
	)

	//添加gin
	g := gin.Default()
	g.GET("/config", func(c *gin.Context) 
		c.JSON(http.StatusOK, Conf)
	)

	if err := g.Run(
		fmt.Sprintf(":%d", Conf.Port)); err != nil 
		panic(err)

接口响应信息

 //请求接口
 curl http://localhost:8080/config

 //响应信息
 
    "Port": 8080,
    "Version": "1.18.3",
    "Context": 
        "language": "zh-CN",
        "url": "ccoder.cc"

5. 参考连接

全文完。

了解更多内容，可以关注我的微信公众号，更多首发文章。

以上是关于Go-viper读取配置文件的主要内容，如果未能解决你的问题，请参考以下文章

写一个 Properties 格式的配置文件，配置类的完整名称。 * 写一个程序，读取这个 Properties 配置文件，获得类的完整名称并加载这个类,用反射的方式运行

ResourceBundle类：读取配置文件

shell 读取一个文件，要按一定格式截取并读取其值

springBoot框架不同环境读取不同的配置文件

配置文件——App.config文件读取和修改

多环境下读取不同的配置文件