From ab0ba9f38cc1288ef0427a27416aace6df816bd4 Mon Sep 17 00:00:00 2001
From: CaIon <1808837298@qq.com>
Date: Sat, 14 Jun 2025 19:47:44 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20feat(database):=20implement=20datab?=
 =?UTF-8?q?ase=20migration=20logic=20for=20PostgreSQL=20and=20add=20fast?=
 =?UTF-8?q?=20migration=20fallback?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 model/main.go | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/model/main.go b/model/main.go
index 289baa2f..965bba93 100644
--- a/model/main.go
+++ b/model/main.go
@@ -232,6 +232,30 @@ func InitLogDB() (err error) {
 }
 
 func migrateDB() error {
+	if !common.UsingPostgreSQL {
+		return migrateDBFast()
+	}
+	err := DB.AutoMigrate(
+		&Channel{},
+		&Token{},
+		&User{},
+		&Option{},
+		&Redemption{},
+		&Ability{},
+		&Log{},
+		&Midjourney{},
+		&TopUp{},
+		&QuotaData{},
+		&Task{},
+		&Setup{},
+	)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func migrateDBFast() error {
 	var wg sync.WaitGroup
 	errChan := make(chan error, 12) // Buffer size matches number of migrations
 
@@ -273,7 +297,6 @@ func migrateDB() error {
 			return err
 		}
 	}
-
 	common.SysLog("database migrated")
 	return nil
 }