From da89fa99bab2450a9397f1f34fb56f8e5f915688 Mon Sep 17 00:00:00 2001
From: demoManito <1430482733@qq.com>
Date: Tue, 2 Jul 2024 19:16:21 +0800
Subject: [PATCH] DB provides Close function

---
 gorm.go | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/gorm.go b/gorm.go
index 117d2fd0..303c32de 100644
--- a/gorm.go
+++ b/gorm.go
@@ -217,6 +217,15 @@ func Open(dialector Dialector, opts ...Option) (db *DB, err error) {
 	return
 }
 
+// Close close current db connection
+func (db *DB) Close() error {
+	sqlDB, err := db.DB()
+	if err != nil {
+		return err
+	}
+	return sqlDB.Close()
+}
+
 // Session create new db session
 func (db *DB) Session(config *Session) *DB {
 	var (