diff --git a/src/main/java/com/uber/cadence/internal/logging/LoggerTag.java b/src/main/java/com/uber/cadence/internal/logging/LoggerTag.java index f31c0fbe4..cdd3aa2dd 100644 --- a/src/main/java/com/uber/cadence/internal/logging/LoggerTag.java +++ b/src/main/java/com/uber/cadence/internal/logging/LoggerTag.java @@ -25,4 +25,5 @@ public class LoggerTag { public static final String TASK_LIST = "TaskList"; public static final String WORKFLOW_ID = "WorkflowID"; public static final String WORKFLOW_TYPE = "WorkflowType"; + public static final String ATTEMPT = "Attempt"; } diff --git a/src/main/java/com/uber/cadence/internal/worker/ActivityWorker.java b/src/main/java/com/uber/cadence/internal/worker/ActivityWorker.java index 9e83156d5..767f63ea1 100644 --- a/src/main/java/com/uber/cadence/internal/worker/ActivityWorker.java +++ b/src/main/java/com/uber/cadence/internal/worker/ActivityWorker.java @@ -144,6 +144,7 @@ public void handle(PollForActivityTaskResponse task) throws Exception { MDC.put(LoggerTag.ACTIVITY_TYPE, task.getActivityType().getName()); MDC.put(LoggerTag.WORKFLOW_ID, task.getWorkflowExecution().getWorkflowId()); MDC.put(LoggerTag.RUN_ID, task.getWorkflowExecution().getRunId()); + MDC.put(LoggerTag.ATTEMPT, String.valueOf(task.getAttempt())); propagateContext(task); Span span = spanFactory.spanForExecuteActivity(task); @@ -175,6 +176,7 @@ public void handle(PollForActivityTaskResponse task) throws Exception { MDC.remove(LoggerTag.ACTIVITY_TYPE); MDC.remove(LoggerTag.WORKFLOW_ID); MDC.remove(LoggerTag.RUN_ID); + MDC.remove(LoggerTag.ATTEMPT); unsetCurrentContext(); } }