From f2bd08f02a0efc5682460b9b3106f86ba7cd295a Mon Sep 17 00:00:00 2001
From: "SPRINX0\\prochazka" <jan.prochazka@sprinx.com>
Date: Tue, 25 Feb 2025 17:59:31 +0100
Subject: [PATCH] mixed mongoDB export #1044

---
 .../dbgate-plugin-mongo/src/backend/driver.js | 27 +++++++++++++------
 1 file changed, 19 insertions(+), 8 deletions(-)

diff --git a/plugins/dbgate-plugin-mongo/src/backend/driver.js b/plugins/dbgate-plugin-mongo/src/backend/driver.js
index b810d1673..23aca12b6 100644
--- a/plugins/dbgate-plugin-mongo/src/backend/driver.js
+++ b/plugins/dbgate-plugin-mongo/src/backend/driver.js
@@ -260,14 +260,25 @@ const driver = {
       highWaterMark: 100,
     });
 
-    exprValue
-      .forEach((row) => pass.write(transformMongoData(row)))
-      .then(() => {
-        pass.end();
-        // pass.end(() => {
-        //   pass.emit('end');
-        // })
-      });
+    const cursorStream = exprValue.stream();
+
+    cursorStream.on('data', (row) => {
+      pass.write(transformMongoData(row));
+    });
+
+    // Called once the cursor is fully read
+    cursorStream.on('end', () => {
+      pass.emit('end');
+    });
+
+    // exprValue
+    //   .forEach((row) => pass.write(transformMongoData(row)))
+    //   .then(() => {
+    //     pass.end();
+    //     // pass.end(() => {
+    //     //   pass.emit('end');
+    //     // })
+    //   });
 
     return pass;
     // return directly stream without header row