open
/
goutil
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

gormLogger2

This commit is contained in:
jiangyong27 2023-12-13 11:08:14 +08:00
parent be39486043
commit 5735c03cc9
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
gorm.go
View File

@ -9,6 +9,7 @@ import (
)
type GORMLogger struct {
Threshold int64
}
func (d *GORMLogger) LogMode(level logger.LogLevel) logger.Interface {
@ -33,6 +34,10 @@ func (d *GORMLogger) Trace(ctx context.Context, begin time.Time, fc func() (sql
if err != nil && err != gorm.ErrRecordNotFound {
log.Errorf("[SQL]sql=%s affect=%d cost=%dms error=%v", sql, affects, time.Since(begin).Milliseconds(), err)
} else {
log.Debugf("[SQL]sql=%s affect=%d cost=%dms", sql, affects, time.Since(begin).Milliseconds())
if d.Threshold > 0 && time.Since(begin).Milliseconds() > d.Threshold {
log.Errorf("[SQL]sql=%s affect=%d cost=%dms", sql, affects, time.Since(begin).Milliseconds())
} else {
log.Debugf("[SQL]sql=%s affect=%d cost=%dms", sql, affects, time.Since(begin).Milliseconds())
}
}
}