Golang实现yaml配置文件的解析_Golang

代码

需要建立3个文件，目录结构如下：

配置文件 conf.yaml

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

go模块文件 go.mod，用于管理项目的依赖关系和版本信息。

module readyaml

主程序 readyaml.go

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) )
}