set seperator instance

This commit is contained in:
yanghao05
2023-04-27 16:31:44 +08:00
parent f1fc4cf7fa
commit 8ae3d19aa9

View File

@@ -22,21 +22,23 @@ func LoadConfig(file, envPrefix string) *Config {
envPrefix = DefaultPrefix envPrefix = DefaultPrefix
} }
viper.SetEnvPrefix(envPrefix) v := viper.NewWithOptions(viper.KeyDelimiter("_"))
viper.AutomaticEnv()
v.SetEnvPrefix(envPrefix)
v.AutomaticEnv()
if !fs.FileExist(file) { if !fs.FileExist(file) {
return &Config{} return &Config{}
} }
// load file // load file
viper.SetConfigFile(file) v.SetConfigFile(file)
if err := viper.ReadInConfig(); err != nil { if err := v.ReadInConfig(); err != nil {
log.Fatal(err) log.Fatal(err)
} }
var config Config var config Config
if err := viper.Unmarshal(&config); err != nil { if err := v.Unmarshal(&config); err != nil {
log.Fatal(err) log.Fatal(err)
} }