20260507213409

code/app/app-go/internal/config/config.go

@@ -0,0 +1,54 @@
+package config
+
+import "time"
+
+type Config struct {
+	App      AppConfig      `mapstructure:"app"`
+	JWT      JwtConfig      `mapstructure:"jwt"`
+	UniqueID UniqueIDConfig `mapstructure:"unique_id"`
+	Postgres PostgresConfig `mapstructure:"postgres"`
+	Wechat   WechatConfig   `mapstructure:"wechat"`
+	AwsS3    AwsS3Config    `mapstructure:"aws_s3"`
+}
+
+type AppConfig struct {
+	Name    string `mapstructure:"name"`
+	Port    int    `mapstructure:"port"`
+	BaseUrl string `mapstructure:"base_url"`
+}
+
+type PostgresConfig struct {
+	Host            string        `mapstructure:"host"`
+	Port            int           `mapstructure:"port"`
+	User            string        `mapstructure:"user"`
+	Password        string        `mapstructure:"password"`
+	Dbname          string        `mapstructure:"dbname"`
+	SslMode         bool          `mapstructure:"ssl_mode"`
+	TimeZone        string        `mapstructure:"timezone"`
+	MaxOpenConns    int32         `mapstructure:"max_open_conns"`
+	MaxIdleConns    int32         `mapstructure:"max_idle_conns"`
+	ConnMaxLifetime time.Duration `mapstructure:"conn_max_lifetime"`
+}
+
+type JwtConfig struct {
+	Secret       string        `mapstructure:"secret"`
+	AccessExpiry time.Duration `mapstructure:"access_expiry"`
+}
+
+type UniqueIDConfig struct {
+	DataCenterID uint16 `mapstructure:"datacenter_id"`
+	WorkerID     uint16 `mapstructure:"worker_id"`
+}
+
+type WechatConfig struct {
+	AppId     string `mapstructure:"app_id"`
+	AppSecret string `mapstructure:"app_secret"`
+}
+
+type AwsS3Config struct {
+	Region    string `mapstructure:"region"`
+	AccessKey string `mapstructure:"access_key"`
+	SecretKey string `mapstructure:"secret_key"`
+	Bucket    string `mapstructure:"bucket"`
+	Endpoint  string `mapstructure:"endpoint"`
+}

code/app/app-go/internal/config/loader.go

@@ -0,0 +1,68 @@
+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")
+
+	// ✅ 默认值
+	setDefaults(v)
+
+	// ✅ 支持环境变量（如 APP_PORT → app.port）
+	v.AutomaticEnv()
+	v.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
+
+	// 基础配置
+	v.AddConfigPath("configs")
+	v.SetConfigName("config")
+
+	if err := v.ReadInConfig(); err != nil {
+		return nil, fmt.Errorf("读取基础配置失败: %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("读取环境配置失败(%s): %w", env, err)
+		}
+	}
+
+	var cfg Config
+	if err := v.Unmarshal(&cfg); err != nil {
+		return nil, fmt.Errorf("解析配置失败: %w", err)
+	}
+
+	return &cfg, nil
+}
+
+// ✅ 默认值统一在这里管理
+func setDefaults(v *viper.Viper) {
+	v.SetDefault("app.name", "allapp-go")
+	v.SetDefault("app.port", 8080)
+	v.SetDefault("app.base_url", "app")
+
+	// 可以继续扩展
+	v.SetDefault("unique_id.datacenter_id", 1)
+	v.SetDefault("unique_id.worker_id", 1)
+}
+
+func getEnv(key, defaultVal string) string {
+	val := os.Getenv(key)
+	if val == "" {
+		return defaultVal
+	}
+	return strings.ToLower(val)
+}