From 02bcd0432a09c259f7740c9a15f33104f5c1920d Mon Sep 17 00:00:00 2001 From: Kevin Franklin Kim Date: Mon, 20 Jun 2022 18:32:29 +0200 Subject: [PATCH] feat: extend helper --- integration/temporal/log.go | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/integration/temporal/log.go b/integration/temporal/log.go index 7987808..8713769 100644 --- a/integration/temporal/log.go +++ b/integration/temporal/log.go @@ -36,12 +36,14 @@ func Debug(ctx workflow.Context, msg string, fields ...zap.Field) { workflow.GetLogger(ctx).Debug(msg, keyvals...) } -func GetWorkflowLogger(ctx workflow.Context) tlog.Logger { - return workflow.GetLogger(ctx) +func GetWorkflowLogger(ctx workflow.Context, fields ...zap.Field) tlog.Logger { + l := workflow.GetLogger(ctx) + return LoggerWith(l, fields...) } -func GetActivityLogger(ctx context.Context) tlog.Logger { - return activity.GetLogger(ctx) +func GetActivityLogger(ctx context.Context, fields ...zap.Field) tlog.Logger { + l := activity.GetLogger(ctx) + return LoggerWith(l, fields...) } func LoggerWith(logger tlog.Logger, fields ...zap.Field) tlog.Logger {