xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-13 03:43:44 +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

5
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,6 +135,7 @@ func InitDB() {
} }
func migrateDB() error { func migrateDB() error {
if env.Bool("StartSqlMigration", false) {
var err error var err error
if err = DB.AutoMigrate(&Channel{}); err != nil { if err = DB.AutoMigrate(&Channel{}); err != nil {
return err return err
@@ -157,6 +161,7 @@ func migrateDB() error {
if err = DB.AutoMigrate(&Channel{}); err != nil { if err = DB.AutoMigrate(&Channel{}); err != nil {
return err return err
} }
}
return nil return nil
} }