From 8b050bc1da0e8a60d37abaa7f2e9f13bdedd5aa0 Mon Sep 17 00:00:00 2001 From: bagas Date: Tue, 3 Sep 2024 13:45:15 +0700 Subject: [PATCH] Handle error when database connection fails --- db/database.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/db/database.go b/db/database.go index e01b291..feabbcf 100644 --- 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 {