目录准备

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

env:
  lang: golang
  system: win10
 
author: lauwen
 
apiTokenKey: Z2luLWNsaTIwMjEwNQ==
apiTokenExpireTime: 1440
apiTokenRebuildTime: 100

image.png

读取实现

安装

读取配置文件这里使用到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")

image.png