Add overrideBuilder to DataflowRunner.java

apache · Jun 7, 2024 · 0081190 · 0081190
1 parent 748c345
commit 0081190
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/...le-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/DataflowRunner.java b/...le-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/DataflowRunner.java
@@ -697,6 +697,11 @@ private List<PTransformOverride> getOverrides(boolean streaming) {
             PTransformOverride.of(
                 PTransformMatchers.classEqualTo(ParDo.SingleOutput.class),
                 new PrimitiveParDoSingleFactory()));
+
+    overridesBuilder.add(
+        PTransformOverride.of(
+            PTransformMatchers.classEqualTo(RedistributeByKey.class),
+            new RedistributeByKeyOverrideFactory()));
 
     if (streaming) {
       // For update compatibility, always use a Read for Create in streaming mode.