94 lines
2.3 KiB
Go
94 lines
2.3 KiB
Go
|
package logger
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"go.uber.org/zap"
|
|||
|
|
gl "gorm.io/gorm/logger"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var gormZap *zap.SugaredLogger
|
|||
|
|
|
|||
|
|
// 基于Gorm的日志实现
|
|||
|
|
type gormLogger struct {
|
|||
|
|
gl.Config
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// LogMode 实现LogMode接口
|
|||
|
|
func (l *gormLogger) LogMode(level gl.LogLevel) gl.Interface {
|
|||
|
|
nl := *l
|
|||
|
|
nl.LogLevel = level
|
|||
|
|
return &nl
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Info 实现Info接口
|
|||
|
|
func (l gormLogger) Info(ctx context.Context, msg string, data ...interface{}) {
|
|||
|
|
if l.LogLevel >= gl.Info {
|
|||
|
|
// // 去掉第一行
|
|||
|
|
// msg = strings.Join(strings.Split(msg, "\n")[1:], " ")
|
|||
|
|
// gormZap.Info(msg)
|
|||
|
|
//
|
|||
|
|
// l.Printf(msg, append([]interface{}{utils.FileWithLineNum()}, data...)...)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Warn 实现Warn接口
|
|||
|
|
func (l gormLogger) Warn(ctx context.Context, msg string, data ...interface{}) {
|
|||
|
|
if l.LogLevel >= gl.Warn {
|
|||
|
|
//
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 实现Error接口
|
|||
|
|
func (l gormLogger) Error(ctx context.Context, msg string, data ...interface{}) {
|
|||
|
|
if l.LogLevel >= gl.Error {
|
|||
|
|
//
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Trace 实现Trace接口
|
|||
|
|
func (l gormLogger) Trace(ctx context.Context, begin time.Time, fc func() (string, int64), err error) {
|
|||
|
|
if l.LogLevel <= gl.Silent {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
elapsed := time.Since(begin)
|
|||
|
|
sql, rows := fc()
|
|||
|
|
msg := fmt.Sprintf("[%v] [rows:%v] %s", elapsed.String(), rows, sql)
|
|||
|
|
if rows == -1 {
|
|||
|
|
msg = fmt.Sprintf("[%s] [-] %s", elapsed.String(), sql)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
switch {
|
|||
|
|
case err != nil && l.LogLevel >= gl.Error && (!errors.Is(err, gl.ErrRecordNotFound) || !l.IgnoreRecordNotFoundError):
|
|||
|
|
gormZap.Errorf("%s -> %s", err.Error(), sql)
|
|||
|
|
case elapsed > l.SlowThreshold && l.SlowThreshold != 0 && l.LogLevel >= gl.Warn:
|
|||
|
|
slowLog := fmt.Sprintf("SLOW SQL >= %v", l.SlowThreshold)
|
|||
|
|
gormZap.Warnf("%v -> %v", slowLog, sql)
|
|||
|
|
case l.LogLevel == gl.Info:
|
|||
|
|
gormZap.Info(msg)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewGormLoggerWithConfig ...
|
|||
|
|
func NewGormLoggerWithConfig(config gl.Config) gl.Interface {
|
|||
|
|
return &gormLogger{config}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// DefaultGormLogger 默认的日志实现
|
|||
|
|
func DefaultGormLogger() gl.Interface {
|
|||
|
|
// 默认日志级别为Info，如果是生产环境，就是Error
|
|||
|
|
logLevel := gl.Info
|
|||
|
|
if config.Mode == Prod {
|
|||
|
|
logLevel = gl.Error
|
|||
|
|
}
|
|||
|
|
return &gormLogger{gl.Config{
|
|||
|
|
SlowThreshold: time.Second, // Slow SQL threshold
|
|||
|
|
IgnoreRecordNotFoundError: false, // 忽略没找到结果的错误
|
|||
|
|
LogLevel: logLevel, // Log level
|
|||
|
|
Colorful: false, // Disable color
|
|||
|
|
}}
|
|||
|
|
}
|