适用场景:使用viper 读取yaml 文件时候,遇到yaml 文件中存在下划线,映射到结构体之后,结构体无法取到yaml 文件中对应的值。
conf.yaml 文件如下:
mysql:
rds_host: "rds_xxxxxx.com"
rds_port: 3306
config.go 部分内容如下:
type Mysql struct {
RdsHost string `yaml:"rds_host" mapstructure:"rds_host"`
RdsPort int `yaml:"rds_port" mapstructure:"rds_port"`
}
type Config struct {
Mysql Mysql `yaml:"mysql"`
}
var Conf *Config
func InitConf(confPath string) {
var err error
v := viper.New()
v.SetConfigName("config")
v.SetConfigType("yaml")
v.AddConfigPath("xxxxxx")
err = v.ReadInConfig()
if err != nil {
logrus.Errorf("[viper] ReadInConfig err:%s", err.Error())
return
}
err = v.Unmarshal(&Conf)
if err != nil {
logrus.Errorf("[viper] Unmarshal err:%s", err.Error())
panic(err)
}
return
}
在读取yaml 映射到Conf 结构体时,发现Mysql 中的RdsHost 和RdsPort 都是空值,仔细检查,发现tag 中的
字段也是对得上的,为何不映射不到?
翻了翻viper 解析的部分源码,发现有些特殊情况,需要自定义kv 的分隔符,有些时候,需要添加一些特殊标记来告诉解析器,这是一个合法的标识符。
将Mysql 的结构体添加mapstructure ,指定对应的字段就行了:
type Mysql struct {
RdsHost string `yaml:"rds_host" mapstructure:"rds_host"`
RdsPort int `yaml:"rds_port" mapstructure:"rds_port"`
}
|