认证授权主要流程实现

init/orm/orm.go

@@ -4,40 +4,39 @@ import (
 	"fmt"
 	"log/slog"
 	"platform/init/env"
-	"platform/init/logs"
+	"platform/web/queries"
 
 	"gorm.io/gorm"
 	"gorm.io/gorm/schema"
 )
 import "gorm.io/driver/postgres"
 
-var DB *gorm.DB
-
 func Init() {
-	logger := logs.Default
 
+	// 连接数据库
 	dsn := fmt.Sprintf(
 		"host=%s user=%s password=%s dbname=%s port=%s sslmode=disable TimeZone=Asia/Shanghai",
-		env.DbName, env.DbUserName, env.DbPassword, env.DbName, env.DbPort,
+		env.DbHost, env.DbUserName, env.DbPassword, env.DbName, env.DbPort,
 	)
-
-	open, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
+	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
 		NamingStrategy: schema.NamingStrategy{
 			SingularTable: true,
 		},
 	})
 	if err != nil {
-		logger.Error("gorm 打开数据库失败", slog.Any("err", err))
+		slog.Error("gorm 初始化数据库失败：", slog.Any("err", err))
 		panic(err)
 	}
 
-	sql, err := open.DB()
+	// 连接池
+	conn, err := db.DB()
 	if err != nil {
-		logger.Error("gorm open db error: ", slog.Any("err", err))
+		slog.Error("gorm 初始化数据库失败：", slog.Any("err", err))
 		panic(err)
 	}
-	sql.SetMaxIdleConns(10)
-	sql.SetMaxOpenConns(100)
+	conn.SetMaxIdleConns(10)
+	conn.SetMaxOpenConns(100)
 
-	DB = open
+	// 初始化查询工具
+	queries.SetDefault(db)
 }