feat: add partitionId to StreamProcessor context

(cherry picked from commit f99ed0d)
camunda · Sep 24, 2021 · 08819dd · 08819dd
1 parent 21e6331
commit 08819dd
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/engine/src/main/java/io/camunda/zeebe/engine/processing/streamprocessor/StreamProcessor.java b/engine/src/main/java/io/camunda/zeebe/engine/processing/streamprocessor/StreamProcessor.java
@@ -31,6 +31,7 @@
 import java.time.Duration;
 import java.util.HashSet;
 import java.util.List;
+import java.util.Map;
 import java.util.Set;
 import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.function.Function;
@@ -97,6 +98,13 @@ public static StreamProcessorBuilder builder() {
     return new StreamProcessorBuilder();
   }
 
+  @Override
+  protected Map<String, String> createContext() {
+    final var context = super.createContext();
+    context.put(ACTOR_PROP_PARTITION_ID, Integer.toString(partitionId));
+    return context;
+  }
+
   @Override
   public String getName() {
     return actorName;