Use arg to control createDatabaseForPostgres()

This commit is contained in:
Yang Luo 2023-07-25 18:36:15 +08:00
parent 949feb18af
commit 6986dad295
2 changed files with 8 additions and 6 deletions

View File

@ -39,7 +39,7 @@ func getCreateDatabaseFlag() bool {
func main() { func main() {
createDatabase := getCreateDatabaseFlag() createDatabase := getCreateDatabaseFlag()
object.InitAdapter() object.InitAdapter(createDatabase)
object.CreateTables(createDatabase) object.CreateTables(createDatabase)
object.DoMigration() object.DoMigration()

View File

@ -42,15 +42,17 @@ func InitConfig() {
beego.BConfig.WebConfig.Session.SessionOn = true beego.BConfig.WebConfig.Session.SessionOn = true
InitAdapter() InitAdapter(true)
CreateTables(true) CreateTables(true)
DoMigration() DoMigration()
} }
func InitAdapter() { func InitAdapter(createDatabase bool) {
err := createDatabaseForPostgres(conf.GetConfigString("driverName"), conf.GetConfigDataSourceName(), conf.GetConfigString("dbName")) if createDatabase {
if err != nil { err := createDatabaseForPostgres(conf.GetConfigString("driverName"), conf.GetConfigDataSourceName(), conf.GetConfigString("dbName"))
panic(err) if err != nil {
panic(err)
}
} }
adapter = NewAdapter(conf.GetConfigString("driverName"), conf.GetConfigDataSourceName(), conf.GetConfigString("dbName")) adapter = NewAdapter(conf.GetConfigString("driverName"), conf.GetConfigDataSourceName(), conf.GetConfigString("dbName"))