Further review comments

apache · Dec 19, 2023 · a3e78ba · a3e78ba
1 parent 83044f9
commit a3e78ba
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/.../flink/src/main/java/org/apache/iceberg/flink/source/reader/IcebergSourceSplitReader.java b/.../flink/src/main/java/org/apache/iceberg/flink/source/reader/IcebergSourceSplitReader.java
@@ -149,6 +149,12 @@ public void handleSplitsChanges(SplitsChange<IcebergSourceSplit> splitsChange) {
     metrics.incrementAssignedBytes(calculateBytes(splitsChange));
   }
 
+  /**
+   * WakeUp is called by Flink to unblock a reader blocked in {@link #fetch()} before calling {@link
+   * #handleSplitsChanges(SplitsChange)} or {@link #pauseOrResumeSplits(Collection, Collection)} to
+   * update the splits or splits state. After {@link #wakeUp()} the specific method will be called
+   * before calling the {@link #fetch()} again.
+   */
   @Override
   public void wakeUp() {
     LOG.info("WakeUp reader {}", indexOfSubtask);

diff --git a/...rc/test/java/org/apache/iceberg/flink/source/TestIcebergSourceWithWatermarkExtractor.java b/...rc/test/java/org/apache/iceberg/flink/source/TestIcebergSourceWithWatermarkExtractor.java
@@ -381,7 +381,7 @@ protected IcebergSource<RowData> source() {
         .project(TestFixtures.TS_SCHEMA)
         .splitSize(100L)
         .streaming(true)
-        .monitorInterval(Duration.ofMillis(100))
+        .monitorInterval(Duration.ofMillis(10))
         .streamingStartingStrategy(StreamingStartingStrategy.TABLE_SCAN_THEN_INCREMENTAL)
         // Read in single row batches, to have more batches for testing
         .flinkConfig(new Configuration().set(SOURCE_READER_FETCH_BATCH_RECORD_COUNT, 1))

diff --git a/...nk/src/test/java/org/apache/iceberg/flink/source/reader/TestIcebergSourceSplitReader.java b/...nk/src/test/java/org/apache/iceberg/flink/source/reader/TestIcebergSourceSplitReader.java
@@ -136,7 +136,10 @@ public void testWakeup() throws Exception {
 
     // Wakeup the reading of the split, and check the result size
     reader.wakeUp();
-    Awaitility.await().atMost(2, TimeUnit.SECONDS).until(() -> gotEmptyResult.get());
+    Awaitility.await()
+        .pollInterval(10, TimeUnit.MILLISECONDS)
+        .atMost(2, TimeUnit.SECONDS)
+        .until(() -> gotEmptyResult.get());
 
     // Unpause and read the data in th next fetch
     reader.pauseOrResumeSplits(ImmutableList.of(), ImmutableList.of(split));