🎨用户接口处理完毕

initialize/init.go

@@ -2,6 +2,7 @@ package initialize
 
 import (
 	"fmt"
+	"gitee.ltd/lxh/logger"
 	"github.com/cowardmrx/go_aliyun_oss"
 	"github.com/go-resty/resty/v2"
 	"github.com/redis/go-redis/v9"
@@ -11,10 +12,8 @@ import (
 	"gorm.io/driver/postgres"
 	"gorm.io/driver/sqlite"
 	"gorm.io/gorm"
-	"gorm.io/gorm/logger"
 	"log"
 	"os"
-	"time"
 	"wireguard-dashboard/client"
 	"wireguard-dashboard/config"
 )
@@ -22,6 +21,7 @@ import (
 // Init
 // @description: 初始化
 func Init() {
+	initLogger()     // 初始化日志
 	initConfig()     // 读取配置文件
 	initWireguard()  // 初始化wireguard客户端
 	initDatabase()   // 初始化数据库
@@ -58,17 +58,6 @@ func initWireguard() {
 // initDatabase
 // @description: 初始化数据库
 func initDatabase() {
-	gormLogger := logger.New(
-		log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer
-		logger.Config{
-			SlowThreshold:             time.Second,   // Slow SQL threshold
-			LogLevel:                  logger.Silent, // Log level
-			IgnoreRecordNotFoundError: true,          // Ignore ErrRecordNotFound error for logger
-			ParameterizedQueries:      true,          // Don't include params in the SQL log
-			Colorful:                  true,          // Disable color
-		},
-	)
-
 	// 不同驱动提供
 	var dbDialector gorm.Dialector
 	switch config.Config.Database.Driver {
@@ -81,7 +70,7 @@ func initDatabase() {
 	}
 
 	db, err := gorm.Open(dbDialector, &gorm.Config{
-		Logger: gormLogger,
+		Logger: logger.DefaultGormLogger(),
 	})
 
 	if err != nil {
@@ -126,3 +115,12 @@ func initOSS() {
 	ossClient := ossConfig.CreateOssConnect()
 	client.OSS = ossClient
 }
+
+// initLogger
+// @description: 初始化日志
+func initLogger() {
+	logger.InitLogger(logger.LogConfig{
+		Mode:       logger.Dev,
+		FileEnable: true,
+	})
+}