From a9f42abb59461cfa2ad1bbdc6ece003922309466 Mon Sep 17 00:00:00 2001 From: suziheng Date: Mon, 2 Dec 2024 10:32:39 +0800 Subject: [PATCH] =?UTF-8?q?=E8=AE=BE=E7=BD=AE=E8=BF=81=E7=A7=BB=E5=BC=80?= =?UTF-8?q?=E5=85=B3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- model/main.go | 60 ++++++++++++++++++++++++++------------------------- 1 file changed, 31 insertions(+), 29 deletions(-) diff --git a/model/main.go b/model/main.go index ff542faf..60b9d849 100644 --- a/model/main.go +++ b/model/main.go @@ -108,36 +108,38 @@ func InitDB(envName string) (db *gorm.DB, err error) { if common.UsingMySQL { _, _ = sqlDB.Exec("DROP INDEX idx_channels_key ON channels;") // TODO: delete this line when most users have upgraded } - logger.SysLog("database migration started") - err = db.AutoMigrate(&Channel{}) - if err != nil { - return nil, err + if env.Bool("StartSqlMigration", false) { + logger.SysLog("database migration started") + err = db.AutoMigrate(&Channel{}) + if err != nil { + return nil, err + } + err = db.AutoMigrate(&Token{}) + if err != nil { + return nil, err + } + err = db.AutoMigrate(&User{}) + if err != nil { + return nil, err + } + err = db.AutoMigrate(&Option{}) + if err != nil { + return nil, err + } + err = db.AutoMigrate(&Redemption{}) + if err != nil { + return nil, err + } + err = db.AutoMigrate(&Ability{}) + if err != nil { + return nil, err + } + err = db.AutoMigrate(&Log{}) + if err != nil { + return nil, err + } + logger.SysLog("database migrated") } - err = db.AutoMigrate(&Token{}) - if err != nil { - return nil, err - } - err = db.AutoMigrate(&User{}) - if err != nil { - return nil, err - } - err = db.AutoMigrate(&Option{}) - if err != nil { - return nil, err - } - err = db.AutoMigrate(&Redemption{}) - if err != nil { - return nil, err - } - err = db.AutoMigrate(&Ability{}) - if err != nil { - return nil, err - } - err = db.AutoMigrate(&Log{}) - if err != nil { - return nil, err - } - logger.SysLog("database migrated") return db, err } else { logger.FatalLog(err)