57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
|
package logger
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"io"
|
|||
|
|
"mini_server/internal/config"
|
|||
|
|
"os"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"github.com/sirupsen/logrus"
|
|||
|
|
"gorm.io/gorm/logger"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var GinLogger io.Writer
|
|||
|
|
var GormLogger logger.Interface
|
|||
|
|
|
|||
|
|
func create_file() *os.File {
|
|||
|
|
f, err := os.OpenFile(config.LogFile, os.O_CREATE|os.O_APPEND|os.O_RDWR, 0644)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Printf("日志文件打开失败，err:%v", err)
|
|||
|
|
os.Exit(-1)
|
|||
|
|
}
|
|||
|
|
return f
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Init_log() {
|
|||
|
|
|
|||
|
|
f := create_file()
|
|||
|
|
// writer := io.MultiWriter(f, os.Stdout)
|
|||
|
|
writer := io.MultiWriter(f)
|
|||
|
|
|
|||
|
|
logrus.SetLevel(config.LogLevel)
|
|||
|
|
//设置日志格式，使用自定义格式
|
|||
|
|
logrus.SetFormatter(&MyFormatter{for_server: true})
|
|||
|
|
// 设置业务日志
|
|||
|
|
gin.DisableConsoleColor()
|
|||
|
|
logrus.SetOutput(writer)
|
|||
|
|
|
|||
|
|
// 设置GIN日志文件
|
|||
|
|
GinLogger = io.MultiWriter(writer)
|
|||
|
|
|
|||
|
|
gormLogrus := logrus.New()
|
|||
|
|
gormLogrus.SetFormatter(&MyFormatter{for_gorm: true})
|
|||
|
|
gormLogrus.SetOutput(writer)
|
|||
|
|
// 设置GORM日志文件
|
|||
|
|
GormLogger = NewGormLogger(
|
|||
|
|
gormLogrus,
|
|||
|
|
logger.Config{
|
|||
|
|
SlowThreshold: time.Second, // 慢SQL阈值
|
|||
|
|
LogLevel: logger.Info,
|
|||
|
|
Colorful: false, // 禁用彩色打印
|
|||
|
|
IgnoreRecordNotFoundError: true, // 不打印未查询到的日志
|
|||
|
|
},
|
|||
|
|
)
|
|||
|
|
}
|