mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-11-08 17:53:41 +08:00
feat: 增加迁移开关
This commit is contained in:
@@ -116,7 +116,10 @@ func InitDB() {
|
||||
return
|
||||
}
|
||||
|
||||
sqlDB := setDBConns(DB)
|
||||
|
||||
if !config.IsMasterNode {
|
||||
return
|
||||
}
|
||||
|
||||
if common.UsingMySQL {
|
||||
@@ -132,6 +135,7 @@ func InitDB() {
|
||||
}
|
||||
|
||||
func migrateDB() error {
|
||||
if env.Bool("StartSqlMigration", false) {
|
||||
var err error
|
||||
if err = DB.AutoMigrate(&Channel{}); err != nil {
|
||||
return err
|
||||
@@ -157,6 +161,7 @@ func migrateDB() error {
|
||||
if err = DB.AutoMigrate(&Channel{}); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user