From 4e113fac27f140bfb2323cd7c6daa2779fac6751 Mon Sep 17 00:00:00 2001
From: JACKYMYPERSON <mayimingperson_person@163.com>
Date: Sat, 13 Sep 2025 20:01:10 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9gorm=20close?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.go | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index 8a6abe1..5bd392d 100644
--- a/main.go
+++ b/main.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"log"
 	"strconv"
 	"toutoukan/config"
 	"toutoukan/controllers/search"
@@ -22,7 +23,21 @@ func main() {
 	if err != nil {
 		return
 	}
-	defer databaseInit.UserDB.Close()
+	defer func() {
+		sqlDB, err := databaseInit.UserDB.DB()
+		if err != nil {
+			// 处理获取失败的情况
+			log.Printf("获取底层数据库连接失败: %v", err)
+			return
+		}
+
+		// 2. 调用 Close() 关闭连接池
+		if err := sqlDB.Close(); err != nil {
+			log.Printf("关闭数据库连接池失败: %v", err)
+		} else {
+			log.Println("数据库连接池已成功关闭")
+		}
+	}()
 	defer redisInit.RedisClient.Close()
 	defer search.CloseIndex()
 }