修改gorm close

This commit is contained in:
JACKYMYPERSON
2025-09-13 20:01:10 +08:00
parent 2bdeff69b3
commit 4e113fac27

17
main.go
View File

@@ -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()
}