/
xloggorm.go
73 lines (65 loc) · 1.88 KB
/
xloggorm.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
package xlog
import (
"context"
"errors"
"fmt"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
gormlog "gorm.io/gorm/logger"
"time"
)
type GormLogger struct {
logger *zap.SugaredLogger
gormlog.Config
}
func (g *GormLogger) LogMode(logLevel gormlog.LogLevel) gormlog.Interface {
var zapLevel zapcore.Level
switch logLevel {
case gormlog.Silent:
zapLevel = zapcore.DPanicLevel
case gormlog.Error:
zapLevel = zapcore.ErrorLevel
case gormlog.Warn:
zapLevel = zapcore.WarnLevel
case gormlog.Info:
zapLevel = zapcore.InfoLevel
}
zapLevel = zapLevel
//level.SetLevel(zapLevel)
return g
}
func (g *GormLogger) Info(ctx context.Context, s string, i ...interface{}) {
//TODO implement me
g.logger.Infof(s, i...)
}
func (g *GormLogger) Warn(ctx context.Context, s string, i ...interface{}) {
//TODO implement me
g.logger.Warnf(s, i...)
}
func (g *GormLogger) Error(ctx context.Context, s string, i ...interface{}) {
//TODO implement me
g.logger.Errorf(s, i...)
}
func (g *GormLogger) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) {
//TODO implement me
if g.logger.Level() >= zapcore.DPanicLevel {
return
}
elapsed := time.Since(begin)
sql, rows := fc()
switch {
case err != nil && g.logger.Level() <= zapcore.ErrorLevel && (!errors.Is(err, gormlog.ErrRecordNotFound) || !g.IgnoreRecordNotFoundError):
g.logger.Errorw(sql, "elapsed", elapsed, "rows", rows, "error", err.Error())
case elapsed > g.SlowThreshold && g.SlowThreshold != 0 && g.logger.Level() >= zapcore.WarnLevel:
slowLog := fmt.Sprintf("SLOW SQL >= %v", g.SlowThreshold)
g.logger.Warnw(sql, "elapsed", elapsed, "rows", rows, "warn", slowLog)
case g.logger.Level() < zapcore.InfoLevel:
g.logger.Debugw(sql, "elapsed", elapsed, "rows", rows)
}
}
func NewGormLogger(conf gormlog.Config) gormlog.Interface {
return &GormLogger{
logger: textLogger,
Config: conf,
}
}