[hotfix] Fix compilation error due to constructor removal of MockSour…

…ceReader (#89)
apache · May 17, 2024 · 6b38e11 · 6b38e11
1 parent 6e6e1fc
commit 6b38e11
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/org/apache/flink/benchmark/MultipleInputBenchmark.java b/src/main/java/org/apache/flink/benchmark/MultipleInputBenchmark.java
@@ -167,7 +167,7 @@ public static void reset() {
         @Override
         public SourceReader<Integer, MockSourceSplit> createReader(
                 SourceReaderContext readerContext) {
-            return new MockSourceReader(true, true) {
+            return new MockSourceReader(MockSourceReader.WaitingForSplits.WAIT_UNTIL_ALL_SPLITS_ASSIGNED, true) {
                 @Override
                 public InputStatus pollNext(ReaderOutput<Integer> sourceOutput) {
                     if (canFinish.isDone() && !canFinish.isCompletedExceptionally()) {