快捷導(dǎo)航

golang讀取yaml配置文件的方法實(shí)現(xiàn)

更新時間：2024年10月25日 09:45:26 作者：code:404-not-found

本文主要介紹了golang讀取yaml配置文件的方法實(shí)現(xiàn),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

Viper

Viper是適用于Go應(yīng)用程序的完整配置解決方案。它被設(shè)計用于在應(yīng)用程序中工作，并且可以處理所有類型的配置需求和格式。它支持以下特性:

設(shè)置默認(rèn)值
從JSON、TOML、YAML、HCL、envfile和Java properties 格式的配置文件讀取配置信息
實(shí)時監(jiān)控和重新讀取配置文件(可選)
從環(huán)境變量中讀取
從遠(yuǎn)程配置系統(tǒng)(etcd或Consul)讀取并監(jiān)控配置變化
從命令行參數(shù)讀取配置
從buffer讀取配置
顯式配置值

安裝

go get github.com/spf13/viper

使用

在項(xiàng)目根目錄創(chuàng)建config文件夾，在文件夾中創(chuàng)建local.yaml

servername: 'myserver'

mysql:
  name: 'mysql'

在main.go中讀取配置文件的信息

package main

import (
	"log"

	"github.com/spf13/viper"
)

func main() {
	v := viper.New()

	// 設(shè)置文件名稱
	v.SetConfigFile("./config/local.yaml")

	if err := v.ReadInConfig(); err != nil {
		log.Panicln("讀取配置文件失敗")
	}

	servername := v.Get("servername")
	mysql := v.Get("mysql.name")
	log.Println(servername, mysql)
}

使用結(jié)構(gòu)體映射yaml

修改配置文件

name: '服務(wù)名'
port: 8080

使用

package main

import (
	"log"

	"github.com/spf13/viper"
)

type ServerConfig struct {
	Name string `mapstructure:"name"`
	Port int    `mapstructure:"port"`
}

func main() {

	cfg := ServerConfig{}
	v := viper.New()

	// 設(shè)置文件名稱
	v.SetConfigFile("./config/local.yaml")

	if err := v.ReadInConfig(); err != nil {
		log.Panicln("讀取配置文件失敗")
	}

	if err := v.Unmarshal(&cfg); err != nil {
		log.Panicln(err)
	}

	log.Println(cfg)
}

輸出

{服務(wù)名 8080}

結(jié)構(gòu)體嵌套讀取配置

配置文件

name: '服務(wù)名'
port: 8080
mysql: 
  host: '127.0.0.1'
  port: 3306

讀取配置

package main

import (
	"log"

	"github.com/spf13/viper"
)

type ServerConfig struct {
	Name        string      `mapstructure:"name"`
	Port        int         `mapstructure:"port"`
	MysqlConfig MysqlConfig `mapstructure:"mysql"` // 嵌套的配置
}

type MysqlConfig struct {
	Host string `mapstructure:"host"`
	Port int    `mapstructure:"port"`
}

func main() {

	cfg := ServerConfig{}
	v := viper.New()

	// 設(shè)置文件名稱
	v.SetConfigFile("./config/local.yaml")

	if err := v.ReadInConfig(); err != nil {
		log.Panicln("讀取配置文件失敗")
	}

	if err := v.Unmarshal(&cfg); err != nil {
		log.Panicln(err)
	}

	log.Println(cfg)
}

輸出

{服務(wù)名 8080 {127.0.0.1 3306}}

根據(jù)不同環(huán)境讀取不同配置文件

在實(shí)際開發(fā)工作中，一般有不同的開發(fā)環(huán)境，會涉及到讀取不同的配置文件

這會運(yùn)用的golang的flag包

package main

import (
	"flag"
	"log"
)


func main() {

	env := flag.String("env", "local", "運(yùn)行環(huán)境請輸入local dev test prod")
	flag.Parse()

	log.Println("當(dāng)前環(huán)境是" + *env)

}

運(yùn)行

go run .\main.go -help

輸出

-env string
運(yùn)行環(huán)境請輸入local dev test prod (default "local")

這樣的話運(yùn)行

go run .\main.go -env dev

輸出

當(dāng)前環(huán)境是dev

利用flag讀取不同環(huán)境的配置文件

在項(xiàng)目更目錄新建config文件夾

并創(chuàng)建4個配置文件 local.yaml dev.yaml test.yaml prod.yaml

其中dev.yaml中

name: 'dev服務(wù)名'
port: 8080
mysql: 
  host: '127.0.0.1'
  port: 3306

在main.go中

package main

import (
	"flag"
	"fmt"
	"log"

	"github.com/spf13/viper"
)

type ServerConfig struct {
	Name        string      `mapstructure:"name"`
	Port        int         `mapstructure:"port"`
	MysqlConfig MysqlConfig `mapstructure:"mysql"` // 嵌套的配置
}

type MysqlConfig struct {
	Host string `mapstructure:"host"`
	Port int    `mapstructure:"port"`
}

func main() {

	env := flag.String("env", "local", "運(yùn)行環(huán)境請輸入local dev test prod")
	flag.Parse()

	log.Println("當(dāng)前環(huán)境是" + *env)

	cfg := ServerConfig{}
	v := viper.New()
	var path string = fmt.Sprintf("./config/%s.yaml", *env)
	// 設(shè)置文件名稱
	v.SetConfigFile(path)

	if err := v.ReadInConfig(); err != nil {
		log.Panicln("讀取配置文件失敗")
	}

	if err := v.Unmarshal(&cfg); err != nil {
		log.Panicln(err)
	}

	log.Println(cfg)
}

運(yùn)行