47 lines
745 B
Go
47 lines
745 B
Go
package config
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
func LoadConfig() (*Config, error) {
|
|
v := viper.New()
|
|
|
|
v.AutomaticEnv()
|
|
|
|
v.SetConfigName("config")
|
|
v.SetConfigType("yaml")
|
|
v.AddConfigPath("./config")
|
|
v.AddConfigPath(".")
|
|
|
|
if err := v.ReadInConfig(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
v.SetConfigName(".env")
|
|
v.SetConfigType("env")
|
|
|
|
if err := v.MergeInConfig(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, key := range v.AllKeys() {
|
|
if strings.Contains(key, "_") {
|
|
nestedKey := strings.Replace(key, "_", ".", 1)
|
|
|
|
if !v.IsSet(nestedKey) || v.Get(nestedKey) == "" {
|
|
v.Set(nestedKey, v.Get(key))
|
|
}
|
|
}
|
|
}
|
|
|
|
var cfg Config
|
|
if err := v.Unmarshal(&cfg); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &cfg, nil
|
|
}
|