mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-10-28 06:22:12 +08:00
30 lines
724 B
Go
30 lines
724 B
Go
package config
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/mitchellh/mapstructure"
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/openimsdk/tools/errs"
|
|
)
|
|
|
|
func LoadConfig(path string, envPrefix string, config any) error {
|
|
v := viper.New()
|
|
v.SetConfigFile(path)
|
|
v.SetEnvPrefix(envPrefix)
|
|
v.AutomaticEnv()
|
|
v.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
|
|
|
|
if err := v.ReadInConfig(); err != nil {
|
|
return errs.WrapMsg(err, "failed to read config file", "path", path, "envPrefix", envPrefix)
|
|
}
|
|
|
|
if err := v.Unmarshal(config, func(config *mapstructure.DecoderConfig) {
|
|
config.TagName = "mapstructure"
|
|
}); err != nil {
|
|
return errs.WrapMsg(err, "failed to unmarshal config", "path", path, "envPrefix", envPrefix)
|
|
}
|
|
return nil
|
|
}
|