复制项目

pkg/common/config/load.go

@@ -0,0 +1,45 @@
+package config
+
+import (
+	"os"
+	"path/filepath"
+	"strings"
+
+	"git.imall.cloud/openim/chat/pkg/common/constant"
+	"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 == constant.KUBERNETES {
+		mountPath := os.Getenv(constant.MountConfigFilePath)
+		if mountPath == "" {
+			return errs.ErrArgs.WrapMsg(constant.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
+}