From 01314bf6509ae97406771771a6a51ea218543eda Mon Sep 17 00:00:00 2001
From: Noel Kwan <noelkwan1998@gmail.com>
Date: Tue, 8 Oct 2024 22:41:49 +0800
Subject: [PATCH] consume all remaining chunks

---
 src/batch/src/executor/postgres_query.rs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/batch/src/executor/postgres_query.rs b/src/batch/src/executor/postgres_query.rs
index 6156cbb7c10a7..404b0c45f61cf 100644
--- a/src/batch/src/executor/postgres_query.rs
+++ b/src/batch/src/executor/postgres_query.rs
@@ -145,6 +145,10 @@ impl PostgresQueryExecutor {
                 yield chunk;
             }
         }
+        if let Some(chunk) = builder.consume_all() {
+            yield chunk;
+        }
+        return Ok(());
     }
 }