/
db_logger.go
88 lines (77 loc) · 2.43 KB
/
db_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
package adapter
import (
"context"
"errors"
"fmt"
"time"
"gorm.io/gorm/utils"
"github.com/duc-cnzj/execit/internal/xlog"
"gorm.io/gorm"
"gorm.io/gorm/logger"
)
type GormLoggerAdapter struct {
level logger.LogLevel
}
func (g *GormLoggerAdapter) LogMode(level logger.LogLevel) logger.Interface {
g.level = level
return g
}
func (g *GormLoggerAdapter) Info(ctx context.Context, s string, i ...any) {
if g.level >= logger.Info {
xlog.Infof(s, i...)
}
}
func (g *GormLoggerAdapter) Warn(ctx context.Context, s string, i ...any) {
if g.level >= logger.Warn {
xlog.Warningf(s, i...)
}
}
func (g *GormLoggerAdapter) Error(ctx context.Context, s string, i ...any) {
if g.level >= logger.Error {
xlog.Errorf(s, i...)
}
}
func (g *GormLoggerAdapter) Trace(ctx context.Context, begin time.Time, fc func() (string, int64), err error) {
const (
traceStr = "[SQL]: [%.3fms] [rows:%v] %s %s"
traceWarnStr = "[SQL]: %s [%.3fms] [rows:%v] %s %s"
traceErrStr = "[SQL]: %s [%.3fms] [rows:%v] %s %s"
slowThreshold = 200 * time.Millisecond
)
if g.level > logger.Silent {
elapsed := time.Since(begin)
switch {
case err != nil && g.level >= logger.Error:
if errors.Is(err, gorm.ErrRecordNotFound) {
sql, rows := fc()
if rows == -1 {
xlog.Debugf(traceErrStr, err, float64(elapsed.Nanoseconds())/1e6, "-", sql, utils.FileWithLineNum())
} else {
xlog.Debugf(traceErrStr, err, float64(elapsed.Nanoseconds())/1e6, rows, sql, utils.FileWithLineNum())
}
return
}
sql, rows := fc()
if rows == -1 {
xlog.Errorf(traceErrStr, err, float64(elapsed.Nanoseconds())/1e6, "-", sql, utils.FileWithLineNum())
} else {
xlog.Errorf(traceErrStr, err, float64(elapsed.Nanoseconds())/1e6, rows, sql, utils.FileWithLineNum())
}
case elapsed > slowThreshold && g.level >= logger.Warn:
sql, rows := fc()
slowLog := fmt.Sprintf("(SLOW SQL) >= %v", slowThreshold)
if rows == -1 {
xlog.Warningf(traceWarnStr, slowLog, float64(elapsed.Nanoseconds())/1e6, "-", sql, utils.FileWithLineNum())
} else {
xlog.Warningf(traceWarnStr, slowLog, float64(elapsed.Nanoseconds())/1e6, rows, sql, utils.FileWithLineNum())
}
case g.level == logger.Info:
sql, rows := fc()
if rows == -1 {
xlog.Infof(traceStr, float64(elapsed.Nanoseconds())/1e6, "-", sql, utils.FileWithLineNum())
} else {
xlog.Infof(traceStr, float64(elapsed.Nanoseconds())/1e6, rows, sql, utils.FileWithLineNum())
}
}
}
}