使用 viper读取配置文件

安装

go get -u github.com/spf13/viper@level

使用

//提前创建配置内容结构体
type TestSetting struct {
    AppName string
    AppPort int
}

//在configs文件夹下创建config.yaml，写入内容
Test:
  AppName: test
  AppPort: 80

//导入
import "github.com/spf13/viper"

func readSetting() {
    vp := viper.New()

    //设置配置文件目录
    vp.AddConfigPath("configs/")

    //设置配置文件文件名
    vp.SetConfigName("config")

    //设置配置文件文件类型
    vp.SetConfigType("yaml")

    //准备配置文件
    err := vp.ReadInConfig()
    if err != nil {
        return err
    }

    //提前定义好
    var testSetting *TestSetting

    //读取指定配置文件并赋值
    err = vp.UnmarshalKey("Test", &testSetting)
    if err != nil {
        return err
    }

    fmt.Println(testSetting)
}