-
Notifications
You must be signed in to change notification settings - Fork 0
/
logger.go
99 lines (89 loc) · 2.65 KB
/
logger.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package logger
import (
"context"
"fmt"
logi "github.com/hopeio/cherry/utils/log"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
"gorm.io/gorm/logger"
"gorm.io/gorm/utils"
"strings"
"time"
)
var (
DefaultV2 = New(logi.GetCallerSkipLogger(0).Logger, &logger.Config{
SlowThreshold: 100 * time.Millisecond,
LogLevel: logger.Warn,
Colorful: true,
})
field = zap.String("comment", "gorm")
)
type Logger struct {
*zap.Logger
*logger.Config
}
type Config struct {
SlowThreshold time.Duration
Colorful bool
LogLevel zapcore.Level
}
func New(loger *zap.Logger, conf *logger.Config) logger.Interface {
if conf == nil {
conf = &logger.Config{LogLevel: logger.Warn}
}
loger.Core().Enabled(zapcore.Level(4 - conf.LogLevel))
return &Logger{Logger: loger, Config: conf}
}
// LogMode log mode
func (l *Logger) LogMode(level logger.LogLevel) logger.Interface {
l.Logger.Core().Enabled(zapcore.Level(4 - level))
l.LogLevel = level
return l
}
// Info print info
func (l *Logger) Info(ctx context.Context, msg string, data ...interface{}) {
l.Logger.Info(fmt.Sprintf(strings.TrimRight(msg, "\n"), data...), field)
}
// Warn print warn messages
func (l *Logger) Warn(ctx context.Context, msg string, data ...interface{}) {
l.Logger.Warn(fmt.Sprintf(strings.TrimRight(msg, "\n"), data...), field)
}
// Error print error messages
func (l *Logger) Error(ctx context.Context, msg string, data ...interface{}) {
l.Logger.Error(fmt.Sprintf(strings.TrimRight(msg, "\n"), data...), field)
}
// Trace print sql message 只有这里的context不是background,看了代码,也没用
func (l *Logger) Trace(ctx context.Context, begin time.Time, fc func() (string, int64), err error) {
if l.LogLevel == logger.Silent {
return
}
elapsed := time.Since(begin)
elapsedms := zap.Float64("elapsedms", float64(elapsed.Nanoseconds())/1e6)
level := logger.Info
var msg string
switch {
case err != nil:
level = logger.Error
msg = err.Error()
case elapsed > l.SlowThreshold && l.SlowThreshold != 0:
level = logger.Warn
msg = fmt.Sprintf("SLOW SQL >= %v", l.SlowThreshold)
}
if l.LogLevel < level {
return
}
switch level {
case logger.Error:
msg = err.Error()
case logger.Warn:
msg = fmt.Sprintf("SLOW SQL >= %v", l.SlowThreshold)
}
sql, rows := fc()
sqlField := zap.String("sql", sql)
rowsField := zap.Int64("rows", rows)
caller := zap.String("caller", utils.FileWithLineNum())
fields := []zap.Field{elapsedms, sqlField, rowsField, caller, logi.TraceIdField(ctx), field}
entry := l.Check(zapcore.Level(4-level), msg)
// entry.Caller = zapcore.NewEntryCaller(0, "", 0, false) utils.FileWithLineNum() or 获取到gorm的gormSourceDir
entry.Write(fields...)
}