wz-golang-server/server/config/config.go

// Package config 应用配置
package config

import (
	"fmt"
	"os"

	"gopkg.in/yaml.v3"
)

// AppConfig 应用配置
var AppConfig = &appConfig{}

type appConfig struct {
	Log           LogConfig           `yaml:"log"`
	Server        ServerConfig        `yaml:"server"`
	Security      SecurityConfig      `yaml:"security"`
	PayloadCrypto PayloadCryptoConfig `yaml:"payload_crypto"`
	RateLimit     RateLimitConfig     `yaml:"rate_limit"`
	Swagger       SwaggerConfig       `yaml:"swagger"`
	Database      DatabaseConfig      `yaml:"database"`
	Redis         RedisConfig         `yaml:"redis"`
	Wechat        WechatConfig        `yaml:"wechat"`
	AppConfig     AppVersionConfig    `yaml:"app_config"`
}

// LogConfig 日志配置
type LogConfig struct {
	Level   string `yaml:"level"`   // 日志级别
	Dir     string `yaml:"dir"`     // 日志目录
	Console bool   `yaml:"console"` // 是否输出到控制台
}

// ServerConfig 服务配置
type ServerConfig struct {
	Port         int `yaml:"port"`          // 服务端口
	WorkerID     int `yaml:"worker_id"`     // 工作机器ID (0-31)，用于雪花算法
	DatacenterID int `yaml:"datacenter_id"` // 数据中心ID (0-31)，用于雪花算法
}

// SecurityConfig 安全配置
type SecurityConfig struct {
	Enable    bool   `yaml:"enable"`     // 是否启用
	HeaderKey string `yaml:"header_key"` // 请求头字段名
	SecretKey string `yaml:"secret_key"` // 签名密钥
}

// PayloadCryptoConfig 请求/响应参数加密配置
type PayloadCryptoConfig struct {
	Enable    bool                         `yaml:"enable"`     // 是否启用
	HeaderKey string                       `yaml:"header_key"` // 加密标记请求头字段名
	SecretKey string                       `yaml:"secret_key"` // 加密密钥
	Whitelist []string                     `yaml:"whitelist"`  // 白名单路径
	Request   PayloadCryptoDirectionConfig `yaml:"request"`    // 请求加密配置
	Response  PayloadCryptoDirectionConfig `yaml:"response"`   // 响应加密配置
}

// PayloadCryptoDirectionConfig 方向配置
type PayloadCryptoDirectionConfig struct {
	Enable   bool `yaml:"enable"`   // 是否启用
	Required bool `yaml:"required"` // 是否强制
}

// RateLimitConfig 限流配置
type RateLimitConfig struct {
	Enable  bool                     `yaml:"enable"`  // 是否启用
	Default RateLimitRule            `yaml:"default"` // 默认规则
	Rules   map[string]RateLimitRule `yaml:"rules"`   // 特定路径规则
}

// RateLimitRule 限流规则
type RateLimitRule struct {
	Interval    int `yaml:"interval"`     // 时间间隔(秒)
	MaxRequests int `yaml:"max_requests"` // 最大请求次数
}

// SwaggerConfig Swagger文档认证配置
type SwaggerConfig struct {
	User     string `yaml:"user"`
	Password string `yaml:"password"`
}

// DatabaseConfig 数据库配置
type DatabaseConfig struct {
	Driver          string `yaml:"driver"`
	Host            string `yaml:"host"`
	Port            int    `yaml:"port"`
	Database        string `yaml:"database"`
	Username        string `yaml:"username"`
	Password        string `yaml:"password"`
	Charset         string `yaml:"charset"`
	MaxIdleConns    int    `yaml:"max_idle_conns"`
	MaxOpenConns    int    `yaml:"max_open_conns"`
	ConnMaxLifetime int    `yaml:"conn_max_lifetime"` // 小时
	LogMode         bool   `yaml:"log_mode"`          // 是否开启SQL日志
}

// RedisConfig Redis配置
type RedisConfig struct {
	Addr     string `yaml:"addr"`
	Password string `yaml:"password"`
	DB       int    `yaml:"db"`
}

// WechatConfig 微信配置
type WechatConfig struct {
	MiniProgram WechatMiniProgramConfig `yaml:"mini_program"`
}

// WechatMiniProgramConfig 微信小程序配置
type WechatMiniProgramConfig struct {
	AppID     string `yaml:"app_id"`
	AppSecret string `yaml:"app_secret"`
}

// AppVersionConfig 小程序版本与配置中心
type AppVersionConfig struct {
	App           AppClientConfig   `yaml:"app"`
	API           AppEndpointConfig `yaml:"api"`
	WebView       AppEndpointConfig `yaml:"webview"`
	TTLSeconds    int               `yaml:"ttl_seconds"`
	Disabled      bool              `yaml:"disabled"`
	DisableReason string            `yaml:"disable_reason"`
	TenantId      string            `yaml:"tenantId"`
}

// AppClientConfig 客户端版本配置
type AppClientConfig struct {
	MinVersion    string `yaml:"min_version"`
	LatestVersion string `yaml:"latest_version"`
	ForceUpdate   bool   `yaml:"force_update"`
}

// AppEndpointConfig 接口或 WebView 配置
type AppEndpointConfig struct {
	BaseURL          string `yaml:"base_url"`
	Version          string `yaml:"version"`
	MinClientVersion string `yaml:"min_client_version"`
}

// LoadConfig 加载配置
func LoadConfig() {
	var configFile string

	// 1. 优先检查命令行参数 (简单解析)
	// 格式: ./app -c config.prod.yaml 或 ./app -config config.prod.yaml
	args := os.Args
	for i, arg := range args {
		if (arg == "-c" || arg == "-config") && i+1 < len(args) {
			configFile = args[i+1]
			break
		}
	}

	// 2. 如果命令行参数未指定，则根据环境变量查找
	if configFile == "" {
		env := os.Getenv("GO_ENV")
		if env == "" {
			env = "dev"
		}

		// 查找顺序：
		// 1. 当前目录 config.{env}.yaml (方便部署时直接放在执行文件旁)
		// 2. config/config.{env}.yaml (开发习惯)
		// 3. ../config/config.{env}.yaml (测试环境习惯)
		searchPaths := []string{
			fmt.Sprintf("config.%s.yaml", env),
			fmt.Sprintf("config/config.%s.yaml", env),
			fmt.Sprintf("../config/config.%s.yaml", env),
		}

		for _, path := range searchPaths {
			if _, err := os.Stat(path); err == nil {
				configFile = path
				break
			}
		}

		// 如果都没找到，默认回退到 config/config.{env}.yaml 以便报错信息准确
		if configFile == "" {
			configFile = fmt.Sprintf("config/config.%s.yaml", env)
		}
	}

	fmt.Printf("正在加载配置文件: %s\n", configFile)

	data, err := os.ReadFile(configFile)
	if err != nil {
		fmt.Printf("读取配置文件失败: %v, 使用默认配置\n", err)
		return
	}

	err = yaml.Unmarshal(data, AppConfig)
	if err != nil {
		fmt.Printf("解析配置文件失败: %v\n", err)
		panic(err)
	}
}