forked from SpectatorNan/gorm-zero
-
Notifications
You must be signed in to change notification settings - Fork 0
/
log.go
96 lines (87 loc) · 2.14 KB
/
log.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
package gormc
import (
"context"
"errors"
"time"
"github.com/zeromicro/go-zero/core/logx"
"gorm.io/gorm"
"gorm.io/gorm/logger"
)
type GormLog struct {
Level logger.LogLevel
IgnoreRecordNotFoundError bool
SlowThreshold time.Duration
}
func (g *GormLog) LogMode(level logger.LogLevel) logger.Interface {
newLog := *g
newLog.Level = level
return &newLog
}
func (g *GormLog) Info(ctx context.Context, format string, data ...any) {
if g.Level < logger.Info {
return
}
logx.WithContext(ctx).Infof(format, data...)
}
func (g *GormLog) Warn(ctx context.Context, format string, data ...any) {
if g.Level < logger.Warn {
return
}
logx.WithContext(ctx).Errorf(format, data...)
}
func (g *GormLog) Error(ctx context.Context, format string, data ...any) {
if g.Level < logger.Error {
return
}
logx.WithContext(ctx).Errorf(format, data...)
}
func (g *GormLog) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) {
if g.Level <= logger.Silent {
return
}
elapsed := time.Since(begin)
switch {
case err != nil && g.Level >= logger.Error && (!errors.Is(err, gorm.ErrRecordNotFound) || !g.IgnoreRecordNotFoundError):
sql, rows := fc()
if rows == -1 {
logx.WithContext(ctx).WithDuration(elapsed).Errorw(
err.Error(),
logx.Field("sql", sql),
)
} else {
logx.WithContext(ctx).WithDuration(elapsed).Errorw(
err.Error(),
logx.Field("sql", sql),
logx.Field("rows", rows),
)
}
case elapsed > g.SlowThreshold && g.SlowThreshold != 0 && g.Level >= logger.Warn:
sql, rows := fc()
if rows == -1 {
logx.WithContext(ctx).WithDuration(elapsed).Sloww(
"slow sql",
logx.Field("sql", sql),
)
} else {
logx.WithContext(ctx).WithDuration(elapsed).Sloww(
"slow sql",
logx.Field("sql", sql),
logx.Field("rows", rows),
)
}
case g.Level == logger.Info:
sql, rows := fc()
if rows == -1 {
logx.WithContext(ctx).WithDuration(elapsed).Infow(
"",
logx.Field("sql", sql),
)
} else {
logx.WithContext(ctx).WithDuration(elapsed).Infow(
"",
logx.Field("sql", sql),
logx.Field("rows", rows),
)
}
}
}