Pass taskListActivitiesPerSecond down to ActivityWorker

cadence-workflow · Oct 23, 2020 · 42eee7a · 42eee7a
1 parent 144c32c
commit 42eee7a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/main/java/com/uber/cadence/worker/Worker.java b/src/main/java/com/uber/cadence/worker/Worker.java
@@ -141,12 +141,13 @@ private static SingleWorkerOptions toActivityOptions(
             .put(MetricsTag.TASK_LIST, taskList)
             .build();
     return new SingleWorkerOptions.Builder()
-        .setDataConverter(options.getDataConverter())
         .setIdentity(options.getIdentity())
+        .setDataConverter(options.getDataConverter())
+        .setTaskExecutorThreadPoolSize(options.getMaxConcurrentActivityExecutionSize())
+        .setTaskListActivitiesPerSecond(options.getTaskListActivitiesPerSecond())
         .setPollerOptions(options.getActivityPollerOptions())
         .setReportCompletionRetryOptions(options.getReportActivityCompletionRetryOptions())
         .setReportFailureRetryOptions(options.getReportActivityFailureRetryOptions())
-        .setTaskExecutorThreadPoolSize(options.getMaxConcurrentActivityExecutionSize())
         .setMetricsScope(options.getMetricsScope().tagged(tags))
         .setEnableLoggingInReplay(options.getEnableLoggingInReplay())
         .setContextPropagators(contextPropagators)

diff --git a/src/main/java/com/uber/cadence/worker/WorkerOptions.java b/src/main/java/com/uber/cadence/worker/WorkerOptions.java
@@ -281,6 +281,10 @@ public int getMaxConcurrentLocalActivityExecutionSize() {
     return maxConcurrentLocalActivityExecutionSize;
   }
 
+  public double getTaskListActivitiesPerSecond() {
+    return taskListActivitiesPerSecond;
+  }
+
   public PollerOptions getActivityPollerOptions() {
     return activityPollerOptions;
   }