u

code/allapp/allapp-go-v3/internal/config/loader.go

@@ -0,0 +1,50 @@
+package config
+
+import (
+	"fmt"
+	"os"
+	"strings"
+
+	"github.com/spf13/viper"
+)
+
+func Load() (*Config, error) {
+	env := getEnv("APP_ENV", "dev")
+
+	v := viper.New()
+	v.SetConfigType("yaml")
+
+	// 基础配置
+	v.AddConfigPath("configs")
+	v.SetConfigName("config")
+
+	if err := v.ReadInConfig(); err != nil {
+		return nil, fmt.Errorf("read base config failed: %w", err)
+	}
+
+	// 环境配置（覆盖）
+	envConfigName := fmt.Sprintf("config.%s", env)
+	v.SetConfigName(envConfigName)
+
+	if err := v.MergeInConfig(); err != nil {
+		// dev 允许没有文件，prod 建议必须存在
+		if env == "prod" {
+			return nil, fmt.Errorf("read env config failed: %w", err)
+		}
+	}
+
+	var cfg Config
+	if err := v.Unmarshal(&cfg); err != nil {
+		return nil, fmt.Errorf("unmarshal config failed: %w", err)
+	}
+
+	return &cfg, nil
+}
+
+func getEnv(key, defaultVal string) string {
+	val := os.Getenv(key)
+	if val == "" {
+		return defaultVal
+	}
+	return strings.ToLower(val)
+}