From 9390ab3c6c40d0a1cb3a14fc7e75baa75d2f2c10 Mon Sep 17 00:00:00 2001
From: Nybkox <nybkox@gmail.com>
Date: Tue, 29 Apr 2025 01:32:55 +0200
Subject: [PATCH] fix: do not skip non-returnin statements

---
 plugins/dbgate-plugin-duckdb/src/backend/driver.js | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/plugins/dbgate-plugin-duckdb/src/backend/driver.js b/plugins/dbgate-plugin-duckdb/src/backend/driver.js
index 7be4ebfda..5e8626eb8 100644
--- a/plugins/dbgate-plugin-duckdb/src/backend/driver.js
+++ b/plugins/dbgate-plugin-duckdb/src/backend/driver.js
@@ -73,16 +73,16 @@ const driver = {
           duckdb.StatementType.LOGICAL_PLAN,
         ];
 
-        if (!returningStatementTypes.includes(stmt.statementType)) {
-          continue;
-        }
-
         const result = await stmt.stream();
         let hasSentColumns = false;
 
         while (true) {
           const chunk = await result.fetchChunk();
 
+          if (!returningStatementTypes.includes(stmt.statementType)) {
+            break;
+          }
+
           if (!chunk || chunk.rowCount === 0) {
             break;
           }