Golang实现yaml配置文件的解析

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

module readyaml

package main
 
import (
    "github.com/spf13/viper"
    "fmt"
)
 
// 配置文件
type Conf struct {
    Redis RedisConfig      `mapstructure:"redis"`
}
 
// redis配置数据
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 
 
    // 设置待读取配置文件的名字
    viper.SetConfigName("conf")
    // 设置待读取配置文件的格式
    viper.SetConfigType("yaml")
    // 添加配置文件的检索路径
    viper.AddConfigPath(".")
 
    // 读取数据
    if err := viper.ReadInConfig(); err != nil {
        fmt.Printf("ReadInConfig error : %s\n", err)
        return cfg
    }
 
    // 测试 1 : 直接获取数据
    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) )
}