From 1e08057cdaa0194d38fe6c22b08d7dcb87fb0579 Mon Sep 17 00:00:00 2001 From: yv1ing Date: Thu, 28 Aug 2025 17:51:50 +0800 Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E7=B3=BB=E7=BB=9F=E6=95=B0?= =?UTF-8?q?=E6=8D=AE=E5=88=9D=E5=A7=8B=E5=8C=96=E6=96=B9=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/core/initialize/database.go | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/internal/core/initialize/database.go b/internal/core/initialize/database.go index 296c211..0f17600 100644 --- a/internal/core/initialize/database.go +++ b/internal/core/initialize/database.go @@ -26,13 +26,22 @@ func dsn() string { ) } +func recreateTables(db *gorm.DB, models ...interface{}) error { + err := db.Migrator().DropTable(models...) + if err != nil { + return err + } + return db.AutoMigrate(models...) +} + func InitDatabase() error { db, err := gorm.Open(mysql.Open(dsn()), &gorm.Config{}) if err != nil { return err } - err = db.AutoMigrate( + err = recreateTables( + db, &systemmodel.User{}, )