适用场景:使用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"`
}