@dbcube/core 3.0.7 → 3.0.8

package/dist/index.cjs

@@ -767,6 +767,7 @@ var import_module2 = require("module");
 var net = __toESM(require("net"));
 var import_child_process2 = require("child_process");
 var globalTcpServers = /* @__PURE__ */ new Map();
+var globalTcpConnections = /* @__PURE__ */ new Map();
 var QueryEngine = class {
   name;
   config;
@@ -885,10 +886,10 @@ var QueryEngine = class {
     if (!serverInfo || !serverInfo.process || serverInfo.process.killed) {
       console.log("\u{1F680} Starting TCP server for ultra-fast queries...");
       await this.startTcpServer();
+      await this.waitForServerReady();
     }
-    await this.waitForServerReady();
     console.log("\u26A1 Using TCP server (ultra-fast)");
-    return this.sendTcpRequest(args);
+    return this.sendTcpRequestFast(args);
   }
   async waitForServerReady() {
     const maxRetries = 10;
@@ -1036,6 +1037,137 @@ var QueryEngine = class {
       });
     });
   }
+  async sendTcpRequestFast(args) {
+    const existingConnection = globalTcpConnections.get(this.connectionId);
+    if (existingConnection && existingConnection.readyState === "open") {
+      try {
+        return await this.sendOnExistingConnection(existingConnection, args);
+      } catch (error) {
+        globalTcpConnections.delete(this.connectionId);
+        existingConnection.destroy();
+      }
+    }
+    return await this.createPersistentConnection(args);
+  }
+  async sendOnExistingConnection(connection, args) {
+    return new Promise((resolve5, reject) => {
+      const dmlIndex = args.findIndex((arg) => arg === "--dml");
+      const dmlJson = dmlIndex !== -1 ? args[dmlIndex + 1] : null;
+      if (!dmlJson) {
+        reject(new Error("No DML found in arguments"));
+        return;
+      }
+      let responseBuffer = "";
+      let isResolved = false;
+      const timeout = setTimeout(() => {
+        if (!isResolved) {
+          isResolved = true;
+          reject(new Error("TCP request timeout"));
+        }
+      }, 5e3);
+      const onData = (data) => {
+        responseBuffer += data.toString();
+        const match = responseBuffer.match(/PROCESS_RESPONSE:(\{.*\})/);
+        if (match && !isResolved) {
+          isResolved = true;
+          clearTimeout(timeout);
+          connection.off("data", onData);
+          connection.off("error", onError);
+          try {
+            const response = JSON.parse(match[1]);
+            resolve5({
+              status: response.status,
+              message: response.message,
+              data: response.data
+            });
+          } catch (error) {
+            reject(new Error("Failed to parse TCP response"));
+          }
+        }
+      };
+      const onError = (error) => {
+        if (!isResolved) {
+          isResolved = true;
+          clearTimeout(timeout);
+          connection.off("data", onData);
+          connection.off("error", onError);
+          reject(error);
+        }
+      };
+      connection.on("data", onData);
+      connection.on("error", onError);
+      const command = {
+        action: "execute",
+        dml: dmlJson
+      };
+      connection.write(JSON.stringify(command));
+    });
+  }
+  async createPersistentConnection(args) {
+    return new Promise((resolve5, reject) => {
+      const client = new net.Socket();
+      let responseBuffer = "";
+      let isResolved = false;
+      client.setNoDelay(true);
+      client.setKeepAlive(true, 6e4);
+      const timeout = setTimeout(() => {
+        if (!isResolved) {
+          isResolved = true;
+          client.destroy();
+          reject(new Error("TCP connection timeout"));
+        }
+      }, 5e3);
+      client.connect(this.tcpPort, "127.0.0.1", () => {
+        clearTimeout(timeout);
+        const dmlIndex = args.findIndex((arg) => arg === "--dml");
+        const dmlJson = dmlIndex !== -1 ? args[dmlIndex + 1] : null;
+        if (!dmlJson) {
+          if (!isResolved) {
+            isResolved = true;
+            client.destroy();
+            reject(new Error("No DML found in arguments"));
+          }
+          return;
+        }
+        globalTcpConnections.set(this.connectionId, client);
+        const command = {
+          action: "execute",
+          dml: dmlJson
+        };
+        client.write(JSON.stringify(command));
+      });
+      client.on("data", (data) => {
+        responseBuffer += data.toString();
+        const match = responseBuffer.match(/PROCESS_RESPONSE:(\{.*\})/);
+        if (match && !isResolved) {
+          isResolved = true;
+          try {
+            const response = JSON.parse(match[1]);
+            resolve5({
+              status: response.status,
+              message: response.message,
+              data: response.data
+            });
+          } catch (error) {
+            reject(new Error("Failed to parse TCP response"));
+          }
+        }
+      });
+      client.on("error", (error) => {
+        if (!isResolved) {
+          isResolved = true;
+          globalTcpConnections.delete(this.connectionId);
+          reject(error);
+        }
+      });
+      client.on("close", () => {
+        globalTcpConnections.delete(this.connectionId);
+        if (!isResolved && !responseBuffer.includes("PROCESS_RESPONSE:")) {
+          reject(new Error("Connection closed without valid response"));
+        }
+      });
+    });
+  }
   async createProcess(binary, args) {
     await this.initializeBinary();
     if (!this.binary) {
@@ -1124,6 +1256,12 @@ var QueryEngine = class {
     });
   }
   async disconnect() {
+    const connection = globalTcpConnections.get(this.connectionId);
+    if (connection && connection.readyState === "open") {
+      connection.write(JSON.stringify({ action: "disconnect" }));
+      connection.destroy();
+      globalTcpConnections.delete(this.connectionId);
+    }
     const serverInfo = globalTcpServers.get(this.connectionId);
     if (serverInfo && serverInfo.process && !serverInfo.process.killed) {
       console.log("\u{1F50C} Stopping TCP server...");
@@ -1131,7 +1269,7 @@ var QueryEngine = class {
       globalTcpServers.delete(this.connectionId);
       return {
         status: 200,
-        message: "TCP server stopped",
+        message: "TCP server and connections stopped",
         data: null
       };
     }