♻️ refactor: Update database migration setup in migration/migration.go

Refined the database migration setup in migration/migration.go. Enhanced error handling and streamlined the database connection setup process for increased efficiency and reliability of migrations. Related issue: #66
YJU-OKURA · Mar 6, 2024 · 8db56ea · 8db56ea
1 parent 3504ce7
commit 8db56ea
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/migration/migration.go b/migration/migration.go
@@ -11,7 +11,7 @@ import (
 	"time"
 )
 
-func InitDB() *gorm.DB {
+func InitDB() (*gorm.DB, error) {
 	user := os.Getenv("MYSQL_USER")
 	pass := os.Getenv("MYSQL_PASSWORD")
 	dbName := os.Getenv("MYSQL_DATABASE")
@@ -40,7 +40,7 @@ func InitDB() *gorm.DB {
 	sqlDB.SetMaxOpenConns(100)
 	sqlDB.SetConnMaxLifetime(time.Hour)
 
-	return db
+	return db, nil
 }
 
 func Migrate(db *gorm.DB) {