Gcsio migration (#29360) (#29465)

apache · Nov 17, 2023 · 65ddc34 · 65ddc34
1 parent a4d6135
commit 65ddc34
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.../flink/translation/wrappers/streaming/io/source/unbounded/FlinkUnboundedSourceReader.java b/.../flink/translation/wrappers/streaming/io/source/unbounded/FlinkUnboundedSourceReader.java
@@ -126,6 +126,9 @@ public InputStatus pollNext(ReaderOutput<WindowedValue<ValueWithRecordId<T>>> ou
     if (reader != null) {
       emitRecord(reader, output);
       return InputStatus.MORE_AVAILABLE;
+    } else if (noMoreSplits()) {
+      LOG.trace("No more splits.");
+      return InputStatus.END_OF_INPUT;
     } else {
       LOG.trace("No data available for now.");
       return InputStatus.NOTHING_AVAILABLE;