You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
1.8 KiB
64 lines
1.8 KiB
package config
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"reflect"
|
|
"strings"
|
|
|
|
"github.com/mitchellh/mapstructure"
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/openimsdk/tools/errs"
|
|
"github.com/openimsdk/tools/utils/runtimeenv"
|
|
)
|
|
|
|
func Load(configDirectory string, configFileName string, envPrefix string, config any) error {
|
|
if runtimeenv.RuntimeEnvironment() == KUBERNETES {
|
|
mountPath := os.Getenv(MountConfigFilePath)
|
|
if mountPath == "" {
|
|
return errs.ErrArgs.WrapMsg(MountConfigFilePath + " env is empty")
|
|
}
|
|
|
|
return loadConfig(filepath.Join(mountPath, configFileName), envPrefix, config)
|
|
}
|
|
|
|
return loadConfig(filepath.Join(configDirectory, configFileName), envPrefix, config)
|
|
}
|
|
|
|
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, ApplyDecoderConfig); err != nil {
|
|
return errs.WrapMsg(err, "failed to unmarshal config", "path", path, "envPrefix", envPrefix)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func ApplyDecoderConfig(config *mapstructure.DecoderConfig) {
|
|
config.TagName = StructTagName
|
|
config.DecodeHook = mapstructure.ComposeDecodeHookFunc(
|
|
mapstructure.StringToTimeDurationHookFunc(),
|
|
mapstructure.StringToSliceHookFunc(","),
|
|
stringToEngineSelectorHookFunc(),
|
|
)
|
|
}
|
|
|
|
func stringToEngineSelectorHookFunc() mapstructure.DecodeHookFuncType {
|
|
engineSelectorType := reflect.TypeOf(EngineSelector{})
|
|
return func(from reflect.Type, to reflect.Type, data any) (any, error) {
|
|
if to != engineSelectorType || from.Kind() != reflect.String {
|
|
return data, nil
|
|
}
|
|
return EngineSelector{Engine: data.(string)}, nil
|
|
}
|
|
}
|