上传初始版，功能未经任何测试，仅为基本框架，目前为完全不可用状态

2026-01-02 01:19:24 +08:00
parent 8c39d85831
commit 2c8afc1275
18 changed files with 2729 additions and 34 deletions
@@ -0,0 +1,169 @@
+package config
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"time"
+
+	"github.com/spf13/viper"
+)
+
+// Config 应用配置结构体
+type Config struct {
+	Server   ServerConfig   `mapstructure:"server"`
+	Database DatabaseConfig `mapstructure:"database"`
+	JWT      JWTConfig      `mapstructure:"jwt"`
+	AI       AIConfig       `mapstructure:"ai"`
+	Quota    QuotaConfig    `mapstructure:"quota"`
+	CORS     CORSConfig     `mapstructure:"cors"`
+}
+
+// ServerConfig 服务器配置
+type ServerConfig struct {
+	Port         string        `mapstructure:"port"`
+	Host         string        `mapstructure:"host"`
+	ReadTimeout  time.Duration `mapstructure:"read_timeout"`
+	WriteTimeout time.Duration `mapstructure:"write_timeout"`
+}
+
+// DatabaseConfig 数据库配置
+type DatabaseConfig struct {
+	DSN          string `mapstructure:"dsn"`
+	MaxIdleConns int    `mapstructure:"max_idle_conns"`
+	MaxOpenConns int    `mapstructure:"max_open_conns"`
+}
+
+// JWTConfig JWT配置
+type JWTConfig struct {
+	SecretKey     string        `mapstructure:"secret_key"`
+	AccessTokenExpire time.Duration `mapstructure:"access_token_expire"`
+	RefreshTokenExpire time.Duration `mapstructure:"refresh_token_expire"`
+}
+
+// AIConfig AI模型配置
+type AIConfig struct {
+	APIURL      string `mapstructure:"api_url"`
+	APIKey      string `mapstructure:"api_key"`
+	ModelName   string `mapstructure:"model_name"`
+	MaxTokens   int64  `mapstructure:"max_tokens"`
+	Temperature float64 `mapstructure:"temperature"`
+}
+
+// QuotaConfig 配额配置
+type QuotaConfig struct {
+	DefaultTotalTokens int64         `mapstructure:"default_total_tokens"`
+	DefaultTokenLimit  int64         `mapstructure:"default_token_limit"`
+	ResetInterval      time.Duration `mapstructure:"reset_interval"`
+}
+
+// CORSConfig CORS配置
+type CORSConfig struct {
+	AllowOrigins     []string `mapstructure:"allow_origins"`
+	AllowMethods     []string `mapstructure:"allow_methods"`
+	AllowHeaders     []string `mapstructure:"allow_headers"`
+	AllowCredentials bool     `mapstructure:"allow_credentials"`
+}
+
+// LoadConfig 加载配置
+func LoadConfig(path string) (*Config, error) {
+	// 设置默认值
+	setDefaults()
+	
+	// 读取配置文件
+	viper.AddConfigPath(path)
+	viper.SetConfigName(".env")
+	viper.SetConfigType("env")
+	
+	// 读取环境变量
+	viper.AutomaticEnv()
+	
+	// 尝试读取配置文件
+	if err := viper.ReadInConfig(); err != nil {
+		if _, ok := err.(viper.ConfigFileNotFoundError); ok {
+			log.Println("警告: 未找到配置文件，将使用环境变量和默认值")
+		} else {
+			return nil, fmt.Errorf("读取配置文件错误: %w", err)
+		}
+	}
+	
+	// 解析配置
+	var config Config
+	if err := viper.Unmarshal(&config); err != nil {
+		return nil, fmt.Errorf("解析配置错误: %w", err)
+	}
+	
+	// 验证必要的配置
+	if err := validateConfig(&config); err != nil {
+		return nil, err
+	}
+	
+	return &config, nil
+}
+
+// 设置默认值
+func setDefaults() {
+	// 服务器默认配置
+	viper.SetDefault("server.port", "8000")
+	viper.SetDefault("server.host", "0.0.0.0")
+	viper.SetDefault("server.read_timeout", 15*time.Second)
+	viper.SetDefault("server.write_timeout", 15*time.Second)
+	
+	// 数据库默认配置（使用SQLite）
+	viper.SetDefault("database.dsn", "./chat.db")
+	viper.SetDefault("database.max_idle_conns", 10)
+	viper.SetDefault("database.max_open_conns", 100)
+	
+	// JWT默认配置
+	viper.SetDefault("jwt.secret_key", "your-secret-key-change-me")
+	viper.SetDefault("jwt.access_token_expire", 24*time.Hour)
+	viper.SetDefault("jwt.refresh_token_expire", 7*24*time.Hour)
+	
+	// AI模型默认配置
+	viper.SetDefault("ai.api_url", "http://localhost:5000/v1/chat/completions")
+	viper.SetDefault("ai.api_key", "")
+	viper.SetDefault("ai.model_name", "default")
+	viper.SetDefault("ai.max_tokens", 2048)
+	viper.SetDefault("ai.temperature", 0.7)
+	
+	// 配额默认配置
+	viper.SetDefault("quota.default_total_tokens", 100000)
+	viper.SetDefault("quota.default_token_limit", 1000)
+	viper.SetDefault("quota.reset_interval", 30*24*time.Hour)
+	
+	// CORS默认配置
+	viper.SetDefault("cors.allow_origins", []string{"*"})
+	viper.SetDefault("cors.allow_methods", []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"})
+	viper.SetDefault("cors.allow_headers", []string{"Origin", "Content-Type", "Authorization"})
+	viper.SetDefault("cors.allow_credentials", true)
+}
+
+// 验证配置
+func validateConfig(config *Config) error {
+	// 验证JWT密钥
+	if config.JWT.SecretKey == "your-secret-key-change-me" {
+		log.Println("警告: 使用默认JWT密钥，建议在生产环境中修改")
+	}
+	
+	// 验证AI API URL
+	if config.AI.APIURL == "" {
+		return fmt.Errorf("AI API URL不能为空")
+	}
+	
+	return nil
+}
+
+// GetDSN 获取数据库DSN
+func (c *DatabaseConfig) GetDSN() string {
+	// 如果是SQLite，确保目录存在
+	if len(c.DSN) > 6 && c.DSN[:6] == "sqlite" {
+		// 提取SQLite文件路径
+		filePath := c.DSN[9:] // 去掉 "sqlite://" 前缀
+		dir := filePath[:len(filePath)-len("/"+filePath[len(filePath)-1:])]
+		if dir != "" {
+			os.MkdirAll(dir, 0755)
+		}
+	}
+	
+	return c.DSN
+}