快捷導(dǎo)航

golang使用viper解析配置文件的示例代碼

更新時(shí)間：2024年08月22日 10:42:53 作者：windcontinuestoblow

Viper是一個(gè)輕量級的、易于使用的配置工具庫,它允許你在Go應(yīng)用中方便地管理配置,Viper支持從多種來源讀取配置,如環(huán)境變量、命令行參數(shù)、文件、甚至是加密的數(shù)據(jù)存儲,本文給大家介紹了golang使用viper解析配置文件,需要的朋友可以參考下

viper讀取配置文件

讀取配置這部分內(nèi)容可以作為初始化內(nèi)容寫在init函數(shù)中，后面直接調(diào)用viper相關(guān)接口即可。
yaml類型的配置文件在path目錄下，通過viper接口可以將配置文件加載到viper中，方便后續(xù)使用。

	//設(shè)置配置文件
	cfgFile="/path/conf/config.yaml"
	viper.SetConfigFile(cfgFile)
	//加載配置文件內(nèi)容
	err = viper.ReadInConfig()
	if err != nil {
		slog.Error("ReadInConfig error")
	}

viper讀取基本數(shù)據(jù)類型內(nèi)容

在config.yaml文件中填寫如下配置：

	#yaml配置文件內(nèi)容
	logger:
	  level: DEBUG #log級別
	  max_size: 10 #log rotate文件大小配置，單位"MB"
	  compress: true #是否壓縮舊文件

可以在go代碼中定義logger結(jié)構(gòu)體，用于統(tǒng)一管理logger節(jié)點(diǎn)下的配置。

	//定義結(jié)構(gòu)體
	type LogConfig struct {
		MaxSize    int    `yaml:"max_size"`
		Compress   bool   `yaml:"compress"`
		Level      string `yaml:"level"`
		
	}

針對int，bool，string基本數(shù)據(jù)類型的配置，viper提供了對應(yīng)的接口進(jìn)行解析。

	//viper接口獲取int,bool,string等類型內(nèi)容
	conf := &LogConfig{
	        MaxSize:    viper.GetInt("logger.max_size"),
	        Compress:   viper.GetBool("logger.compress"),
	        Level:      viper.GetString("logger.level"),
	}

除了上述幾個(gè)數(shù)據(jù)類型還包括viper.GetChan接口，基本上可以覆蓋大多數(shù)解析情況。

viper讀取map類型配置

如果希望在配置文件中填入map形式的數(shù)據(jù)，這是viper沒有提供對應(yīng)數(shù)據(jù)類型的接口，需要自定義解析方式。

config.yaml配置中內(nèi)容：

	#yaml配置文件內(nèi)容
	res:
	  enable_flag: false
	  res_list: {"001":"aa","002":"bb"}

采用viper.AllSettings()方法獲取全部配置，yaml數(shù)據(jù)以map形式存儲，逐步解析可以獲得res_list map類型數(shù)據(jù)。

	//通過viper獲取全部配置并解析
	resList := make(map[string]string)
	as := viper.AllSettings()
	res, ok := as["res"].(map[string]interface{})
	if !ok {
	    slog.Error("獲取res配置文件信息失敗")
	    return nil
	}
	res_list, ok := res["res_list"].(map[string]interface{})
	if !ok {
	    slog.Error("獲取res_list配置文件信息失敗")
	    return nil
	}
	for k, v := range res_list {
	    resList[k] = v.(string)
	}

到此這篇關(guān)于golang使用viper解析配置文件的示例代碼的文章就介紹到這了,更多相關(guān)golang viper解析配置文件內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: