From 224c57f8f31d862691513ef9b33fc4d7e17ac580 Mon Sep 17 00:00:00 2001
From: KantaaHasegawa <gyoza20000@gmail.com>
Date: Sat, 3 Jun 2023 21:13:36 +0900
Subject: [PATCH] refactor: Detecting nil in dbConnector.GetDBConn()

---
 gorm.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/gorm.go b/gorm.go
index 48e307cc..92fc8dd4 100644
--- a/gorm.go
+++ b/gorm.go
@@ -376,7 +376,9 @@ func (db *DB) DB() (*sql.DB, error) {
 	connPool := db.ConnPool
 
 	if dbConnector, ok := connPool.(GetDBConnector); ok && dbConnector != nil {
-		return dbConnector.GetDBConn()
+		if sqldb, err := dbConnector.GetDBConn(); sqldb != nil && err != nil {
+			return sqldb, err
+		}
 	}
 
 	if sqldb, ok := connPool.(*sql.DB); ok && sqldb != nil {