怎么在Golang利用viper包對yaml的配置信息進行讀取？針對這個問題，這篇文章詳細介紹了相對應(yīng)的分析和解答，希望可以幫助更多想解決這個問題的小伙伴找到更簡單易行的方法。

成都創(chuàng)新互聯(lián)公司專注于民和網(wǎng)站建設(shè)服務(wù)及定制，我們擁有豐富的企業(yè)做網(wǎng)站經(jīng)驗。 熱誠為您提供民和營銷型網(wǎng)站建設(shè)，民和網(wǎng)站制作、民和網(wǎng)頁設(shè)計、民和網(wǎng)站官網(wǎng)定制、重慶小程序開發(fā)服務(wù)，打造民和網(wǎng)絡(luò)公司原創(chuàng)品牌,更為您提供民和網(wǎng)站排名全網(wǎng)營銷落地服務(wù)。

1、安裝

go get github.com/spf13/viper

2、編寫一個yaml的配置文件，config.yaml

database:
 host: 127.0.0.1
 user: root
 dbname: test
 pwd: 123456

3、編寫學(xué)習(xí)腳本main.go，讀取config.yaml配置信息

package main 
import (
 "fmt"
 "os" 
 "github.com/spf13/viper"
)
 
func main() {
 //獲取項目的執(zhí)行路徑
 path, err := os.Getwd()
 if err != nil {
 panic(err)
 }
 
 config := viper.New() 
 config.AddConfigPath(path)  //設(shè)置讀取的文件路徑
 config.SetConfigName("config") //設(shè)置讀取的文件名
 config.SetConfigType("yaml") //設(shè)置文件的類型
 //嘗試進行配置讀取
 if err := config.ReadInConfig(); err != nil {
 panic(err)
 }
 
 //打印文件讀取出來的內(nèi)容:
 fmt.Println(config.Get("database.host"))
 fmt.Println(config.Get("database.user"))
 fmt.Println(config.Get("database.dbname"))
 fmt.Println(config.Get("database.pwd")) 
}

4、執(zhí)行g(shù)o run main.go

輸出:

127.0.0.1
root
test
123456

ok!

補充：go基于viper實現(xiàn)配置文件熱更新及其源碼分析

go第三方庫 github.com/spf13/viper 實現(xiàn)了對配置文件的讀取并注入到結(jié)構(gòu)中，好用方便。

其中以

viperInstance := viper.New() // viper實例
viperInstance.WatchConfig()
viperInstance.OnConfigChange(func(e fsnotify.Event) {
 log.Print("Config file updated.")
 viperLoadConf(viperInstance) // 加載配置的方法
})

可實現(xiàn)配置的熱更新，不用重啟項目新配置即可生效（實現(xiàn)熱加載的方法也不止這一種，比如以文件的上次修改時間來判斷等）。

為什么這么寫？這樣寫為什么就能立即生效？基于這兩個問題一起來看看viper是怎樣實現(xiàn)熱更新的。

上面代碼的核心一共兩處：WatchConfig()方法、OnConfigChange()方法。WatchConfig()方法用來開啟事件監(jiān)聽，確定用戶操作文件后該文件是否可正常讀取，并將內(nèi)容注入到viper實例的config字段，先來看看WatchConfig()方法：

func (v *Viper) WatchConfig() {
 go func() {
   // 建立新的監(jiān)視處理程序，開啟一個協(xié)程開始等待事件
   // 從I/O完成端口讀取，將事件注入到Event對象中：Watcher.Events
 watcher, err := fsnotify.NewWatcher() 
 if err != nil {
 log.Fatal(err)
 }
 defer watcher.Close()
 
 // we have to watch the entire directory to pick up renames/atomic saves in a cross-platform way
 filename, err := v.getConfigFile() 
 if err != nil {
 log.Println("error:", err)
 return
 }
 
 configFile := filepath.Clean(filename)  //配置文件E:\etc\bizsvc\config.yml
 configDir, _ := filepath.Split(configFile) // E:\etc\bizsvc\
 
 done := make(chan bool)
 go func() {
 for {
 select {
    // 讀取的event對象有兩個屬性，Name為E:\etc\bizsvc\config.yml，Op為write（對文件的操作）
 case event := <-watcher.Events:
 // 清除內(nèi)部的..和他前面的元素，清除當(dāng)前路徑.，用來判斷操作的文件是否是configFile
  if filepath.Clean(event.Name) == configFile {
    // 如果對該文件進行了創(chuàng)建操作或?qū)懖僮?  if event.Op&fsnotify.Write == fsnotify.Write || event.Op&fsnotify.Create == fsnotify.Create {
  err := v.ReadInConfig()
  if err != nil {
  log.Println("error:", err)
  }
  v.onConfigChange(event)
  }
  }
 case err := <-watcher.Errors:
     // 有錯誤將打印
  log.Println("error:", err)
 }
 }
 }() 
 watcher.Add(configDir)
 <-done
 }()
}

其中，fsnotify是用來監(jiān)控目錄及文件的第三方庫； watcher, err := fsnotify.NewWatcher() 用來建立新的監(jiān)視處理程序，它會開啟一個協(xié)程開始等待讀取事件，完成 從I / O完成端口讀取任務(wù)，將事件注入到Event對象中，即Watcher.Events；

執(zhí)行v.ReadInConfig()后配置文件的內(nèi)容將重新讀取到viper實例中，如下圖：

執(zhí)行完v.ReadInConfig()后,config字段的內(nèi)容已經(jīng)是用戶修改的新內(nèi)容了；

其中這行v.onConfigChange(event)的onConfigChange是核心結(jié)構(gòu)體Viper的一個屬性，類型是func：

type Viper struct {
 // Delimiter that separates a list of keys
 // used to access a nested value in one go
 keyDelim string
 
 // A set of paths to look for the config file in
 configPaths []string
 
 // The filesystem to read config from.
 fs afero.Fs
 
 // A set of remote providers to search for the configuration
 remoteProviders []*defaultRemoteProvider
 
 // Name of file to look for inside the path
 configName string
 configFile string
 configType string
 envPrefix string
 
 automaticEnvApplied bool
 envKeyReplacer  *strings.Replacer
 
 config   map[string]interface{}
 override  map[string]interface{}
 defaults  map[string]interface{}
 kvstore  map[string]interface{}
 pflags   map[string]FlagValue
 env   map[string]string
 aliases  map[string]string
 typeByDefValue bool
 
 // Store read properties on the object so that we can write back in order with comments.
 // This will only be used if the configuration read is a properties file.
 properties *properties.Properties
 
 onConfigChange func(fsnotify.Event)
}

它用來傳入本次event來執(zhí)行你寫的函數(shù)。為什么修改會立即生效？相信第二個疑問已經(jīng)得到解決了。

接下來看看OnConfigChange(func(e fsnotify.Event) {...... })的運行情況：

func (v *Viper) OnConfigChange(run func(in fsnotify.Event)) {
 v.onConfigChange = run
}

方法參數(shù)為一個函數(shù)，類型為func(in fsnotify.Event)) {}，這就意味著開發(fā)者需要把你自己的執(zhí)行邏輯放到這個func里面，在監(jiān)聽到event時就會執(zhí)行你寫的函數(shù)，所以就可以這樣寫：

 viperInstance.OnConfigChange(func(e fsnotify.Event) {
 log.Print("Config file updated.")
 viperLoadConf(viperInstance) // viperLoadConf函數(shù)就是將新配置注入到自定義結(jié)構(gòu)體對象的邏輯
 })

而OnConfigChange方法的參數(shù)會賦值給形參run并傳到viper實例的onConfigChange屬性，以WatchConfig()方法中的v.onConfigChange(event)來執(zhí)行這個函數(shù)。

關(guān)于怎么在Golang利用viper包對yaml的配置信息進行讀取問題的解答就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，如果你還有很多疑惑沒有解開，可以關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道了解更多相關(guān)知識。

標(biāo)題名稱：怎么在Golang利用viper包對yaml的配置信息進行讀取-創(chuàng)新互聯(lián)
網(wǎng)頁網(wǎng)址：http://jinyejixie.com/article44/dchiee.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供外貿(mào)建站、App開發(fā)、網(wǎng)站改版、網(wǎng)站制作、網(wǎng)站設(shè)計公司、用戶體驗

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源： 創(chuàng)新互聯(lián)