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

2026-04-23 00:43:23 +08:00
parent 15f08ee2ca
commit 5b765c8b5e
17 changed files with 626 additions and 205 deletions
--- a/backend/README.md
+++ b/backend/README.md
@@ -24,7 +24,7 @@ AI 网关后端服务，提供统一的大模型 API 代理接口。
 - **语言**: Go 1.26+
 - **HTTP 框架**: Gin
 - **ORM**: GORM
- **数据库**: SQLite
+- **数据库**: SQLite / MySQL
 - **日志**: zap + lumberjack
 - **配置**: Viper + pflag（多层配置：CLI > 环境变量 > 配置文件 > 默认值）
 - **验证**: go-playground/validator/v10
@@ -294,7 +294,14 @@ server:
  write_timeout: 30s

 database:
-  path: ~/.nex/config.db
+  driver: sqlite               # sqlite 或 mysql
+  path: ~/.nex/config.db       # SQLite 数据库文件路径
+  # --- MySQL 配置（driver=mysql 时生效）---
+  # host: localhost
+  # port: 3306
+  # user: nex
+  # password: ""
+  # dbname: nex
  max_idle_conns: 10
  max_open_conns: 100
  conn_max_lifetime: 1h
@@ -316,6 +323,14 @@ log:
 export NEX_SERVER_PORT=9000
 export NEX_DATABASE_PATH=/data/nex.db
 export NEX_LOG_LEVEL=debug
+
+# MySQL 模式
+export NEX_DATABASE_DRIVER=mysql
+export NEX_DATABASE_HOST=db.example.com
+export NEX_DATABASE_PORT=3306
+export NEX_DATABASE_USER=nex
+export NEX_DATABASE_PASSWORD=secret
+export NEX_DATABASE_DBNAME=nex
 ```

 命名规则：配置路径转大写 + 下划线 + `NEX_` 前缀（如 `server.port` → `NEX_SERVER_PORT`）。
@@ -332,7 +347,7 @@ export NEX_LOG_LEVEL=debug

 ```
 服务器:   --server-port, --server-read-timeout, --server-write-timeout
-数据库:   --database-path, --database-max-idle-conns, --database-max-open-conns, --database-conn-max-lifetime
+数据库:   --database-driver, --database-path, --database-host, --database-port, --database-user, --database-password, --database-dbname, --database-max-idle-conns, --database-max-open-conns, --database-conn-max-lifetime
 日志:     --log-level, --log-path, --log-max-size, --log-max-backups, --log-max-age, --log-compress
 通用:     --config (指定配置文件路径)
 ```
@@ -352,15 +367,20 @@ export NEX_LOG_LEVEL=debug
 # Docker 部署
 docker run -d -e NEX_SERVER_PORT=9000 -e NEX_LOG_LEVEL=info nex-server

+# MySQL 模式
+./server --database-driver mysql --database-host db.example.com --database-user nex --database-password secret --database-dbname nex
+
 # 自定义配置文件
 ./server --config /path/to/custom.yaml
 ```

 数据文件：
 - `~/.nex/config.yaml` - 配置文件
- `~/.nex/config.db` - SQLite 数据库
+- `~/.nex/config.db` - SQLite 数据库（MySQL 模式下不使用本地数据库文件）
 - `~/.nex/log/` - 日志目录

+**MySQL 连接说明**：MySQL 连接使用 DSN 格式: `user:password@tcp(host:port)/dbname?charset=utf8mb4&parseTime=true&loc=Local`，最低支持 MySQL 8.0+。
+
 ## 测试

 ```bash
--- a/backend/cmd/desktop/main.go
+++ b/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)

--- a/backend/cmd/server/main.go
+++ b/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"})
 	})
--- a/backend/go.mod
+++ b/backend/go.mod
@@ -24,6 +24,7 @@ require (
 	go.uber.org/zap v1.27.1
 	gopkg.in/lumberjack.v2 v2.0.0
 	gopkg.in/yaml.v3 v3.0.1
+	gorm.io/driver/mysql v1.6.0
 	gorm.io/driver/sqlite v1.6.0
 	gorm.io/gorm v1.31.1
 	nex/embedfs v0.0.0-00010101000000-000000000000
@@ -32,6 +33,7 @@ require (
 require (
 	4d63.com/gocheckcompilerdirectives v1.3.0 // indirect
 	4d63.com/gochecknoglobals v0.2.2 // indirect
+	filippo.io/edwards25519 v1.2.0 // indirect
 	github.com/4meepo/tagalign v1.4.2 // indirect
 	github.com/Abirdcfly/dupword v0.1.3 // indirect
 	github.com/Antonboom/errname v1.0.0 // indirect
@@ -90,6 +92,7 @@ require (
 	github.com/go-critic/go-critic v0.12.0 // indirect
 	github.com/go-playground/locales v0.14.1 // indirect
 	github.com/go-playground/universal-translator v0.18.1 // indirect
+	github.com/go-sql-driver/mysql v1.9.3 // indirect
 	github.com/go-stack/stack v1.8.0 // indirect
 	github.com/go-toolsmith/astcast v1.1.0 // indirect
 	github.com/go-toolsmith/astcopy v1.1.0 // indirect
--- a/backend/go.sum
+++ b/backend/go.sum
@@ -35,6 +35,8 @@ cloud.google.com/go/storage v1.6.0/go.mod h1:N7U0C8pVQ/+NIKOBQyamJIeKQKkZ+mxpohl
 cloud.google.com/go/storage v1.8.0/go.mod h1:Wv1Oy7z6Yz3DshWRJFhqM/UCfaWIRTdp0RXyy7KQOVs=
 cloud.google.com/go/storage v1.10.0/go.mod h1:FLPqc6j+Ki4BU591ie1oL6qBQGu2Bl/tZ9ullr3+Kg0=
 dmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9/go.mod h1:H6x//7gZCb22OMCxBHrMx7a5I7Hp++hsVxbQ4BYO7hU=
+filippo.io/edwards25519 v1.2.0 h1:crnVqOiS4jqYleHd9vaKZ+HKtHfllngJIiOpNpoJsjo=
+filippo.io/edwards25519 v1.2.0/go.mod h1:xzAOLCNug/yB62zG1bQ8uziwrIqIuxhctzJT18Q77mc=
 github.com/4meepo/tagalign v1.4.2 h1:0hcLHPGMjDyM1gHG58cS73aQF8J4TdVR96TZViorO9E=
 github.com/4meepo/tagalign v1.4.2/go.mod h1:+p4aMyFM+ra7nb41CnFG6aSDXqRxU/w1VQqScKqDARI=
 github.com/Abirdcfly/dupword v0.1.3 h1:9Pa1NuAsZvpFPi9Pqkd93I7LIYRURj+A//dFd5tgBeE=
@@ -206,6 +208,8 @@ github.com/go-playground/validator/v10 v10.30.2 h1:JiFIMtSSHb2/XBUbWM4i/MpeQm9ZK
 github.com/go-playground/validator/v10 v10.30.2/go.mod h1:mAf2pIOVXjTEBrwUMGKkCWKKPs9NheYGabeB04txQSc=
 github.com/go-quicktest/qt v1.101.0 h1:O1K29Txy5P2OK0dGo59b7b0LR6wKfIhttaAhHUyn7eI=
 github.com/go-quicktest/qt v1.101.0/go.mod h1:14Bz/f7NwaXPtdYEgzsx46kqSxVwTbzVZsDC26tQJow=
+github.com/go-sql-driver/mysql v1.9.3 h1:U/N249h2WzJ3Ukj8SowVFjdtZKfu9vlLZxjPXV1aweo=
+github.com/go-sql-driver/mysql v1.9.3/go.mod h1:qn46aNg1333BRMNU69Lq93t8du/dwxI64Gl8i5p1WMU=
 github.com/go-stack/stack v1.8.0 h1:5SgMzNM5HxrEjV0ww2lTmX6E2Izsfxas4+YHWRs3Lsk=
 github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY=
 github.com/go-task/slim-sprig/v3 v3.0.0 h1:sUs3vkvUymDpBKi3qH1YSqBQk9+9D/8M2mN1vB6EwHI=
@@ -1052,6 +1056,8 @@ gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
 gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gorm.io/driver/mysql v1.6.0 h1:eNbLmNTpPpTOVZi8MMxCi2aaIm0ZpInbORNXDwyLGvg=
+gorm.io/driver/mysql v1.6.0/go.mod h1:D/oCC2GWK3M/dqoLxnOlaNKmXz8WNTfcS9y5ovaSqKo=
 gorm.io/driver/sqlite v1.6.0 h1:WHRRrIiulaPiPFmDcod6prc4l2VGVWHz80KspNsxSfQ=
 gorm.io/driver/sqlite v1.6.0/go.mod h1:AO9V1qIQddBESngQUKWL9yoH93HIeA1X6V633rBwyT8=
 gorm.io/gorm v1.31.1 h1:7CA8FTFz/gRfgqgpeKIBcervUn3xSyPUmr6B2WXJ7kg=
--- a/backend/internal/config/config.go
+++ b/backend/internal/config/config.go
@@ -32,7 +32,13 @@ type ServerConfig struct {

 // DatabaseConfig 数据库配置
 type DatabaseConfig struct {
-	Path            string        `yaml:"path" mapstructure:"path" validate:"required"`
+	Driver          string        `yaml:"driver" mapstructure:"driver" validate:"required,oneof=sqlite mysql"`
+	Path            string        `yaml:"path" mapstructure:"path" validate:"required_if=Driver sqlite"`
+	Host            string        `yaml:"host" mapstructure:"host" validate:"required_if=Driver mysql"`
+	Port            int           `yaml:"port" mapstructure:"port" validate:"required_if=Driver mysql,min=1,max=65535"`
+	User            string        `yaml:"user" mapstructure:"user" validate:"required_if=Driver mysql"`
+	Password        string        `yaml:"password" mapstructure:"password"`
+	DBName          string        `yaml:"dbname" mapstructure:"dbname" validate:"required_if=Driver mysql"`
 	MaxIdleConns    int           `yaml:"max_idle_conns" mapstructure:"max_idle_conns" validate:"required,min=1"`
 	MaxOpenConns    int           `yaml:"max_open_conns" mapstructure:"max_open_conns" validate:"required,min=1"`
 	ConnMaxLifetime time.Duration `yaml:"conn_max_lifetime" mapstructure:"conn_max_lifetime" validate:"required"`
@@ -61,7 +67,13 @@ func DefaultConfig() *Config {
 			WriteTimeout: 30 * time.Second,
 		},
 		Database: DatabaseConfig{
+			Driver:          "sqlite",
 			Path:            filepath.Join(nexDir, "config.db"),
+			Host:            "",
+			Port:            3306,
+			User:            "",
+			Password:        "",
+			DBName:          "nex",
 			MaxIdleConns:    10,
 			MaxOpenConns:    100,
 			ConnMaxLifetime: 1 * time.Hour,
@@ -117,7 +129,13 @@ func setupDefaults(v *viper.Viper) {
 	v.SetDefault("server.read_timeout", "30s")
 	v.SetDefault("server.write_timeout", "30s")

+	v.SetDefault("database.driver", "sqlite")
 	v.SetDefault("database.path", filepath.Join(nexDir, "config.db"))
+	v.SetDefault("database.host", "")
+	v.SetDefault("database.port", 3306)
+	v.SetDefault("database.user", "")
+	v.SetDefault("database.password", "")
+	v.SetDefault("database.dbname", "nex")
 	v.SetDefault("database.max_idle_conns", 10)
 	v.SetDefault("database.max_open_conns", 100)
 	v.SetDefault("database.conn_max_lifetime", "1h")
@@ -138,7 +156,13 @@ func setupFlags(v *viper.Viper, flagSet *pflag.FlagSet) {
 	flagSet.Duration("server-read-timeout", 0, "读超时")
 	flagSet.Duration("server-write-timeout", 0, "写超时")

+	flagSet.String("database-driver", "", "数据库驱动：sqlite/mysql")
 	flagSet.String("database-path", "", "数据库文件路径")
+	flagSet.String("database-host", "", "MySQL 主机地址")
+	flagSet.Int("database-port", 0, "MySQL 端口")
+	flagSet.String("database-user", "", "MySQL 用户名")
+	flagSet.String("database-password", "", "MySQL 密码")
+	flagSet.String("database-dbname", "", "MySQL 数据库名")
 	flagSet.Int("database-max-idle-conns", 0, "最大空闲连接数")
 	flagSet.Int("database-max-open-conns", 0, "最大打开连接数")
 	flagSet.Duration("database-conn-max-lifetime", 0, "连接最大生命周期")
@@ -156,7 +180,13 @@ func setupFlags(v *viper.Viper, flagSet *pflag.FlagSet) {
 	v.BindPFlag("server.read_timeout", flagSet.Lookup("server-read-timeout"))
 	v.BindPFlag("server.write_timeout", flagSet.Lookup("server-write-timeout"))

+	v.BindPFlag("database.driver", flagSet.Lookup("database-driver"))
 	v.BindPFlag("database.path", flagSet.Lookup("database-path"))
+	v.BindPFlag("database.host", flagSet.Lookup("database-host"))
+	v.BindPFlag("database.port", flagSet.Lookup("database-port"))
+	v.BindPFlag("database.user", flagSet.Lookup("database-user"))
+	v.BindPFlag("database.password", flagSet.Lookup("database-password"))
+	v.BindPFlag("database.dbname", flagSet.Lookup("database-dbname"))
 	v.BindPFlag("database.max_idle_conns", flagSet.Lookup("database-max-idle-conns"))
 	v.BindPFlag("database.max_open_conns", flagSet.Lookup("database-max-open-conns"))
 	v.BindPFlag("database.conn_max_lifetime", flagSet.Lookup("database-conn-max-lifetime"))
@@ -268,7 +298,7 @@ func SaveConfig(cfg *Config) error {
 		return appErrors.Wrap(appErrors.ErrInternal, err)
 	}

-	return os.WriteFile(configPath, data, 0644)
+	return os.WriteFile(configPath, data, 0600)
 }

 // Validate validates the config
@@ -285,7 +315,13 @@ func (c *Config) PrintSummary() {
 	fmt.Println("\nAI Gateway 启动配置")
 	fmt.Println("==================")
 	fmt.Printf("服务器端口:     %d\n", c.Server.Port)
-	fmt.Printf("数据库路径:     %s\n", c.Database.Path)
+	if c.Database.Driver == "mysql" {
+		fmt.Printf("数据库类型:     mysql\n")
+		fmt.Printf("数据库地址:     %s:%d/%s\n", c.Database.Host, c.Database.Port, c.Database.DBName)
+	} else {
+		fmt.Printf("数据库类型:     sqlite\n")
+		fmt.Printf("数据库路径:     %s\n", c.Database.Path)
+	}
 	fmt.Printf("日志级别:       %s\n", c.Log.Level)
 	fmt.Println("\n配置来源:")
 	configPath, _ := GetConfigPath()
--- a/backend/internal/config/config_test.go
+++ b/backend/internal/config/config_test.go
@@ -19,6 +19,12 @@ func TestDefaultConfig(t *testing.T) {
 	assert.Equal(t, 30*time.Second, cfg.Server.ReadTimeout)
 	assert.Equal(t, 30*time.Second, cfg.Server.WriteTimeout)

+	assert.Equal(t, "sqlite", cfg.Database.Driver)
+	assert.Equal(t, "", cfg.Database.Host)
+	assert.Equal(t, 3306, cfg.Database.Port)
+	assert.Equal(t, "", cfg.Database.User)
+	assert.Equal(t, "", cfg.Database.Password)
+	assert.Equal(t, "nex", cfg.Database.DBName)
 	assert.Equal(t, 10, cfg.Database.MaxIdleConns)
 	assert.Equal(t, 100, cfg.Database.MaxOpenConns)
 	assert.Equal(t, 1*time.Hour, cfg.Database.ConnMaxLifetime)
@@ -86,11 +92,76 @@ func TestConfig_Validate(t *testing.T) {
 			wantErr: false,
 		},
 		{
-			name:    "数据库路径为空无效",
+			name:    "SQLite模式路径为空无效",
 			modify:  func(c *Config) { c.Database.Path = "" },
 			wantErr: true,
 			errMsg:  "配置验证失败",
 		},
+		{
+			name:    "driver值不合法",
+			modify:  func(c *Config) { c.Database.Driver = "postgres" },
+			wantErr: true,
+			errMsg:  "配置验证失败",
+		},
+		{
+			name: "MySQL配置有效",
+			modify: func(c *Config) {
+				c.Database.Driver = "mysql"
+				c.Database.Host = "localhost"
+				c.Database.Port = 3306
+				c.Database.User = "root"
+				c.Database.DBName = "nex"
+				c.Database.Path = ""
+			},
+			wantErr: false,
+		},
+		{
+			name: "MySQL模式host为空无效",
+			modify: func(c *Config) {
+				c.Database.Driver = "mysql"
+				c.Database.Host = ""
+				c.Database.User = "root"
+				c.Database.DBName = "nex"
+				c.Database.Path = ""
+			},
+			wantErr: true,
+			errMsg:  "配置验证失败",
+		},
+		{
+			name: "MySQL模式user为空无效",
+			modify: func(c *Config) {
+				c.Database.Driver = "mysql"
+				c.Database.Host = "localhost"
+				c.Database.User = ""
+				c.Database.DBName = "nex"
+				c.Database.Path = ""
+			},
+			wantErr: true,
+			errMsg:  "配置验证失败",
+		},
+		{
+			name: "MySQL模式dbname为空无效",
+			modify: func(c *Config) {
+				c.Database.Driver = "mysql"
+				c.Database.Host = "localhost"
+				c.Database.User = "root"
+				c.Database.DBName = ""
+				c.Database.Path = ""
+			},
+			wantErr: true,
+			errMsg:  "配置验证失败",
+		},
+		{
+			name: "MySQL模式忽略path字段",
+			modify: func(c *Config) {
+				c.Database.Driver = "mysql"
+				c.Database.Host = "localhost"
+				c.Database.User = "root"
+				c.Database.DBName = "nex"
+				c.Database.Path = ""
+			},
+			wantErr: false,
+		},
 	}

 	for _, tt := range tests {
@@ -140,7 +211,10 @@ func TestSaveAndLoadConfig(t *testing.T) {
 			WriteTimeout: 20 * time.Second,
 		},
 		Database: DatabaseConfig{
+			Driver:          "sqlite",
 			Path:            filepath.Join(dir, "test.db"),
+			Port:            3306,
+			DBName:          "nex",
 			MaxIdleConns:    5,
 			MaxOpenConns:    50,
 			ConnMaxLifetime: 30 * time.Minute,
@@ -210,6 +284,9 @@ func TestConfigPriority(t *testing.T) {
 		assert.Equal(t, 9826, cfg.Server.Port)
 		assert.Equal(t, 30*time.Second, cfg.Server.ReadTimeout)
 		assert.Equal(t, 30*time.Second, cfg.Server.WriteTimeout)
+		assert.Equal(t, "sqlite", cfg.Database.Driver)
+		assert.Equal(t, 3306, cfg.Database.Port)
+		assert.Equal(t, "nex", cfg.Database.DBName)
 		assert.Equal(t, 10, cfg.Database.MaxIdleConns)
 		assert.Equal(t, 100, cfg.Database.MaxOpenConns)
 		assert.Equal(t, 1*time.Hour, cfg.Database.ConnMaxLifetime)
@@ -222,11 +299,19 @@ func TestConfigPriority(t *testing.T) {
 }

 func TestPrintSummary(t *testing.T) {
-	// 测试配置摘要输出
-	t.Run("打印配置摘要", func(t *testing.T) {
+	t.Run("SQLite模式摘要", func(t *testing.T) {
 		cfg := DefaultConfig()
-		// PrintSummary 只是打印，不会返回错误
-		// 这里主要验证不会 panic
+		assert.NotPanics(t, func() {
+			cfg.PrintSummary()
+		})
+	})
+	t.Run("MySQL模式摘要", func(t *testing.T) {
+		cfg := DefaultConfig()
+		cfg.Database.Driver = "mysql"
+		cfg.Database.Host = "db.example.com"
+		cfg.Database.Port = 3306
+		cfg.Database.User = "nex"
+		cfg.Database.DBName = "nex"
 		assert.NotPanics(t, func() {
 			cfg.PrintSummary()
 		})
--- a/backend/internal/database/database.go
+++ b/backend/internal/database/database.go
@@ -0,0 +1,126 @@
+package database
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"path/filepath"
+	"runtime"
+
+	"github.com/pressly/goose/v3"
+	"go.uber.org/zap"
+	"gorm.io/driver/mysql"
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+
+	"nex/backend/internal/config"
+)
+
+func Init(cfg *config.DatabaseConfig, zapLogger *zap.Logger) (*gorm.DB, error) {
+	db, err := initDB(cfg)
+	if err != nil {
+		return nil, fmt.Errorf("初始化数据库失败: %w", err)
+	}
+
+	if err := runMigrations(db, cfg.Driver); err != nil {
+		return nil, fmt.Errorf("数据库迁移失败: %w", err)
+	}
+
+	configurePool(db, cfg)
+
+	return db, nil
+}
+
+func Close(db *gorm.DB) {
+	sqlDB, err := db.DB()
+	if err != nil {
+		return
+	}
+	sqlDB.Close()
+}
+
+func initDB(cfg *config.DatabaseConfig) (*gorm.DB, error) {
+	gormConfig := &gorm.Config{
+		Logger: logger.Default.LogMode(logger.Info),
+	}
+
+	switch cfg.Driver {
+	case "mysql":
+		dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=true&loc=Local",
+			cfg.User, cfg.Password, cfg.Host, cfg.Port, cfg.DBName)
+		return gorm.Open(mysql.Open(dsn), gormConfig)
+	default:
+		dbDir := filepath.Dir(cfg.Path)
+		if err := os.MkdirAll(dbDir, 0755); err != nil {
+			return nil, fmt.Errorf("创建数据库目录失败: %w", err)
+		}
+		return gorm.Open(sqlite.Open(cfg.Path), gormConfig)
+	}
+}
+
+func runMigrations(db *gorm.DB, driver string) error {
+	sqlDB, err := db.DB()
+	if err != nil {
+		return err
+	}
+
+	migrationsDir := getMigrationsDir(driver)
+	if _, err := os.Stat(migrationsDir); os.IsNotExist(err) {
+		return fmt.Errorf("迁移目录不存在: %s", migrationsDir)
+	}
+
+	gooseDialect := "sqlite3"
+	migrationsSubDir := "sqlite"
+	if driver == "mysql" {
+		gooseDialect = "mysql"
+		migrationsSubDir = "mysql"
+	}
+
+	goose.SetDialect(gooseDialect)
+	if err := goose.Up(sqlDB, migrationsDir); err != nil {
+		return err
+	}
+
+	log.Printf("使用 %s 方言执行迁移，目录: %s", gooseDialect, migrationsSubDir)
+	return nil
+}
+
+func configurePool(db *gorm.DB, cfg *config.DatabaseConfig) {
+	if cfg.Driver == "sqlite" {
+		if err := db.Exec("PRAGMA journal_mode=WAL").Error; err != nil {
+			log.Printf("警告: 启用 WAL 模式失败: %v", err)
+		}
+	}
+
+	sqlDB, err := db.DB()
+	if err != nil {
+		return
+	}
+	sqlDB.SetMaxIdleConns(cfg.MaxIdleConns)
+	sqlDB.SetMaxOpenConns(cfg.MaxOpenConns)
+	sqlDB.SetConnMaxLifetime(cfg.ConnMaxLifetime)
+
+	log.Printf("数据库连接池配置: MaxIdle=%d, MaxOpen=%d, MaxLifetime=%v",
+		cfg.MaxIdleConns, cfg.MaxOpenConns, cfg.ConnMaxLifetime)
+}
+
+func getMigrationsDir(driver string) string {
+	_, filename, _, ok := runtime.Caller(0)
+	if ok {
+		subDir := "sqlite"
+		if driver == "mysql" {
+			subDir = "mysql"
+		}
+		dir := filepath.Join(filepath.Dir(filename), "..", "..", "migrations", subDir)
+		if abs, err := filepath.Abs(dir); err == nil {
+			return abs
+		}
+	}
+	return "./migrations"
+}
+
+func BuildDSN(cfg *config.DatabaseConfig) string {
+	return fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=true&loc=Local",
+		cfg.User, cfg.Password, cfg.Host, cfg.Port, cfg.DBName)
+}
--- a/backend/internal/database/database_test.go
+++ b/backend/internal/database/database_test.go
@@ -0,0 +1,75 @@
+package database
+
+import (
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"nex/backend/internal/config"
+)
+
+func TestInit_SQLite(t *testing.T) {
+	dir := t.TempDir()
+	cfg := &config.DatabaseConfig{
+		Driver:          "sqlite",
+		Path:            filepath.Join(dir, "test.db"),
+		MaxIdleConns:    5,
+		MaxOpenConns:    10,
+		ConnMaxLifetime: 0,
+	}
+
+	db, err := Init(cfg, nil)
+	require.NoError(t, err)
+	require.NotNil(t, db)
+	defer Close(db)
+
+	sqlDB, err := db.DB()
+	require.NoError(t, err)
+	require.NotNil(t, sqlDB)
+}
+
+func TestClose(t *testing.T) {
+	dir := t.TempDir()
+	cfg := &config.DatabaseConfig{
+		Driver:          "sqlite",
+		Path:            filepath.Join(dir, "test.db"),
+		MaxIdleConns:    5,
+		MaxOpenConns:    10,
+		ConnMaxLifetime: 0,
+	}
+
+	db, err := Init(cfg, nil)
+	require.NoError(t, err)
+	require.NotNil(t, db)
+
+	Close(db)
+}
+
+func TestBuildDSN(t *testing.T) {
+	cfg := &config.DatabaseConfig{
+		Driver:   "mysql",
+		Host:     "db.example.com",
+		Port:     3306,
+		User:     "nexuser",
+		Password: "secretpass",
+		DBName:   "nexdb",
+	}
+
+	dsn := BuildDSN(cfg)
+	assert.Equal(t, "nexuser:secretpass@tcp(db.example.com:3306)/nexdb?charset=utf8mb4&parseTime=true&loc=Local", dsn)
+}
+
+func TestBuildDSN_EmptyPassword(t *testing.T) {
+	cfg := &config.DatabaseConfig{
+		Driver: "mysql",
+		Host:   "localhost",
+		Port:   3306,
+		User:   "root",
+		DBName: "nex",
+	}
+
+	dsn := BuildDSN(cfg)
+	assert.Equal(t, "root:@tcp(localhost:3306)/nex?charset=utf8mb4&parseTime=true&loc=Local", dsn)
+}
--- a/backend/migrations/mysql/20260421000001_initial_schema.sql
+++ b/backend/migrations/mysql/20260421000001_initial_schema.sql
@@ -0,0 +1,44 @@
+-- +goose Up
+-- MySQL 方言初始迁移：providers、models、usage_stats 完整表结构
+
+CREATE TABLE IF NOT EXISTS providers (
+    id VARCHAR(36) PRIMARY KEY,
+    name VARCHAR(255) NOT NULL,
+    api_key VARCHAR(255) NOT NULL,
+    base_url VARCHAR(255) NOT NULL,
+    protocol VARCHAR(50) DEFAULT 'openai',
+    enabled BOOLEAN DEFAULT TRUE,
+    created_at DATETIME(3),
+    updated_at DATETIME(3)
+);
+
+CREATE TABLE IF NOT EXISTS models (
+    id VARCHAR(36) PRIMARY KEY,
+    provider_id VARCHAR(36) NOT NULL,
+    model_name VARCHAR(255) NOT NULL,
+    enabled BOOLEAN DEFAULT TRUE,
+    created_at DATETIME(3),
+    FOREIGN KEY (provider_id) REFERENCES providers(id) ON DELETE CASCADE,
+    UNIQUE(provider_id, model_name)
+);
+
+CREATE TABLE IF NOT EXISTS usage_stats (
+    id INT UNSIGNED AUTO_INCREMENT PRIMARY KEY,
+    provider_id VARCHAR(36) NOT NULL,
+    model_name VARCHAR(255) NOT NULL,
+    request_count INT DEFAULT 0,
+    date DATE NOT NULL,
+    UNIQUE(provider_id, model_name, date)
+);
+
+CREATE INDEX idx_models_provider_id ON models(provider_id);
+CREATE INDEX idx_models_model_name ON models(model_name);
+CREATE INDEX idx_usage_stats_provider_model_date ON usage_stats(provider_id, model_name, date);
+
+-- +goose Down
+DROP INDEX IF EXISTS idx_usage_stats_provider_model_date ON usage_stats;
+DROP INDEX IF EXISTS idx_models_model_name ON models;
+DROP INDEX IF EXISTS idx_models_provider_id ON models;
+DROP TABLE IF EXISTS usage_stats;
+DROP TABLE IF EXISTS models;
+DROP TABLE IF EXISTS providers;
--- a/backend/migrations/sqlite/20260421000001_initial_schema.sql
+++ b/backend/migrations/sqlite/20260421000001_initial_schema.sql
--- a/backend/tests/config/config_test.go
+++ b/backend/tests/config/config_test.go
@@ -158,7 +158,10 @@ func TestSaveAndLoadConfig(t *testing.T) {
 			WriteTimeout: 45 * time.Second,
 		},
 		Database: config.DatabaseConfig{
+			Driver:          "sqlite",
 			Path:            filepath.Join(tmpDir, "test.db"),
+			Port:            3306,
+			DBName:          "nex",
 			MaxIdleConns:    15,
 			MaxOpenConns:    150,
 			ConnMaxLifetime: 2 * time.Hour,