feat: 新增 MySQL 数据库驱动支持，支持跨设备数据同步

backend/cmd/desktop/main.go

@@ -17,16 +17,13 @@ import (
 	"github.com/gin-gonic/gin"
 	"github.com/getlantern/systray"
 	"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)