Golang读取配置文件实现（yaml）

目录准备

在项目根目录新建config配置文件目录，同时创建一个yaml格式的配置文件config.yaml，写入一些测试数据

env:
  lang: golang
  system: win10

author: lauwen

apiTokenKey: Z2luLWNsaTIwMjEwNQ==
apiTokenExpireTime: 1440
apiTokenRebuildTime: 100

读取实现

安装

读取配置文件这里使用到spf13/viper这个库，先安装

go get -u github.com/spf13/viper

读取键值格式 

[配置文件名].[yaml文件具体配置]，例如confing.yaml文件下env.lang配置，传递时写为config.env.lang

 

获取文件名filename和要获取的键名key

index := strings.Index(fileKey, ".")
fileName := fileKey[0:index]
key := fileKey[index+1:]

viper读取

// 初始化viper
viper := viper2.New()

// 设置配置文件名，即上一步提取的
viper.SetConfigName(fileName)

// 设置配置文件格式
viper.SetConfigType("yaml")

// 设置配置文件所在目录，这里固定为config，可以根据需要更改
viper.AddConfigPath("config")

具体实现代码

package config

import (
	viper2 "github.com/spf13/viper"
	"strings"
)

func Get(fileKey string) interface{}  {
	index := strings.Index(fileKey, ".")
	fileName := fileKey[0:index]
	key := fileKey[index+1:]

	viper := viper2.New()
	viper.SetConfigName(fileName)
	viper.SetConfigType("yaml")
	viper.AddConfigPath("config")

	if err := viper.ReadInConfig(); err != nil {
		panic(err.Error())
	}
	return viper.Get(key)
}

调用

config.Get("config.env.lang")