feat: 日志类型

logx/gorm.go

+package logx
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+	"time"
+
+	"gorm.io/gorm/logger"
+)
+
+type gormLogger struct {
+	log *slog.Logger
+}
+
+func NewGormLogger(log *slog.Logger) logger.Interface {
+	return &gormLogger{
+		log: log.With("type", "gorm"), // 添加类型
+	}
+}
+
+func (g *gormLogger) LogMode(level logger.LogLevel) logger.Interface {
+	return g
+}
+
+func (g *gormLogger) Info(ctx context.Context, s string, i ...any) {
+	g.log.InfoContext(ctx, fmt.Sprintf(s, i...))
+}
+
+func (g *gormLogger) Warn(ctx context.Context, s string, i ...interface{}) {
+	g.log.WarnContext(ctx, fmt.Sprintf(s, i...))
+}
+
+func (g *gormLogger) Error(ctx context.Context, s string, i ...interface{}) {
+	g.log.ErrorContext(ctx, fmt.Sprintf(s, i...))
+}
+
+func (g *gormLogger) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) {
+	// 执行耗时
+
+	latency := time.Since(begin)
+	sql, rows := fc()
+	if err != nil {
+		g.log.Error("sql trace", "latency", latency.Milliseconds(), "sql", sql, "rows", rows, "err", err)
+	} else {
+		g.log.Debug("sql trace", "latency", latency.Milliseconds(), "sql", sql, "rows", rows)
+	}
+}

restapi/app.go

@@ -29,7 +29,7 @@ func NewApp(appName string, log *slog.Logger) *fiber.App {
 	middlewares := []any{
 		requestid.New(),
 		Recover(),
-		Logger(LoggerConfig{Log: log}),
+		Logger(LoggerConfig{Log: log.With("type", "http")}),
 		Error(),
 		cors.New(),
 	}