New warehouse

2026-05-25 02:24:05 +08:00 · 2025-10-24 00:09:45 +08:00
commit ac07e27908
75 changed files with 26814 additions and 0 deletions
--- a/config/config.go
+++ b/config/config.go
@@ -0,0 +1,79 @@
+package config
+
+import (
+	"bytes"
+	_ "embed"
+	"errors"
+	"io/fs"
+	"os"
+
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/viper"
+)
+
+//go:embed config.json
+var DefaultConfig string
+
+// Init 初始化配置文件
+func Init(cfgFilePath string) {
+	viper.SetConfigFile(cfgFilePath)
+	viper.SetConfigType("json")
+	viper.AddConfigPath(".")
+
+	if err := viper.ReadInConfig(); err != nil {
+		var pathError *fs.PathError
+		if errors.As(err, &pathError) {
+			log.Warn("未找到配置文件，使用默认配置")
+			err = os.WriteFile(cfgFilePath, []byte(DefaultConfig), 0o644)
+			if err != nil {
+				log.WithFields(
+					log.Fields{
+						"err": err,
+					},
+				).Error("写入默认配置文件失败")
+			} else {
+				log.WithFields(
+					log.Fields{
+						"file": cfgFilePath,
+					},
+				).Info("写入默认配置文件成功")
+			}
+			// 写完默认配置后再读一次
+			err = viper.ReadConfig(bytes.NewBuffer([]byte(DefaultConfig)))
+			if err != nil {
+				log.WithFields(
+					log.Fields{
+						"err": err,
+					},
+				).Error("读取默认配置文件失败")
+			} else {
+				log.Info("已成功读取默认配置")
+			}
+		} else {
+			log.WithFields(
+				log.Fields{
+					"err": err,
+				},
+			).Fatal("配置文件解析错误")
+		}
+	}
+	log.WithFields(
+		log.Fields{
+			"file": viper.ConfigFileUsed(),
+		},
+	).Info("使用配置文件")
+
+	// 验证配置并设置默认值
+	if _, err := ValidateAndSetDefaults(); err != nil {
+		log.WithFields(
+			log.Fields{
+				"err": err,
+			},
+		).Fatal("配置验证失败")
+	}
+}
+
+// CreateDefaultConfig 创建默认配置文件
+func CreateDefaultConfig(filePath string) error {
+	return os.WriteFile(filePath, []byte(DefaultConfig), 0o644)
+}
--- a/config/validator.go
+++ b/config/validator.go
@@ -0,0 +1,410 @@
+package config
+
+import (
+	"errors"
+	"fmt"
+	"net"
+	"os"
+	"path/filepath"
+	"strconv"
+	"strings"
+
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/viper"
+)
+
+// ServerConfig 服务器配置结构体
+// 包含HTTP服务器的基本配置信息
+type ServerConfig struct {
+	Host string `json:"host" mapstructure:"host"` // 服务器监听地址
+	Port int    `json:"port" mapstructure:"port"` // 服务器监听端口
+	Mode string `json:"mode" mapstructure:"mode"` // 运行模式（debug/release）
+	Dist string `json:"dist" mapstructure:"dist"` // 静态文件目录
+}
+
+// DatabaseConfig 数据库配置结构体
+// 支持MySQL和SQLite两种数据库类型
+type DatabaseConfig struct {
+	Type   string       `json:"type" mapstructure:"type"`     // 数据库类型（mysql/sqlite）
+	MySQL  MySQLConfig  `json:"mysql" mapstructure:"mysql"`   // MySQL配置
+	SQLite SQLiteConfig `json:"sqlite" mapstructure:"sqlite"` // SQLite配置
+}
+
+// MySQLConfig MySQL数据库配置结构体
+// 包含MySQL数据库连接和连接池的配置信息
+type MySQLConfig struct {
+	Host         string `json:"host" mapstructure:"host"`                     // 数据库主机地址
+	Port         int    `json:"port" mapstructure:"port"`                     // 数据库端口
+	Username     string `json:"username" mapstructure:"username"`             // 数据库用户名
+	Password     string `json:"password" mapstructure:"password"`             // 数据库密码
+	Database     string `json:"database" mapstructure:"database"`             // 数据库名称
+	Charset      string `json:"charset" mapstructure:"charset"`               // 字符集
+	MaxIdleConns int    `json:"max_idle_conns" mapstructure:"max_idle_conns"` // 最大空闲连接数
+	MaxOpenConns int    `json:"max_open_conns" mapstructure:"max_open_conns"` // 最大打开连接数
+}
+
+// SQLiteConfig SQLite数据库配置结构体
+// 包含SQLite数据库文件路径配置
+type SQLiteConfig struct {
+	Path string `json:"path" mapstructure:"path"` // 数据库文件路径
+}
+
+// RedisConfig Redis配置结构体
+// 包含Redis缓存服务器的连接配置
+type RedisConfig struct {
+	Host     string `json:"host" mapstructure:"host"`         // Redis服务器地址
+	Port     int    `json:"port" mapstructure:"port"`         // Redis服务器端口
+	Password string `json:"password" mapstructure:"password"` // Redis密码
+	DB       int    `json:"db" mapstructure:"db"`             // Redis数据库编号
+}
+
+// LogConfig 日志配置结构体
+// 包含日志记录的相关配置信息
+type LogConfig struct {
+	Level      string `json:"level" mapstructure:"level"`             // 日志级别
+	File       string `json:"file" mapstructure:"file"`               // 日志文件路径
+	MaxSize    int    `json:"max_size" mapstructure:"max_size"`       // 单个日志文件最大大小(MB)
+	MaxBackups int    `json:"max_backups" mapstructure:"max_backups"` // 保留的旧日志文件数量
+	MaxAge     int    `json:"max_age" mapstructure:"max_age"`         // 日志文件保留天数
+}
+
+// SecurityConfig 安全配置结构体
+// 包含应用程序安全相关的配置信息
+type SecurityConfig struct {
+	JWTSecret                string `json:"jwt_secret" mapstructure:"jwt_secret"`                                   // JWT签名密钥
+	EncryptionKey            string `json:"encryption_key" mapstructure:"encryption_key"`                           // 数据加密密钥
+	JWTRefreshThresholdHours int    `json:"jwt_refresh_threshold_hours" mapstructure:"jwt_refresh_threshold_hours"` // JWT令牌刷新阈值（小时）
+}
+
+// AppConfig 应用配置结构体
+type AppConfig struct {
+	Server   ServerConfig   `json:"server" mapstructure:"server"`
+	Database DatabaseConfig `json:"database" mapstructure:"database"`
+	Redis    RedisConfig    `json:"redis" mapstructure:"redis"`
+	Log      LogConfig      `json:"log" mapstructure:"log"`
+	Security SecurityConfig `json:"security" mapstructure:"security"`
+}
+
+// ValidateAndSetDefaults 验证配置并设置默认值
+func ValidateAndSetDefaults() (*AppConfig, error) {
+	var config AppConfig
+
+	// 解析配置到结构体
+	if err := viper.Unmarshal(&config); err != nil {
+		return nil, fmt.Errorf("解析配置失败: %w", err)
+	}
+
+	// 设置默认值
+	setDefaults(&config)
+
+	// 验证配置
+	if err := validateConfig(&config); err != nil {
+		return nil, fmt.Errorf("配置验证失败: %w", err)
+	}
+
+	log.Info("配置验证通过")
+	return &config, nil
+}
+
+// setDefaults 设置默认值
+func setDefaults(config *AppConfig) {
+	// 服务器默认值
+	if config.Server.Host == "" {
+		config.Server.Host = "0.0.0.0"
+	}
+	if config.Server.Port == 0 {
+		config.Server.Port = 8080
+	}
+	if config.Server.Mode == "" {
+		config.Server.Mode = "debug"
+	}
+
+	// 数据库默认值
+	if config.Database.Type == "" {
+		config.Database.Type = "sqlite"
+	}
+	if config.Database.MySQL.Port == 0 {
+		config.Database.MySQL.Port = 3306
+	}
+	if config.Database.MySQL.Charset == "" {
+		config.Database.MySQL.Charset = "utf8mb4"
+	}
+	if config.Database.MySQL.MaxIdleConns == 0 {
+		config.Database.MySQL.MaxIdleConns = 10
+	}
+	if config.Database.MySQL.MaxOpenConns == 0 {
+		config.Database.MySQL.MaxOpenConns = 100
+	}
+	if config.Database.SQLite.Path == "" {
+		config.Database.SQLite.Path = "./recharge.db"
+	}
+
+	// Redis默认值
+	if config.Redis.Host == "" {
+		config.Redis.Host = "localhost"
+	}
+	if config.Redis.Port == 0 {
+		config.Redis.Port = 6379
+	}
+
+	// 日志默认值
+	if config.Log.Level == "" {
+		config.Log.Level = "info"
+	}
+	// 不为空的日志文件路径设置默认值，保持为空表示只输出到控制台
+	if config.Log.MaxSize == 0 {
+		config.Log.MaxSize = 100
+	}
+	if config.Log.MaxBackups == 0 {
+		config.Log.MaxBackups = 5
+	}
+	if config.Log.MaxAge == 0 {
+		config.Log.MaxAge = 30
+	}
+
+	// 安全配置默认值
+	if config.Security.JWTSecret == "" || config.Security.JWTSecret == "your-jwt-secret-key" {
+		config.Security.JWTSecret = "default-jwt-secret-change-in-production"
+	}
+	if config.Security.EncryptionKey == "" || config.Security.EncryptionKey == "your-encryption-key" {
+		config.Security.EncryptionKey = "default-encryption-key-change-in-production"
+	}
+	if config.Security.JWTRefreshThresholdHours == 0 {
+		config.Security.JWTRefreshThresholdHours = 6
+	}
+}
+
+// validateConfig 验证配置
+func validateConfig(config *AppConfig) error {
+	// 验证服务器配置
+	if err := validateServerConfig(&config.Server); err != nil {
+		return fmt.Errorf("服务器配置错误: %w", err)
+	}
+
+	// 验证数据库配置
+	if err := validateDatabaseConfig(&config.Database); err != nil {
+		return fmt.Errorf("数据库配置错误: %w", err)
+	}
+
+	// 验证Redis配置
+	if err := validateRedisConfig(&config.Redis); err != nil {
+		return fmt.Errorf("redis配置错误: %w", err)
+	}
+
+	// 验证日志配置
+	if err := validateLogConfig(&config.Log); err != nil {
+		return fmt.Errorf("日志配置错误: %w", err)
+	}
+
+	// 验证安全配置
+	if err := validateSecurityConfig(&config.Security); err != nil {
+		return fmt.Errorf("安全配置错误: %w", err)
+	}
+
+	return nil
+}
+
+// validateServerConfig 验证服务器配置
+func validateServerConfig(config *ServerConfig) error {
+	// 验证主机地址
+	if config.Host != "" {
+		if ip := net.ParseIP(config.Host); ip == nil && config.Host != "localhost" {
+			return fmt.Errorf("无效的主机地址: %s", config.Host)
+		}
+	}
+
+	// 验证端口
+	if config.Port < 1 || config.Port > 65535 {
+		return fmt.Errorf("无效的端口号: %d，端口号必须在1-65535之间", config.Port)
+	}
+
+	// 验证运行模式
+	validModes := []string{"debug", "release", "test"}
+	if !contains(validModes, config.Mode) {
+		return fmt.Errorf("无效的运行模式: %s，支持的模式: %s", config.Mode, strings.Join(validModes, ", "))
+	}
+
+	return nil
+}
+
+// validateDatabaseConfig 验证数据库配置
+func validateDatabaseConfig(config *DatabaseConfig) error {
+	// 验证数据库类型
+	validTypes := []string{"mysql", "sqlite"}
+	if !contains(validTypes, config.Type) {
+		return fmt.Errorf("不支持的数据库类型: %s，支持的类型: %s", config.Type, strings.Join(validTypes, ", "))
+	}
+
+	// 根据类型验证具体配置
+	switch config.Type {
+	case "mysql":
+		return validateMySQLConfig(&config.MySQL)
+	case "sqlite":
+		return validateSQLiteConfig(&config.SQLite)
+	}
+
+	return nil
+}
+
+// validateMySQLConfig 验证MySQL配置
+func validateMySQLConfig(config *MySQLConfig) error {
+	if config.Host == "" {
+		return errors.New("MySQL主机地址不能为空")
+	}
+	if config.Port < 1 || config.Port > 65535 {
+		return fmt.Errorf("无效的MySQL端口号: %d", config.Port)
+	}
+	if config.Username == "" {
+		return errors.New("MySQL用户名不能为空")
+	}
+	if config.Database == "" {
+		return errors.New("MySQL数据库名不能为空")
+	}
+	if config.MaxIdleConns < 0 {
+		return errors.New("MySQL最大空闲连接数不能为负数")
+	}
+	if config.MaxOpenConns < 0 {
+		return errors.New("MySQL最大打开连接数不能为负数")
+	}
+	return nil
+}
+
+// validateSQLiteConfig 验证SQLite配置
+func validateSQLiteConfig(config *SQLiteConfig) error {
+	if config.Path == "" {
+		return errors.New("SQLite数据库路径不能为空")
+	}
+
+	// 检查目录是否存在，不存在则创建
+	dir := filepath.Dir(config.Path)
+	if _, err := os.Stat(dir); os.IsNotExist(err) {
+		if err := os.MkdirAll(dir, 0755); err != nil {
+			return fmt.Errorf("创建SQLite数据库目录失败: %w", err)
+		}
+	}
+
+	return nil
+}
+
+// validateRedisConfig 验证Redis配置
+func validateRedisConfig(config *RedisConfig) error {
+	if config.Host == "" {
+		return errors.New("Redis主机地址不能为空")
+	}
+	if config.Port < 1 || config.Port > 65535 {
+		return fmt.Errorf("无效的Redis端口号: %d", config.Port)
+	}
+	if config.DB < 0 || config.DB > 15 {
+		return fmt.Errorf("无效的Redis数据库索引: %d，必须在0-15之间", config.DB)
+	}
+	return nil
+}
+
+// validateLogConfig 验证日志配置
+func validateLogConfig(config *LogConfig) error {
+	// 验证日志级别
+	validLevels := []string{"trace", "debug", "info", "warn", "error", "fatal", "panic"}
+	if !contains(validLevels, config.Level) {
+		return fmt.Errorf("无效的日志级别: %s，支持的级别: %s", config.Level, strings.Join(validLevels, ", "))
+	}
+
+	// 检查日志文件目录（仅当日志文件路径不为空时）
+	if config.File != "" {
+		dir := filepath.Dir(config.File)
+		if _, err := os.Stat(dir); os.IsNotExist(err) {
+			if err := os.MkdirAll(dir, 0755); err != nil {
+				return fmt.Errorf("创建日志目录失败: %w", err)
+			}
+		}
+	}
+	// 当日志文件路径为空时，不进行目录检查和创建
+
+	// 验证日志轮转配置
+	if config.MaxSize <= 0 {
+		return errors.New("日志文件最大大小必须大于0")
+	}
+	if config.MaxBackups < 0 {
+		return errors.New("日志备份文件数量不能为负数")
+	}
+	if config.MaxAge < 0 {
+		return errors.New("日志文件保留天数不能为负数")
+	}
+
+	return nil
+}
+
+// validateSecurityConfig 验证安全配置
+func validateSecurityConfig(config *SecurityConfig) error {
+	if len(config.JWTSecret) < 16 {
+		return errors.New("JWT密钥长度不能少于16个字符")
+	}
+
+	if len(config.EncryptionKey) < 16 {
+		return errors.New("加密密钥长度不能少于16个字符")
+	}
+
+	if config.JWTRefreshThresholdHours < 1 || config.JWTRefreshThresholdHours > 23 {
+		return errors.New("JWT令牌刷新阈值必须在1-23小时之间")
+	}
+
+	// 检查是否使用默认值（生产环境警告）
+	if strings.Contains(config.JWTSecret, "default") {
+		log.Warn("检测到使用默认JWT密钥，生产环境请更换为安全的密钥")
+	}
+
+	if strings.Contains(config.EncryptionKey, "default") {
+		log.Warn("检测到使用默认加密密钥，生产环境请更换为安全的密钥")
+	}
+
+	return nil
+}
+
+// contains 检查切片是否包含指定元素
+func contains(slice []string, item string) bool {
+	for _, s := range slice {
+		if s == item {
+			return true
+		}
+	}
+	return false
+}
+
+// GetConfigValue 获取配置值，支持类型转换和默认值
+func GetConfigValue[T any](key string, defaultValue T) T {
+	if !viper.IsSet(key) {
+		return defaultValue
+	}
+
+	value := viper.Get(key)
+	if result, ok := value.(T); ok {
+		return result
+	}
+
+	// 尝试类型转换
+	if converted, err := convertValue[T](value); err == nil {
+		return converted
+	}
+
+	return defaultValue
+}
+
+// convertValue 尝试类型转换
+func convertValue[T any](value interface{}) (T, error) {
+	var zero T
+	str := fmt.Sprintf("%v", value)
+
+	switch any(zero).(type) {
+	case int:
+		if i, err := strconv.Atoi(str); err == nil {
+			return any(i).(T), nil
+		}
+	case string:
+		return any(str).(T), nil
+	case bool:
+		if b, err := strconv.ParseBool(str); err == nil {
+			return any(b).(T), nil
+		}
+	}
+
+	return zero, fmt.Errorf("无法转换类型")
+}