From 35d2ba9f087b9608dc2abe29121d132e424ee2f2 Mon Sep 17 00:00:00 2001
From: qqxhb <1252905006@qq.com>
Date: Fri, 18 Aug 2023 19:33:38 +0800
Subject: [PATCH] feat: nil

---
 gorm.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/gorm.go b/gorm.go
index 61633c33..32193870 100644
--- a/gorm.go
+++ b/gorm.go
@@ -374,8 +374,10 @@ func (db *DB) AddError(err error) error {
 
 // DB returns `*sql.DB`
 func (db *DB) DB() (*sql.DB, error) {
-	connPool := db.Statement.ConnPool
-
+	connPool := db.ConnPool
+	if db.Statement != nil && db.Statement.ConnPool != nil {
+		connPool = db.Statement.ConnPool
+	}
 	if tx, ok := connPool.(*sql.Tx); ok && tx != nil {
 		return (*sql.DB)(reflect.ValueOf(tx).Elem().FieldByName("db").UnsafePointer()), nil
 	}