使用go读取配置文件

配置工具的选择

通常,在一个或多个项目中会有多种格式的配置文件,比如PHP的php.ini文件、Nginx的server.conf文件,那么使用Golang怎么去读取这些不同格式的配置文件呢?

比如常见的有 JSON文件、INI文件、YAML文件和TOML文件等等。其中这些文件,对应的Golang处理库如下:

  • encoding/json – 标准库中的包,可以处理JSON配置文件,缺点是不能加注释
  • gcfg&goconfig – 处理INI配置文件
  • toml – 处理TOML配置文件
  • viper – 处理JSON, TOML, YAML, HCL以及Java properties配置文件

通常情况下,推荐使用viper库来读取配置文件,虽然它不支持ini格式的配置文件,但我们可以使用goconfig 或gcfg.v1库读取ini 格式配置文件。

viper 支持以下功能:

  • 支持Yaml、Json、 TOML、HCL 等格式的配置文件
  • 可以从文件、 io.Reader 、环境变量、cli命令行读取配置
  • 支持自动转换的类型解析
  • 可以远程从Key/Value中读取配置,需要导入 viper/remote 包
  • 监听配置文件。以往我们修改配置文件后需要重启服务生效,而Viper使用watch函数可以让配置自动生效。

安装viper

1
2
go get github.com/spf13/viper
go get github.com/fsnotify/fsnotify

使用viper读取JSON配置文件

假设现在有一份 json 格式的配置文件 config.json

1
2
3
4
5
6
7
8
{
"date": "2019-04-30",
"mysql": {
"url": "127.0.0.1:3306",
"username": "root",
"password": "123456"
}
}

读取json配置文件

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
package main

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

func main() {
viper.SetConfigName("config") //设置配置文件的名字
viper.AddConfigPath(".") //添加配置文件所在的路径
viper.SetConfigType("json") //设置配置文件类型,可选
err := viper.ReadInConfig()
if err != nil {
fmt.Printf("config file error: %s\n", err)
os.Exit(1)
}
urlValue := viper.Get("mysql.url")
fmt.Println("mysql url:", urlValue)
fmt.Printf("mysql url: %s\n mysql username: %s\n mysql password: %s", viper.Get("mysql.url"), viper.Get("mysql.username"), viper.Get("mysql.password"))
}

运行程序,查看效果

1
2
3
4
5
$ go run viper_json.go
mysql url: 127.0.0.1:3306
mysql url: 127.0.0.1:3306
mysql username: root
mysql password: 123456

使用viper读取yaml配置文件

假设现在有一份yaml格式的配置文件 config_yaml.yaml

1
2
3
4
5
port: 10666
mysql:
url: "127.0.0.1:3306"
username: root
password: 123456

读取yaml配置文件

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
package main

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

func main() {
viper.SetConfigName("config_yaml") //把json文件换成yaml文件,只需要配置文件名 (不带后缀)即可
viper.AddConfigPath(".") //添加配置文件所在的路径
//viper.SetConfigType("json") //设置配置文件类型
err := viper.ReadInConfig()
if err != nil {
fmt.Printf("config file error: %s\n", err)
os.Exit(1)
}

viper.WatchConfig() //监听配置变化
viper.OnConfigChange(func(e fsnotify.Event) {
fmt.Println("配置发生变更:", e.Name)
})

urlValue := viper.Get("mysql.url")
fmt.Println("mysql url:", urlValue)
fmt.Printf("mysql url: %s\nmysql username: %s\nmysql password: %s", viper.Get("mysql.url"), viper.Get("mysql.username"), viper.GetString("mysql.password"))
}

viper其他重要功能

获取子级配置
当配置层级关系较多的时候,有时候我们需要直接获取某个子级的所有配置,可以这样操作:

1
2
3
4
5
6
7
app:
cache1:
max-items: 100
item-size: 64
cache2:
max-items: 200
item-size: 80

如果要读取cache1下的max-items,只需要执行viper.Get(“app.cache1.max-items”)就可以了。

解析配置
可以将配置绑定到某个结构体、map上,有两个方法可以做到这一点:

1
2
3
4
5
6
7
8
9
10
11
12
13
Unmarshal(rawVal interface{}) : error
UnmarshalKey(key string, rawVal interface{}) : error

var config Config
var mysql MySQL
err := Unmarshal(&config) // 将配置解析到 config 变量
if err != nil {
t.Fatalf("unable to decode into struct, %v", err)
}
err := UnmarshalKey("mysql", &mysql) // 将配置解析到 mysql 变量
if err != nil {
t.Fatalf("unable to decode into struct, %v", err)
}

获取值
在Viper中,有一些根据值的类型获取值的方法,存在以下方法:

1
2
3
4
5
6
7
8
9
10
11
Get(key string) : interface{}
GetBool(key string) : bool
GetFloat64(key string) : float64
GetInt(key string) : int
GetString(key string) : string
GetStringMap(key string) : map[string]interface{}
GetStringMapString(key string) : map[string]string
GetStringSlice(key string) : []string
GetTime(key string) : time.Time
GetDuration(key string) : time.Duration
IsSet(key string) : bool

如果 Get 函数未找到值,则返回对应类型的一个零值。可以通过 IsSet() 方法来检测一个健是否存在。

1
2
3
4
viper.GetString("logfile") 
if viper.GetBool("verbose") {
fmt.Println("verbose enabled")
}

修改对应的配置

1
2
viper.Set("Verbose", true)
viper.Set("LogFile", LogFile)

使用goconfig读取ini配置文件

安装goconfig

1
go get github.com/Unknwon/goconfig

假设database.conf配置文件,如下所示

1
2
3
4
5
6
[mysql]
username=root
password=123456
url=127.0.0.1:3306
[redis]
address=127.0.0.1:6379

使用goconfig读取ini格式配置文件

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
package main

import (
"fmt"
"github.com/Unknwon/goconfig"
"os"
)

var cfg *goconfig.ConfigFile

func init() {
config, err := goconfig.LoadConfigFile("database.conf") //加载配置文件
if err != nil {
fmt.Println("get config file error")
os.Exit(-1)
}
cfg = config
}

func GlobalConfig() {
glob, _ := cfg.GetSection("mysql") //读取全部mysql配置
fmt.Println(glob)
}

func main() {
password, _ := cfg.GetValue("mysql", "password") //读取单个值
fmt.Println(password)
username, _ := cfg.GetValue("mysql", "username") //读取单个值
fmt.Println(username)
err := cfg.Reload() //重载配置
if err != nil {
fmt.Printf("reload config file error: %s", err)
}
GlobalConfig()
}

加载完全局配置后,该配置长驻内存,需要动态加载的话,使用cfg.Reload()方法。

运行程序,效果如下。

1
2
3
4
$ go run goconfig.go
123456
root
map[password:123456 url:127.0.0.1:3306 username:root]

参考
https://blog.biezhi.me/2018/10/load-config-with-viper.html

0%