eos set false

TongchengOpenSource · Apr 30, 2024 · 37cdf07 · 37cdf07
1 parent 8d8137c
commit 37cdf07
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.../apache/seatunnel/connectors/seatunnel/starrocks/client/source/StarRocksBeReadClient.java b/.../apache/seatunnel/connectors/seatunnel/starrocks/client/source/StarRocksBeReadClient.java
@@ -91,6 +91,7 @@ public StarRocksBeReadClient(String beNodeInfo, SourceConfig sourceConfig) {
     }
 
     public void openScanner(QueryPartition partition, SeaTunnelRowType seaTunnelRowType) {
+        eos.set(false);
         this.seaTunnelRowType = seaTunnelRowType;
         Set<Long> tabletIds = partition.getTabletIds();
         TScanOpenParams params = new TScanOpenParams();
@@ -139,6 +140,7 @@ public void openScanner(QueryPartition partition, SeaTunnelRowType seaTunnelRowT
     public boolean hasNext() {
         boolean hasNext = false;
         // Arrow data was acquired synchronously during the iterative process
+        System.out.println("eos.get(): " + eos.get());
         if (!eos.get() && (rowBatch == null || !rowBatch.hasNext())) {
             if (rowBatch != null) {
                 readerOffset += rowBatch.getReadRowCount();