Handle error when database connection fails

2024-09-03 13:45:15 +07:00
parent 541b3be419
commit 8b050bc1da
1 changed files with 4 additions and 0 deletions
--- a/db/database.go
+++ b/db/database.go
@ -64,6 +64,10 @@ func NewMYSQLdb(username, password, host, port, dbName string) Database {
 		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)
 	if count <= 0 {
 		if err := initDB.Exec("CREATE DATABASE IF NOT EXISTS " + dbName).Error; err != nil {