go 使用viper包读取toml和yml配置文件

什么是viper包？

Viper是适用于Go应用程序的完整配置解决方案。它被设计用于在应用程序中工作，并且可以处理所有类型的配置需求和格式。有了viper包，不用再费尽心思去寻找不同格式的配置文件对应的操作库，轻松读取配置信息。目前viper包支持JSON、TOML、YAML、HCL、envfile和Java properties等格式。
viper非常的容易上手。

读取toml文件

test.toml

[database]
server = "192.168.1.1"
ports = [ 8001, 8001, 8002 ]

[servers]
[servers.alpha]
ip = "10.0.0.1"

main.go

package main

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

func main() {
	//获取当前项目目录
	work, _ := os.Getwd()
	//设置文件名和文件后缀
	viper.SetConfigName("test")
	viper.SetConfigType("toml")
	//配置文件所在的文件夹
	viper.AddConfigPath(work + "/_viper")
	err := viper.ReadInConfig()
	if err != nil {
		panic("err")
	}
	//获取全部文件内容
	fmt.Println("all settings: ", viper.AllSettings())
	fmt.Println("--------------")
		//根据内容类型，解析出不同类型
	fmt.Println(viper.GetString("database.server"))
	fmt.Println(viper.GetIntSlice("database.ports"))
	fmt.Println("--------------")
	fmt.Println(viper.GetString("servers.alpha.ip"))
}

输出

all settings:  map[database:map[ports:[8001 8001 8002] server:192.168.1.1] serve
rs:map[alpha:map[ip:10.0.0.1]]]
--------------
192.168.1.1
[8001 8001 8002]
--------------
10.0.0.1

读取yml文件

test2.yml

database:
  server: 8090
  ports:
    - 8001
    - 8001
    - 8002

servers:
  alpha:
    ip: 10.0.0.1

main.go

func main() {
	//获取当前项目目录
	work, _ := os.Getwd()
	//设置文件名和文件后缀
	viper.SetConfigName("test2")
	viper.SetConfigType("yml")
	//配置文件所在的文件夹
	viper.AddConfigPath(work + "/_viper")
	err := viper.ReadInConfig()
	if err != nil {
		panic(err)
	}
	//获取全部文件内容
	fmt.Println("all settings: ", viper.AllSettings())
	fmt.Println("--------------")
	//根据内容类型，解析出不同类型
	fmt.Println(viper.GetString("database.server"))
	fmt.Println(viper.GetIntSlice("database.ports"))
	fmt.Println("--------------")
	fmt.Println(viper.GetString("servers.alpha.ip"))
}