From 8212c93c6f59a5c4f6f16493b408aa24cba828c6 Mon Sep 17 00:00:00 2001
From: chenjian2664 <chenjian2664@gmail.com>
Date: Sun, 28 Sep 2025 01:15:42 +0800
Subject: [PATCH] Fix batch size condition in AsyncResultIterator

Previously, rows were not handled in batch
---
 .../src/main/java/io/trino/jdbc/AsyncResultIterator.java        | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/client/trino-jdbc/src/main/java/io/trino/jdbc/AsyncResultIterator.java b/client/trino-jdbc/src/main/java/io/trino/jdbc/AsyncResultIterator.java
index 24fa9d1ebc9..cf72441ebb2 100644
--- a/client/trino-jdbc/src/main/java/io/trino/jdbc/AsyncResultIterator.java
+++ b/client/trino-jdbc/src/main/java/io/trino/jdbc/AsyncResultIterator.java
@@ -75,7 +75,7 @@ public class AsyncResultIterator
                     warningsManager.addWarnings(results.getWarnings());
                     for (List<Object> row : client.currentRows()) {
                         rowQueue.put(row);
-                        if (rowsProcessed++ % BATCH_SIZE == 0) {
+                        if (++rowsProcessed == BATCH_SIZE) {
                             semaphore.release(rowsProcessed);
                             rowsProcessed = 0;
                         }