New

internal/db/migrations/migrations.go

@@ -0,0 +1,36 @@
+// Filename: internal/db/migrations/migrations.go (全新)
+package migrations
+
+import (
+	"gemini-balancer/internal/models"
+
+	"github.com/sirupsen/logrus"
+	"gorm.io/gorm"
+)
+
+// RunMigrations 负责执行所有的数据库模式迁移。
+func RunMigrations(db *gorm.DB, logger *logrus.Logger) error {
+	log := logger.WithField("component", "migrations")
+	log.Info("Running database schema migrations...")
+	// 集中管理所有需要被创建或更新的表。
+	err := db.AutoMigrate(
+		&models.UpstreamEndpoint{},
+		&models.ProxyConfig{},
+		&models.APIKey{},
+		&models.KeyGroup{},
+		&models.GroupModelMapping{},
+		&models.AuthToken{},
+		&models.RequestLog{},
+		&models.StatsHourly{},
+		&models.FileRecord{},
+		&models.Setting{},
+		&models.GroupSettings{},
+		&models.GroupAPIKeyMapping{},
+	)
+	if err != nil {
+		log.Errorf("Database schema migration failed: %v", err)
+		return err
+	}
+	log.Info("Database schema migrations completed successfully.")
+	return nil
+}

internal/db/migrations/versioned_migrations.go

@@ -0,0 +1,62 @@
+// Filename: internal/db/migrations/versioned_migrations.go
+
+package migrations
+
+import (
+	"fmt"
+	"gemini-balancer/internal/config"
+
+	"github.com/sirupsen/logrus"
+	"gorm.io/gorm"
+)
+
+// RunVersionedMigrations 负责运行所有已注册的版本化迁移。
+func RunVersionedMigrations(db *gorm.DB, cfg *config.Config, logger *logrus.Logger) error {
+	log := logger.WithField("component", "versioned_migrations")
+	log.Info("Checking for versioned database migrations...")
+
+	if err := db.AutoMigrate(&MigrationHistory{}); err != nil {
+		log.Errorf("Failed to create migration history table: %v", err)
+		return err
+	}
+
+	var executedMigrations []MigrationHistory
+	db.Find(&executedMigrations)
+	executedVersions := make(map[string]bool)
+	for _, m := range executedMigrations {
+		executedVersions[m.Version] = true
+	}
+
+	for _, migration := range migrationRegistry {
+		if !executedVersions[migration.Version] {
+			log.Infof("Running migration %s: %s", migration.Version, migration.Description)
+			if err := migration.Migrate(db, cfg, log); err != nil {
+				log.Errorf("Migration %s failed: %v", migration.Version, err)
+				return fmt.Errorf("migration %s failed: %w", migration.Version, err)
+			}
+			db.Create(&MigrationHistory{Version: migration.Version})
+			log.Infof("Migration %s completed successfully.", migration.Version)
+		}
+	}
+
+	log.Info("All versioned migrations are up to date.")
+	return nil
+}
+
+type MigrationFunc func(db *gorm.DB, cfg *config.Config, logger *logrus.Entry) error
+type VersionedMigration struct {
+	Version     string
+	Description string
+	Migrate     MigrationFunc
+}
+type MigrationHistory struct {
+	Version string `gorm:"primaryKey"`
+}
+
+var migrationRegistry = []VersionedMigration{
+	/*{
+		Version:     "20250828_encrypt_existing_auth_tokens",
+		Description: "Encrypt plaintext tokens and populate new crypto columns in auth_tokens table.",
+		Migrate:     MigrateAuthTokenEncryption,
+	},*/
+}