bagas/filekeeper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Handle error when database connection fails

Browse Source
This commit is contained in:
bagas
2024-09-03 13:45:15 +07:00
parent 541b3be419
commit 8b050bc1da
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
db/database.go
View File

@ -64,6 +64,10 @@ func NewMYSQLdb(username, password, host, port, dbName string) Database {
Logger: gormLogger.Default.LogMode(gormLogger.Silent), Logger: gormLogger.Default.LogMode(gormLogger.Silent),
}) })
if err != nil {
panic("failed to connect database: " + err.Error())
}
initDB.Raw("SELECT count(*) FROM information_schema.SCHEMATA WHERE SCHEMA_NAME = ?", dbName).Scan(&count) initDB.Raw("SELECT count(*) FROM information_schema.SCHEMATA WHERE SCHEMA_NAME = ?", dbName).Scan(&count)
if count <= 0 { if count <= 0 {
if err := initDB.Exec("CREATE DATABASE IF NOT EXISTS " + dbName).Error; err != nil { if err := initDB.Exec("CREATE DATABASE IF NOT EXISTS " + dbName).Error; err != nil {