mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-04-05 05:12:45 +08:00
24ab940875
* build: implement services image build. * remove unused tools * update test. * update images. * update dockerfile and go mod. * update go mod. * Add comments. * update go pkg. * update loadConfig and discovery logic in kubernetes. * update go pkg and discovery field. * update Load method args.
44 lines
1.2 KiB
Go
44 lines
1.2 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/mitchellh/mapstructure"
|
|
"github.com/openimsdk/tools/errs"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
func Load(configDirectory string, configFileName string, envPrefix string, runtimeEnv string, config any) error {
|
|
if runtimeEnv == 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, func(config *mapstructure.DecoderConfig) {
|
|
config.TagName = "mapstructure"
|
|
}); err != nil {
|
|
return errs.WrapMsg(err, "failed to unmarshal config", "path", path, "envPrefix", envPrefix)
|
|
}
|
|
return nil
|
|
}
|