Adding more wrappers

Author: Elliot Forbes

error.go

@@ -1,8 +1,17 @@
 package ctxlog
 
-import "context"
+import (
+	"context"
+
+	"github.com/sirupsen/logrus"
+)
 
 func (l *CtxLogger) Error(ctx context.Context, msg string) {
 	fields := convertFieldsToLogrusFields(getFields(ctx))
 	l.Log.WithFields(fields).Error(msg)
 }
+
+func (l *CtxLogger) ErrorFn(ctx context.Context, logFunc logrus.LogFunction) {
+	fields := convertFieldsToLogrusFields(getFields(ctx))
+	l.Log.WithFields(fields).Logger.ErrorFn(logFunc)
+}

info.go

@@ -2,9 +2,16 @@ package ctxlog
 
 import (
 	"context"
+
+	"github.com/sirupsen/logrus"
 )
 
 func (l *CtxLogger) Info(ctx context.Context, msg string) {
 	fields := convertFieldsToLogrusFields(getFields(ctx))
 	l.Log.WithFields(fields).Info(msg)
 }
+
+func (l *CtxLogger) InfoFn(ctx context.Context, logFun logrus.LogFunction) {
+	fields := convertFieldsToLogrusFields(getFields(ctx))
+	l.Log.WithFields(fields).InfoFn(logFun)
+}

warn.go

@@ -1,8 +1,17 @@
 package ctxlog
 
-import "context"
+import (
+	"context"
+
+	"github.com/sirupsen/logrus"
+)
 
 func (l *CtxLogger) Warn(ctx context.Context, msg string) {
 	fields := convertFieldsToLogrusFields(getFields(ctx))
 	l.Log.WithFields(fields).Warn(msg)
 }
+
+func (l *CtxLogger) Warn(ctx context.Context, logFunc logrus.LogFunction) {
+	fields := convertFieldsToLogrusFields(getFields(ctx))
+	l.Log.WithFields(fields).WarnFn(logFunc)
+}