open-im-server/pkg/common/config/load_config.go
2024-04-03 16:51:00 +08:00

28 lines
518 B
Go

package config
import (
"github.com/mitchellh/mapstructure"
"github.com/spf13/viper"
"strings"
)
func LoadConfig(path string, prefix string, config any) error {
v := viper.New()
v.SetConfigFile(path)
v.SetEnvPrefix(prefix)
v.AutomaticEnv()
v.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
if err := v.ReadInConfig(); err != nil {
return err
}
if err := v.Unmarshal(config, func(config *mapstructure.DecoderConfig) {
config.TagName = "mapstructure"
}); err != nil {
return err
}
return nil
}