xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-12 11:23:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: 增加迁移开关

Browse Source
This commit is contained in:
suziheng
2024-12-24 10:08:38 +08:00
parent f8fcb1d258
commit 6eb4e788c7
1 changed files with 29 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
model/main.go
View File

@@ -116,7 +116,10 @@ func InitDB() {
return return
} }
sqlDB := setDBConns(DB)
if !config.IsMasterNode {
return
} }
if common.UsingMySQL { if common.UsingMySQL {
@@ -132,30 +135,32 @@ func InitDB() {
} }
func migrateDB() error { func migrateDB() error {
var err error if env.Bool("StartSqlMigration", false) {
if err = DB.AutoMigrate(&Channel{}); err != nil { var err error
return err if err = DB.AutoMigrate(&Channel{}); err != nil {
} return err
if err = DB.AutoMigrate(&Token{}); err != nil { }
return err if err = DB.AutoMigrate(&Token{}); err != nil {
} return err
if err = DB.AutoMigrate(&User{}); err != nil { }
return err if err = DB.AutoMigrate(&User{}); err != nil {
} return err
if err = DB.AutoMigrate(&Option{}); err != nil { }
return err if err = DB.AutoMigrate(&Option{}); err != nil {
} return err
if err = DB.AutoMigrate(&Redemption{}); err != nil { }
return err if err = DB.AutoMigrate(&Redemption{}); err != nil {
} return err
if err = DB.AutoMigrate(&Ability{}); err != nil { }
return err if err = DB.AutoMigrate(&Ability{}); err != nil {
} return err
if err = DB.AutoMigrate(&Log{}); err != nil { }
return err if err = DB.AutoMigrate(&Log{}); err != nil {
} return err
if err = DB.AutoMigrate(&Channel{}); err != nil { }
return err if err = DB.AutoMigrate(&Channel{}); err != nil {
return err
}
} }
return nil return nil
} }