快捷導(dǎo)航

Go?viper讀取配置文件的示例詳解

更新時(shí)間：2023年08月29日 10:01:04 作者：copyLeft

這篇文章主要為大家詳細(xì)介紹了Go語(yǔ)言如何利用viper實(shí)現(xiàn)讀取配置文件,文中的示例代碼講解詳細(xì),具有一定的借鑒價(jià)值,需要的可以參考一下

安裝

    go get github.com/spf13/viper

簡(jiǎn)單例子

/*
  目錄
    - main.go
    - config.yaml
*/
func main(){
	viper.SetConfigName("config") // 設(shè)置配置名稱
	viper.SetConfigFile("config.yaml") // 設(shè)置配置文件路徑
    // 讀取配置
	if err := viper.ReadInConfig(); err != nil{
		log.Fatal(err)
	}
    // 獲取配置數(shù)據(jù)
	summary := viper.GetString("Summary")
	fmt.Println(summary)
}

新建 Viper

viper 提供默認(rèn) Viper對(duì)象，可直接使用。也通過(guò) New 方法創(chuàng)建自定義Viper

// 直接使用默認(rèn)對(duì)象
viper.GetInt("count")
// 獲取全局Viper對(duì)象
globalViper := viper.GetViper()
// 新建Viper
conf := viper.New()
conf.SetConfigFile("config.yaml")

讀取配置文件

從配置讀取

 viper.SetConfigName("config")
 viper.SetConfigFile("config.yaml")
 if err := viper.ReadInConfig(); err != nil{
     log.Fatal(err)
 }
 version := viper.GetString("version")

從 io.Reader 讀取

func main(){
    file, err := os.Open("./config.yaml")
    if err != nil{
        log.Fatal(err)
    }
    defer file.Close()
    viper.SetConfig("yaml")
    version := viper.GetString("version")
    // 這里需要設(shè)置配置類型, 否則無(wú)法正確解釋配置內(nèi)容
    // 如果未配置類型，也可以通過(guò) Get 方法獲取數(shù)據(jù)自行解析。
}

從 flag 讀取

import (
	"fmt"
	"github.com/spf13/pflag"
	"github.com/spf13/viper"
)
func main() {
	pflag.String("ip", "127.0.0.1", "Server running address")
	pflag.Int64("port", 8080, "Server running port")
	pflag.Parse()
	viper.BindPFlags(pflag.CommandLine)
	fmt.Printf("ip :%s , port:%s", viper.GetString("ip"), viper.GetString("port"))
}

獲取值

值得獲取方法名為 Get[type] 形式

接口類型,既未作解析

Get

布爾

GetBool

時(shí)間

GetDuration
GetTime

數(shù)字

GetInt
GetInt32
GetInt64
GetFloat64
GetIntSlice
....

字符

GetString
GetStringSlice
....

設(shè)置默認(rèn)值

setDefault

 viper.setConfigFile("config.yaml")
 viper.SetDefault("port", 8000)
 println(viper.GetInt("port") // 8000
 // 這里未讀取配置文件，將返回默認(rèn)值

將配置映射到結(jié)構(gòu)體

Unmarshal 將配置屬性映射到 struct 中, 匹配模式類似 JSON 解析，只匹配大寫開(kāi)頭的屬性

type User struct{
    ID string
    Name string
    nickName string // 小寫屬性將不做匹配
}
func main(){
    /* config.yaml
        id: id11
        name: Rogan
        nickName: "wolf"  
    */
    ...
    user := &User{}
    viper.Unmarshal(user)
    println(&user) // { id11, Rogan } 
}

UnmarshalExact 用法與 Unmarshal 相同，不同的是struct 必須與配置屬性完全匹配，否則加報(bào)錯(cuò)

 /*
  config.yaml
    id: 111
    name: Rogan
 */
 type Conf_1 struct{
     Id string
     Name string
 }
 // 將報(bào)錯(cuò)
 type Conf_2 struct{
     Id string
 }

UnmarshalKey 匹配某一字段

 /*
  config.yaml
    server: 
        port: 8000
        host: localhost
 */
type ServerConf struct{
    Port int
    host string
}
func main(){
    ...
    serverConf := &ServerConf{}
    viper.UnmarshalKey("server", serverConfig)
}

混合配置 Merge[type]

MergeConfig

func main(){
    viper.SetConfigName("config")
	viper.SetConfigFile("config.yaml")
	viper.ReadInConfig()
	log.Println(viper.GetString("Summary"), viper.InConfig("Summary"))
	buff := []byte(`Name: jeck`)
	viper.MergeConfig(bytes.NewReader(buff))
	log.Println(viper.GetString("Name"))
}

監(jiān)聽(tīng)文件

WatchConfig

func main(){
	viper.SetConfigName("config")
	viper.SetConfigFile("config.yaml")
	viper.ReadInConfig()
	viper.WatchConfig()
	for {
		log.Println(viper.GetString("version"))
		time.Sleep(time.Second * 5)
	}
}
// 初始 versioni: 1
// --> 1
// 修改config.yaml version: 2
// --> 2

OnConfigChange 響應(yīng)配置變化

以上就是Go viper讀取配置文件的示例詳解的詳細(xì)內(nèi)容，更多關(guān)于go viper的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: