consume all remaining chunks

risingwavelabs · Oct 8, 2024 · 01314bf · 01314bf
1 parent ad7db51
commit 01314bf
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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(());
     }
 }