From 0b8737756925331d159df7ce859e17c34ab8c7b8 Mon Sep 17 00:00:00 2001
From: bugtower100 <2889352839@qq.com>
Date: Wed, 8 Jan 2025 16:34:21 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E6=AD=A3PostgreSQL=E5=92=8CMyS?=
 =?UTF-8?q?QL=E7=9A=84=E6=97=A5=E5=BF=97=E4=B8=8D=E8=83=BD=E6=AD=A3?=
 =?UTF-8?q?=E7=A1=AE=E6=89=93=E5=8D=B0=E5=88=B0database.log=E7=9A=84?=
 =?UTF-8?q?=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dice/model/database/mysql.go | 15 +++------------
 dice/model/database/pgsql.go | 14 ++------------
 2 files changed, 5 insertions(+), 24 deletions(-)

diff --git a/dice/model/database/mysql.go b/dice/model/database/mysql.go
index c7310bcd..f4035690 100644
--- a/dice/model/database/mysql.go
+++ b/dice/model/database/mysql.go
@@ -1,10 +1,6 @@
 package database
 
 import (
-	"log"
-	"os"
-	"time"
-
 	"gorm.io/driver/mysql"
 	"gorm.io/gorm"
 	"gorm.io/gorm/logger"
@@ -15,14 +11,9 @@ import (
 func MySQLDBInit(dsn string) (*gorm.DB, error) {
 	// 构建 MySQL DSN (Data Source Name)
 	// 使用 GORM 连接 MySQL
-	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{Logger: logger.New(
-		log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer
-		logger.Config{
-			SlowThreshold: time.Second, // 慢 SQL 阈值
-			LogLevel:      logger.Info, // 记录所有SQL操作
-			Colorful:      true,        // 是否启用彩色打印
-		},
-	)})
+	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
+		// 注意，这里虽然是Info,但实际上打印就变成了Debug.
+		Logger: logger.Default.LogMode(logger.Info)})
 	if err != nil {
 		return nil, err
 	}
diff --git a/dice/model/database/pgsql.go b/dice/model/database/pgsql.go
index 1f74cb05..aa24e54a 100644
--- a/dice/model/database/pgsql.go
+++ b/dice/model/database/pgsql.go
@@ -1,10 +1,6 @@
 package database
 
 import (
-	"log"
-	"os"
-	"time"
-
 	"gorm.io/driver/postgres"
 	"gorm.io/gorm"
 	"gorm.io/gorm/logger"
@@ -17,14 +13,8 @@ func PostgresDBInit(dsn string) (*gorm.DB, error) {
 
 	// 使用 GORM 连接 PostgreSQL
 	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
-		Logger: logger.New(
-			log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer
-			logger.Config{
-				SlowThreshold: time.Second, // 慢 SQL 阈值
-				LogLevel:      logger.Info, // 记录所有SQL操作
-				Colorful:      true,        // 是否启用彩色打印
-			},
-		),
+		// 注意，这里虽然是Info,但实际上打印就变成了Debug.
+		Logger: logger.Default.LogMode(logger.Info),
 	})
 	if err != nil {
 		return nil, err