diff --git a/model/main.go b/model/main.go index 6d567e17..c71243b6 100644 --- a/model/main.go +++ b/model/main.go @@ -116,7 +116,10 @@ func InitDB() { return } + sqlDB := setDBConns(DB) + if !config.IsMasterNode { + return } if common.UsingMySQL { @@ -132,30 +135,32 @@ func InitDB() { } func migrateDB() error { - var err error - if err = DB.AutoMigrate(&Channel{}); 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(&Option{}); 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(&Log{}); err != nil { - return err - } - if err = DB.AutoMigrate(&Channel{}); err != nil { - return err + if env.Bool("StartSqlMigration", false) { + var err error + if err = DB.AutoMigrate(&Channel{}); 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(&Option{}); 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(&Log{}); err != nil { + return err + } + if err = DB.AutoMigrate(&Channel{}); err != nil { + return err + } } return nil }