whatcanGOwrong

go/pkg/mod/github.com/jackc/pgx/v4@v4.18.3/log/logrusadapter/adapter.go

@@ -0,0 +1,42 @@
+// Package logrusadapter provides a logger that writes to a github.com/sirupsen/logrus.Logger
+// log.
+package logrusadapter
+
+import (
+	"context"
+
+	"github.com/jackc/pgx/v4"
+	"github.com/sirupsen/logrus"
+)
+
+type Logger struct {
+	l logrus.FieldLogger
+}
+
+func NewLogger(l logrus.FieldLogger) *Logger {
+	return &Logger{l: l}
+}
+
+func (l *Logger) Log(ctx context.Context, level pgx.LogLevel, msg string, data map[string]interface{}) {
+	var logger logrus.FieldLogger
+	if data != nil {
+		logger = l.l.WithFields(data)
+	} else {
+		logger = l.l
+	}
+
+	switch level {
+	case pgx.LogLevelTrace:
+		logger.WithField("PGX_LOG_LEVEL", level).Debug(msg)
+	case pgx.LogLevelDebug:
+		logger.Debug(msg)
+	case pgx.LogLevelInfo:
+		logger.Info(msg)
+	case pgx.LogLevelWarn:
+		logger.Warn(msg)
+	case pgx.LogLevelError:
+		logger.Error(msg)
+	default:
+		logger.WithField("INVALID_PGX_LOG_LEVEL", level).Error(msg)
+	}
+}