部分更改

srv/internal/common/database-client.go

@@ -1,6 +1,7 @@
 package common
 
 import (
+	"Crimson-Gatekeeper/internal/query"
 	"fmt"
 
 	"gorm.io/driver/postgres"
@@ -12,7 +13,7 @@ import (
 // 返回值
 //   - 一个 gorm.DB 的指针
 //   - 一个清理所有数据库相关数据的函数
-func GetDataBaseClient() (*gorm.DB, func() error) {
+func GetDataBaseClient() (*gorm.DB, func(), error) {
 	dsn := "host=localhost " +
 		"user=gatekeeper " +
 		"dbname=crimson " +
@@ -24,17 +25,30 @@ func GetDataBaseClient() (*gorm.DB, func() error) {
 	client, ero := gorm.Open(postgres.Open(dsn), &gorm.Config{})
 	if ero != nil {
 		fmt.Println("数据库链接建立失败")
-		panic(ero)
+		return nil, nil, ero
 	}
 
 	poolCfg, ero := client.DB()
 	if ero != nil {
 		fmt.Println("获取数据库链接池失败")
-		panic(ero)
+
+		return nil, nil, ero
+	}
+
+	cleanPoolCfg := func() {
+		ero := poolCfg.Close()
+		if ero != nil {
+			fmt.Println("清理过程中出现错误")
+			panic(ero)
+		}
 	}
 
 	poolCfg.SetConnMaxIdleTime(3)
 	poolCfg.SetMaxOpenConns(10)
 
-	return client, poolCfg.Close
+	return client, cleanPoolCfg, nil
+}
+
+func GetQuery(db *gorm.DB) *query.Query {
+	return query.Use(db)
 }