100 lines
2.3 KiB
Go
100 lines
2.3 KiB
Go
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"admin/lib/xlog"
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"gorm.io/gorm/logger"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type gormLogger struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *gormLogger) LogMode(logger.LogLevel) logger.Interface {
|
||
|
|
return l
|
||
|
|
}
|
||
|
|
func (l *gormLogger) Info(ctx context.Context, format string, args ...interface{}) {
|
||
|
|
xlog.Infof(fmt.Sprintf("[GORM LOGGER] "+format, args...))
|
||
|
|
}
|
||
|
|
func (l *gormLogger) Warn(ctx context.Context, format string, args ...interface{}) {
|
||
|
|
xlog.Warnf(fmt.Sprintf("[GORM LOGGER] "+format, args...))
|
||
|
|
}
|
||
|
|
func (l *gormLogger) Error(ctx context.Context, format string, args ...interface{}) {
|
||
|
|
xlog.Errorf(fmt.Sprintf("[GORM LOGGER] "+format, args...))
|
||
|
|
}
|
||
|
|
func (l *gormLogger) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) {
|
||
|
|
sqlStr, aff := fc()
|
||
|
|
if strings.Contains(sqlStr, "SHOW STATUS") {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if err != nil {
|
||
|
|
xlog.Errorf("[GORM LOGGER] "+"sql:%v affected:%v error:%v", sqlStr, aff, err)
|
||
|
|
cmd, table := getSqlCmdAndTable(sqlStr)
|
||
|
|
if cmd == "" || table == "" {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
//metricsMysqlErrorCount.LabelValues(cmd, table).Add(1)
|
||
|
|
} else {
|
||
|
|
dura := time.Now().Sub(begin)
|
||
|
|
if dura.Milliseconds() > 2000 {
|
||
|
|
xlog.Warnf("[GORM LOGGER] [SLOW] "+"sql:%v affected:%v use %vms", sqlStr, aff, dura.Milliseconds())
|
||
|
|
} else {
|
||
|
|
xlog.Tracef("[GORM LOGGER] "+"sql:%v affected:%v", sqlStr, aff)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func getSqlCmdAndTable(sql string) (string, string) {
|
||
|
|
cmd := ""
|
||
|
|
table := ""
|
||
|
|
idx := strings.IndexAny(sql, " ")
|
||
|
|
if idx <= 0 {
|
||
|
|
cmd = sql
|
||
|
|
} else {
|
||
|
|
cmd = sql[:idx]
|
||
|
|
}
|
||
|
|
switch cmd {
|
||
|
|
case "SELECT":
|
||
|
|
if idx := strings.Index(sql, "FROM"); idx >= 0 {
|
||
|
|
for i := idx + 5; i < len(sql); i++ {
|
||
|
|
if sql[i] == ' ' {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
table += string(sql[i])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
case "UPDATE":
|
||
|
|
if idx := strings.Index(sql, "UPDATE"); idx >= 0 {
|
||
|
|
for i := idx + 7; i < len(sql); i++ {
|
||
|
|
if sql[i] == ' ' {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
table += string(sql[i])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
case "INSERT":
|
||
|
|
if idx := strings.Index(sql, "INTO"); idx >= 0 {
|
||
|
|
for i := idx + 5; i < len(sql); i++ {
|
||
|
|
if sql[i] == ' ' || sql[i] == '(' {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
table += string(sql[i])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
case "DELETE":
|
||
|
|
if idx := strings.Index(sql, "FROM"); idx >= 0 {
|
||
|
|
for i := idx + 5; i < len(sql); i++ {
|
||
|
|
if sql[i] == ' ' {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
table += string(sql[i])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
default:
|
||
|
|
}
|
||
|
|
|
||
|
|
return cmd, table
|
||
|
|
}
|