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

5
gorm.go
View File

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