Skip to content

Commit 6d0c6e7

Browse files
authored
Add wf type tag in decider metrics scope (#499)
1 parent f139c7a commit 6d0c6e7

File tree

3 files changed

+12
-2
lines changed

3 files changed

+12
-2
lines changed

src/main/java/com/uber/cadence/internal/replay/ReplayDecider.java

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,11 @@
2929
import com.uber.cadence.WorkflowExecutionSignaledEventAttributes;
3030
import com.uber.cadence.WorkflowExecutionStartedEventAttributes;
3131
import com.uber.cadence.WorkflowQuery;
32+
import com.uber.cadence.WorkflowType;
3233
import com.uber.cadence.common.RetryOptions;
3334
import com.uber.cadence.internal.common.OptionsUtils;
3435
import com.uber.cadence.internal.common.Retryer;
36+
import com.uber.cadence.internal.metrics.MetricsTag;
3537
import com.uber.cadence.internal.metrics.MetricsType;
3638
import com.uber.cadence.internal.replay.HistoryHelper.DecisionEvents;
3739
import com.uber.cadence.internal.replay.HistoryHelper.DecisionEventsIterator;
@@ -43,6 +45,7 @@
4345
import com.uber.cadence.workflow.Functions;
4446
import com.uber.m3.tally.Scope;
4547
import com.uber.m3.tally.Stopwatch;
48+
import com.uber.m3.util.ImmutableMap;
4649
import java.time.Duration;
4750
import java.util.ArrayList;
4851
import java.util.Iterator;
@@ -83,14 +86,18 @@ class ReplayDecider implements Decider {
8386
ReplayDecider(
8487
IWorkflowService service,
8588
String domain,
89+
WorkflowType workflowType,
8690
ReplayWorkflow workflow,
8791
DecisionsHelper decisionsHelper,
8892
SingleWorkerOptions options,
8993
BiFunction<LocalActivityWorker.Task, Duration, Boolean> laTaskPoller) {
9094
this.service = service;
9195
this.workflow = workflow;
9296
this.decisionsHelper = decisionsHelper;
93-
this.metricsScope = options.getMetricsScope();
97+
this.metricsScope =
98+
options
99+
.getMetricsScope()
100+
.tagged(ImmutableMap.of(MetricsTag.WORKFLOW_TYPE, workflowType.getName()));
94101
PollForDecisionTaskResponse decisionTask = decisionsHelper.getTask();
95102

96103
startedEvent =

src/main/java/com/uber/cadence/internal/replay/ReplayDecisionTaskHandler.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -278,6 +278,7 @@ private Decider createDecider(PollForDecisionTaskResponse decisionTask) throws E
278278
}
279279
DecisionsHelper decisionsHelper = new DecisionsHelper(decisionTask);
280280
ReplayWorkflow workflow = workflowFactory.getWorkflow(workflowType);
281-
return new ReplayDecider(service, domain, workflow, decisionsHelper, options, laTaskPoller);
281+
return new ReplayDecider(
282+
service, domain, workflowType, workflow, decisionsHelper, options, laTaskPoller);
282283
}
283284
}

src/test/java/com/uber/cadence/internal/replay/ReplayDeciderCacheTests.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import com.uber.cadence.PollForDecisionTaskResponse;
2828
import com.uber.cadence.WorkflowExecution;
2929
import com.uber.cadence.WorkflowQuery;
30+
import com.uber.cadence.WorkflowType;
3031
import com.uber.cadence.internal.metrics.MetricsTag;
3132
import com.uber.cadence.internal.metrics.MetricsType;
3233
import com.uber.cadence.internal.metrics.NoopScope;
@@ -256,6 +257,7 @@ private ReplayDecider createFakeDecider(PollForDecisionTaskResponse response) {
256257
return new ReplayDecider(
257258
new TestWorkflowService(),
258259
"domain",
260+
new WorkflowType().setName("workflow"),
259261
new ReplayWorkflow() {
260262
@Override
261263
public void start(HistoryEvent event, DecisionContext context) {}

0 commit comments

Comments
 (0)