快捷導(dǎo)航

Golang實(shí)現(xiàn)yaml配置文件的解析

更新時(shí)間：2024年12月27日 09:20:13 作者：瘋狂吧小飛牛

這篇文章主要介紹了Golang實(shí)現(xiàn)yaml配置文件的解析,文中通過圖文結(jié)合的方式講解的非常詳細(xì),對大家的學(xué)習(xí)或工作有一定的幫助,需要的朋友可以參考下

代碼

需要建立3個(gè)文件，目錄結(jié)構(gòu)如下：

配置文件 conf.yaml

redis:  
  host: "127.0.0.1"
  port: 6379
  db: 11  
  pass: "aaaaaa"

go模塊文件 go.mod，用于管理項(xiàng)目的依賴關(guān)系和版本信息。

module readyaml

主程序 readyaml.go

package main
 
import (
    "github.com/spf13/viper"
    "fmt"
)
 
// 配置文件
type Conf struct {
    Redis RedisConfig      `mapstructure:"redis"`
}
 
// redis配置數(shù)據(jù)
type RedisConfig struct{
    HOST    string    `mapstructure:"host"`
    PORT    int       `mapstructure:"port"`
    DB      int       `mapstructure:"db"`
    PASS    string    `mapstructure:"pass"`
}
 
func getConf() Conf{
    var cfg Conf 
 
    // 設(shè)置待讀取配置文件的名字
    viper.SetConfigName("conf")
    // 設(shè)置待讀取配置文件的格式
    viper.SetConfigType("yaml")
    // 添加配置文件的檢索路徑
    viper.AddConfigPath(".")
 
    // 讀取數(shù)據(jù)
    if err := viper.ReadInConfig(); err != nil {
        fmt.Printf("ReadInConfig error : %s\n", err)
        return cfg
    }
 
    // 測試 1 : 直接獲取數(shù)據(jù)
    fmt.Printf("Host %s\n", viper.GetString("redis.host"))
    fmt.Printf("PORT %d\n", viper.GetInt("redis.port"))
    fmt.Printf("DB %d\n", viper.GetInt("redis.db"))
    fmt.Printf("PASS %s\n", viper.GetString("redis.pass"))
 
    // 通過viper模塊，反序列化成對象
    if err := viper.Unmarshal(&cfg); err != nil {
        fmt.Printf("Unmarshal error : %s\n", err)
        return cfg
    }
 
    return cfg 
}
 
var cfg Conf = getConf()
 
func main() {
    // 打印測試
    fmt.Printf("HOST %s, PORT %d, DB %d, PASS %s len(PASS) %d\n", cfg.Redis.HOST, cfg.Redis.PORT, cfg.Redis.DB, cfg.Redis.PASS, len(cfg.Redis.PASS) )
}