nikkiing
1 year ago
parent
1e610cd927
commit
3bf07e0b28
@ -0,0 +1,56 @@ |
|||||||
|
package gl |
||||||
|
|
||||||
|
import ( |
||||||
|
"context" |
||||||
|
"github.com/go-kratos/kratos/v2/log" |
||||||
|
"go.opentelemetry.io/otel" |
||||||
|
"go.opentelemetry.io/otel/attribute" |
||||||
|
semconv "go.opentelemetry.io/otel/semconv/v1.4.0" |
||||||
|
"go.opentelemetry.io/otel/trace" |
||||||
|
"gorm.io/gorm/logger" |
||||||
|
"strings" |
||||||
|
"time" |
||||||
|
) |
||||||
|
|
||||||
|
type Logger struct { |
||||||
|
*log.Helper |
||||||
|
} |
||||||
|
|
||||||
|
func (l *Logger) LogMode(level logger.LogLevel) logger.Interface { |
||||||
|
return l |
||||||
|
} |
||||||
|
|
||||||
|
func (l *Logger) Info(ctx context.Context, s string, i ...interface{}) { |
||||||
|
l.WithContext(ctx).Info(s, i) |
||||||
|
} |
||||||
|
|
||||||
|
func (l *Logger) Warn(ctx context.Context, s string, i ...interface{}) { |
||||||
|
l.WithContext(ctx).Warn(s, i) |
||||||
|
} |
||||||
|
|
||||||
|
func (l *Logger) Error(ctx context.Context, s string, i ...interface{}) { |
||||||
|
l.WithContext(ctx).Error(s, i) |
||||||
|
} |
||||||
|
|
||||||
|
func (l *Logger) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) { |
||||||
|
sql, rowsAffected := fc() |
||||||
|
MysqlTracer(ctx, begin, sql, int(rowsAffected)) |
||||||
|
l.WithContext(ctx).Infof("[%s]db trace sql:%s, rowsAffected:%d, err:%v", begin, sql, rowsAffected, err) |
||||||
|
} |
||||||
|
|
||||||
|
func MysqlTracer(ctx context.Context, begin time.Time, sql string, rowsAffected int) { |
||||||
|
tracer := otel.Tracer("Mysql") |
||||||
|
kind := trace.SpanKindInternal |
||||||
|
strs := strings.Split(sql, " ") |
||||||
|
ctx, span := tracer.Start(ctx, |
||||||
|
strs[0], |
||||||
|
trace.WithSpanKind(kind), |
||||||
|
trace.WithTimestamp(begin), |
||||||
|
) |
||||||
|
var attrs []attribute.KeyValue |
||||||
|
attrs = append(attrs, semconv.DBSystemMySQL) |
||||||
|
attrs = append(attrs, semconv.DBStatementKey.String(sql)) |
||||||
|
attrs = append(attrs, attribute.Int("db.rows_affected", rowsAffected)) |
||||||
|
span.SetAttributes(attrs...) |
||||||
|
span.End(trace.WithTimestamp(time.Now())) |
||||||
|
} |
||||||
Loading…
Reference in new issue