Merge branch 'dev-mysql-support' into master

- 新增 MySQL 数据库驱动支持，支持跨设备数据同步
- 新增 MySQL 专项测试能力（并发、约束、迁移）
- 重构迁移目录结构：migrations/sqlite 和 migrations/mysql
- 修复 statsRepo 并发竞态条件，使用 upsert 保证原子性
- Makefile 合并：保留完整命令体系 + 新增 MySQL 测试命令

backend/cmd/desktop/main.go

@@ -17,16 +17,13 @@ import (
 	"github.com/getlantern/systray"
 	"github.com/gin-gonic/gin"
 	"github.com/gofrs/flock"
-	"github.com/pressly/goose/v3"
 	"go.uber.org/zap"
-	"gorm.io/driver/sqlite"
-	"gorm.io/gorm"
-	"gorm.io/gorm/logger"
 
 	"nex/backend/internal/config"
 	"nex/backend/internal/conversion"
 	"nex/backend/internal/conversion/anthropic"
 	"nex/backend/internal/conversion/openai"
+	"nex/backend/internal/database"
 	"nex/backend/internal/handler"
 	"nex/backend/internal/handler/middleware"
 	"nex/backend/internal/provider"
@@ -79,12 +76,12 @@ func main() {
 	}
 	defer zapLogger.Sync()
 
-	db, err := initDatabase(cfg)
+	db, err := database.Init(&cfg.Database, zapLogger)
 	if err != nil {
 		showError("Nex Gateway", fmt.Sprintf("初始化数据库失败: %v", err))
 		os.Exit(1)
 	}
-	defer closeDB(db)
+	defer database.Close(db)
 
 	providerRepo := repository.NewProviderRepository(db)
 	modelRepo := repository.NewModelRepository(db)
@@ -159,76 +156,6 @@ func main() {
 	setupSystray(port)
 }
 
-func initDatabase(cfg *config.Config) (*gorm.DB, error) {
-	dbDir := filepath.Dir(cfg.Database.Path)
-	if err := os.MkdirAll(dbDir, 0755); err != nil {
-		return nil, fmt.Errorf("创建数据库目录失败: %w", err)
-	}
-
-	db, err := gorm.Open(sqlite.Open(cfg.Database.Path), &gorm.Config{
-		Logger: logger.Default.LogMode(logger.Info),
-	})
-	if err != nil {
-		return nil, err
-	}
-
-	if err := runMigrations(db); err != nil {
-		return nil, fmt.Errorf("数据库迁移失败: %w", err)
-	}
-
-	if err := db.Exec("PRAGMA journal_mode=WAL").Error; err != nil {
-		log.Printf("警告: 启用 WAL 模式失败: %v", err)
-	}
-
-	sqlDB, err := db.DB()
-	if err != nil {
-		return nil, err
-	}
-	sqlDB.SetMaxIdleConns(cfg.Database.MaxIdleConns)
-	sqlDB.SetMaxOpenConns(cfg.Database.MaxOpenConns)
-	sqlDB.SetConnMaxLifetime(cfg.Database.ConnMaxLifetime)
-
-	return db, nil
-}
-
-func runMigrations(db *gorm.DB) error {
-	sqlDB, err := db.DB()
-	if err != nil {
-		return err
-	}
-
-	migrationsDir := getMigrationsDir()
-	if _, err := os.Stat(migrationsDir); os.IsNotExist(err) {
-		return fmt.Errorf("迁移目录不存在: %s", migrationsDir)
-	}
-
-	goose.SetDialect("sqlite3")
-	if err := goose.Up(sqlDB, migrationsDir); err != nil {
-		return err
-	}
-
-	return nil
-}
-
-func getMigrationsDir() string {
-	_, filename, _, ok := runtime.Caller(0)
-	if ok {
-		dir := filepath.Join(filepath.Dir(filename), "..", "..", "migrations")
-		if abs, err := filepath.Abs(dir); err == nil {
-			return abs
-		}
-	}
-	return "./migrations"
-}
-
-func closeDB(db *gorm.DB) {
-	sqlDB, err := db.DB()
-	if err != nil {
-		return
-	}
-	sqlDB.Close()
-}
-
 func setupRoutes(r *gin.Engine, proxyHandler *handler.ProxyHandler, providerHandler *handler.ProviderHandler, modelHandler *handler.ModelHandler, statsHandler *handler.StatsHandler) {
 	r.Any("/v1/*path", proxyHandler.HandleProxy)
 

backend/cmd/server/main.go

@@ -7,22 +7,17 @@ import (
 	"net/http"
 	"os"
 	"os/signal"
-	"path/filepath"
-	"runtime"
 	"syscall"
 	"time"
 
 	"github.com/gin-gonic/gin"
-	"github.com/pressly/goose/v3"
 	"go.uber.org/zap"
-	"gorm.io/driver/sqlite"
-	"gorm.io/gorm"
-	"gorm.io/gorm/logger"
 
 	"nex/backend/internal/config"
 	"nex/backend/internal/conversion"
 	"nex/backend/internal/conversion/anthropic"
 	"nex/backend/internal/conversion/openai"
+	"nex/backend/internal/database"
 	"nex/backend/internal/handler"
 	"nex/backend/internal/handler/middleware"
 	"nex/backend/internal/provider"
@@ -32,16 +27,13 @@ import (
 )
 
 func main() {
-	// 1. 加载配置（已包含 CLI 参数解析、环境变量绑定、配置文件读取和验证）
 	cfg, err := config.LoadConfig()
 	if err != nil {
 		log.Fatalf("加载配置失败: %v", err)
 	}
 
-	// 2. 打印配置摘要
 	cfg.PrintSummary()
 
-	// 3. 初始化日志
 	zapLogger, err := pkgLogger.New(pkgLogger.Config{
 		Level:      cfg.Log.Level,
 		Path:       cfg.Log.Path,
@@ -55,37 +47,31 @@ func main() {
 	}
 	defer zapLogger.Sync()
 
-	// 3. 初始化数据库
-	db, err := initDatabase(cfg)
+	db, err := database.Init(&cfg.Database, zapLogger)
 	if err != nil {
 		zapLogger.Fatal("初始化数据库失败", zap.String("error", err.Error()))
 	}
-	defer closeDB(db)
+	defer database.Close(db)
 
-	// 4. 初始化 repository 层
 	providerRepo := repository.NewProviderRepository(db)
 	modelRepo := repository.NewModelRepository(db)
 	statsRepo := repository.NewStatsRepository(db)
 
-	// 5. 初始化缓存
 	routingCache := service.NewRoutingCache(modelRepo, providerRepo, zapLogger)
 	if err := routingCache.Preload(); err != nil {
 		zapLogger.Warn("缓存预热失败，将使用懒加载", zap.Error(err))
 	}
 
-	// 6. 初始化统计缓冲
 	statsBuffer := service.NewStatsBuffer(statsRepo, zapLogger,
 		service.WithFlushInterval(5*time.Second),
 		service.WithFlushThreshold(100))
 	statsBuffer.Start()
 
-	// 7. 初始化 service 层
 	providerService := service.NewProviderService(providerRepo, modelRepo, routingCache)
 	modelService := service.NewModelService(modelRepo, providerRepo, routingCache)
 	routingService := service.NewRoutingService(routingCache)
 	statsService := service.NewStatsService(statsRepo, statsBuffer)
 
-	// 8. 创建 ConversionEngine
 	registry := conversion.NewMemoryRegistry()
 	if err := registry.Register(openai.NewAdapter()); err != nil {
 		zapLogger.Fatal("注册 OpenAI 适配器失败", zap.String("error", err.Error()))
@@ -95,16 +81,13 @@ func main() {
 	}
 	engine := conversion.NewConversionEngine(registry, zapLogger)
 
-	// 9. 初始化 provider client
 	providerClient := provider.NewClient()
 
-	// 10. 初始化 handler 层
 	proxyHandler := handler.NewProxyHandler(engine, providerClient, routingService, providerService, statsService)
 	providerHandler := handler.NewProviderHandler(providerService)
 	modelHandler := handler.NewModelHandler(modelService)
 	statsHandler := handler.NewStatsHandler(statsService)
 
-	// 11. 创建 Gin 引擎
 	gin.SetMode(gin.ReleaseMode)
 	r := gin.New()
 
@@ -115,9 +98,8 @@ func main() {
 
 	setupRoutes(r, proxyHandler, providerHandler, modelHandler, statsHandler)
 
-	// 12. 启动服务器
 	srv := &http.Server{
-		Addr:         formatAddr(cfg.Server.Port),
+		Addr:         fmt.Sprintf(":%d", cfg.Server.Port),
 		Handler:      r,
 		ReadTimeout:  cfg.Server.ReadTimeout,
 		WriteTimeout: cfg.Server.WriteTimeout,
@@ -148,83 +130,9 @@ func main() {
 	zapLogger.Info("服务器已关闭")
 }
 
-func initDatabase(cfg *config.Config) (*gorm.DB, error) {
-	db, err := gorm.Open(sqlite.Open(cfg.Database.Path), &gorm.Config{
-		Logger: logger.Default.LogMode(logger.Info),
-	})
-	if err != nil {
-		return nil, err
-	}
-
-	if err := runMigrations(db); err != nil {
-		return nil, fmt.Errorf("数据库迁移失败: %w", err)
-	}
-
-	if err := db.Exec("PRAGMA journal_mode=WAL").Error; err != nil {
-		log.Printf("警告: 启用 WAL 模式失败: %v", err)
-	}
-
-	sqlDB, err := db.DB()
-	if err != nil {
-		return nil, err
-	}
-	sqlDB.SetMaxIdleConns(cfg.Database.MaxIdleConns)
-	sqlDB.SetMaxOpenConns(cfg.Database.MaxOpenConns)
-	sqlDB.SetConnMaxLifetime(cfg.Database.ConnMaxLifetime)
-
-	log.Printf("数据库连接池配置: MaxIdle=%d, MaxOpen=%d, MaxLifetime=%v",
-		cfg.Database.MaxIdleConns, cfg.Database.MaxOpenConns, cfg.Database.ConnMaxLifetime)
-
-	return db, nil
-}
-
-func runMigrations(db *gorm.DB) error {
-	sqlDB, err := db.DB()
-	if err != nil {
-		return err
-	}
-
-	migrationsDir := getMigrationsDir()
-	if _, err := os.Stat(migrationsDir); os.IsNotExist(err) {
-		return fmt.Errorf("迁移目录不存在: %s", migrationsDir)
-	}
-
-	goose.SetDialect("sqlite3")
-	if err := goose.Up(sqlDB, migrationsDir); err != nil {
-		return err
-	}
-
-	return nil
-}
-
-func getMigrationsDir() string {
-	_, filename, _, ok := runtime.Caller(0)
-	if ok {
-		dir := filepath.Join(filepath.Dir(filename), "..", "..", "migrations")
-		if abs, err := filepath.Abs(dir); err == nil {
-			return abs
-		}
-	}
-	return "./migrations"
-}
-
-func closeDB(db *gorm.DB) {
-	sqlDB, err := db.DB()
-	if err != nil {
-		return
-	}
-	sqlDB.Close()
-}
-
-func formatAddr(port int) string {
-	return fmt.Sprintf(":%d", port)
-}
-
 func setupRoutes(r *gin.Engine, proxyHandler *handler.ProxyHandler, providerHandler *handler.ProviderHandler, modelHandler *handler.ModelHandler, statsHandler *handler.StatsHandler) {
-	// 统一代理入口: /{protocol}/{path}
 	r.Any("/:protocol/*path", proxyHandler.HandleProxy)
 
-	// 供应商管理 API
 	providers := r.Group("/api/providers")
 	{
 		providers.GET("", providerHandler.ListProviders)
@@ -234,7 +142,6 @@ func setupRoutes(r *gin.Engine, proxyHandler *handler.ProxyHandler, providerHand
 		providers.DELETE("/:id", providerHandler.DeleteProvider)
 	}
 
-	// 模型管理 API
 	models := r.Group("/api/models")
 	{
 		models.GET("", modelHandler.ListModels)
@@ -244,14 +151,12 @@ func setupRoutes(r *gin.Engine, proxyHandler *handler.ProxyHandler, providerHand
 		models.DELETE("/:id", modelHandler.DeleteModel)
 	}
 
-	// 统计查询 API
 	stats := r.Group("/api/stats")
 	{
 		stats.GET("", statsHandler.GetStats)
 		stats.GET("/aggregate", statsHandler.AggregateStats)
 	}
 
-	// 健康检查
 	r.GET("/health", func(c *gin.Context) {
 		c.JSON(200, gin.H{"status": "ok"})
 	})