@dbcube/core 3.0.14 → 3.0.16

package/dist/index.cjs

@@ -768,39 +768,9 @@ var net = __toESM(require("net"));
 var import_child_process2 = require("child_process");
 var globalTcpServers = /* @__PURE__ */ new Map();
 var globalTcpConnections = /* @__PURE__ */ new Map();
-var connectionPools = /* @__PURE__ */ new Map();
-var MAX_POOL_SIZE = 3;
-var LRUCache = class {
-  cache = /* @__PURE__ */ new Map();
-  maxSize;
-  constructor(maxSize) {
-    this.maxSize = maxSize;
-  }
-  get(key) {
-    if (this.cache.has(key)) {
-      const value = this.cache.get(key);
-      this.cache.delete(key);
-      this.cache.set(key, value);
-      return value;
-    }
-    return void 0;
-  }
-  set(key, value) {
-    if (this.cache.has(key)) {
-      this.cache.delete(key);
-    } else if (this.cache.size >= this.maxSize) {
-      const firstKey = this.cache.keys().next().value;
-      if (firstKey !== void 0) {
-        this.cache.delete(firstKey);
-      }
-    }
-    this.cache.set(key, value);
-  }
-  size() {
-    return this.cache.size;
-  }
-};
-var queryCache = new LRUCache(500);
+var queryCache = /* @__PURE__ */ new Map();
+var cacheSize = 0;
+var MAX_CACHE_SIZE = 500;
 var QueryEngine = class {
   name;
   config;
@@ -919,7 +889,6 @@ var QueryEngine = class {
     if (!serverInfo || !serverInfo.process || serverInfo.process.killed) {
       await this.startTcpServer();
       await this.waitForServerReady();
-      await this.initializeConnectionPool();
     }
     return this.sendTcpRequestFast(args);
   }
@@ -955,30 +924,15 @@ var QueryEngine = class {
   sleep(ms) {
     return new Promise((resolve5) => setTimeout(resolve5, ms));
   }
-  async initializeConnectionPool() {
-    console.log("\u{1F3CA} Initializing connection pool...");
-    if (!connectionPools.has(this.connectionId)) {
-      connectionPools.set(this.connectionId, []);
-    }
-    const pool = connectionPools.get(this.connectionId);
-    for (let i = 0; i < MAX_POOL_SIZE; i++) {
-      try {
-        const connection = await this.createPooledConnection();
-        pool.push(connection);
-        console.log(`\u2705 Pool connection ${i + 1}/${MAX_POOL_SIZE} created`);
-      } catch (error) {
-        console.error(`\u274C Failed to create pool connection ${i + 1}:`, error);
-      }
-    }
-    console.log(`\u{1F680} Connection pool ready with ${pool.length} connections`);
-  }
   getCachedDML(dmlJson) {
-    const cached = queryCache.get(dmlJson);
-    if (cached) {
-      return cached;
+    if (queryCache.has(dmlJson)) {
+      return queryCache.get(dmlJson);
     }
     const parsed = JSON.parse(dmlJson);
-    queryCache.set(dmlJson, parsed);
+    if (cacheSize < MAX_CACHE_SIZE) {
+      queryCache.set(dmlJson, parsed);
+      cacheSize++;
+    }
     return parsed;
   }
   async startTcpServer() {
@@ -1002,6 +956,7 @@ var QueryEngine = class {
         if (output.includes("TCP Server listening on")) {
           if (!started) {
             started = true;
+            clearTimeout(timeout);
             globalTcpServers.set(this.connectionId, {
               port: this.tcpPort,
               process: serverProcess
@@ -1018,6 +973,7 @@ var QueryEngine = class {
       });
       serverProcess.on("error", (error) => {
         if (!started) {
+          clearTimeout(timeout);
           reject(error);
         }
       });
@@ -1041,6 +997,7 @@ var QueryEngine = class {
         if (!dmlJson) {
           if (!isResolved) {
             isResolved = true;
+            clearTimeout(timeout);
             client.destroy();
             reject(new Error("No DML found in arguments"));
           }
@@ -1057,6 +1014,7 @@ var QueryEngine = class {
         const match = responseBuffer.match(/PROCESS_RESPONSE:(\{.*\})/);
         if (match && !isResolved) {
           isResolved = true;
+          clearTimeout(timeout);
           client.destroy();
           try {
             const response = JSON.parse(match[1]);
@@ -1073,11 +1031,13 @@ var QueryEngine = class {
       client.on("error", (error) => {
         if (!isResolved) {
           isResolved = true;
+          clearTimeout(timeout);
           reject(error);
         }
       });
       client.on("close", () => {
         if (!isResolved) {
+          clearTimeout(timeout);
           if (!responseBuffer.includes("PROCESS_RESPONSE:")) {
             reject(new Error("Connection closed without valid response"));
           }
@@ -1086,24 +1046,18 @@ var QueryEngine = class {
     });
   }
   async sendTcpRequestFast(args) {
-    if (!connectionPools.has(this.connectionId)) {
-      return await this.createAndPoolConnection(args);
-    }
-    const pool = connectionPools.get(this.connectionId);
-    if (pool.length > 0) {
-      const connection = pool.pop();
+    const existingConnection = globalTcpConnections.get(this.connectionId);
+    if (existingConnection && existingConnection.readyState === "open") {
       try {
-        const result = await this.sendUltraFast(connection, args);
-        pool.push(connection);
-        return result;
+        return await this.sendOnExistingConnection(existingConnection, args);
       } catch (error) {
-        connection.destroy();
-        return await this.createAndPoolConnection(args);
+        globalTcpConnections.delete(this.connectionId);
+        existingConnection.destroy();
       }
     }
-    return await this.createAndPoolConnection(args);
+    return await this.createPersistentConnection(args);
   }
-  async sendUltraFast(connection, 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;
@@ -1113,96 +1067,20 @@ var QueryEngine = class {
       }
       let responseBuffer = "";
       let isResolved = false;
-      const onData = (data) => {
-        responseBuffer += data.toString();
-        const match = responseBuffer.match(/PROCESS_RESPONSE:(\{.*\})/);
-        if (match && !isResolved) {
-          isResolved = true;
-          connection.off("data", onData);
-          connection.off("error", onError);
-          connection.off("close", onClose);
-          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;
-          connection.off("data", onData);
-          connection.off("error", onError);
-          connection.off("close", onClose);
-          reject(error);
-        }
-      };
-      const onClose = () => {
+      const timeout = setTimeout(() => {
         if (!isResolved) {
           isResolved = true;
-          connection.off("data", onData);
-          connection.off("error", onError);
-          connection.off("close", onClose);
-          connection.off("close", onClose);
-          reject(new Error("Connection closed unexpectedly"));
+          reject(new Error("TCP request timeout"));
         }
-      };
-      connection.on("data", onData);
-      connection.on("error", onError);
-      connection.on("close", onClose);
-      const command = `{"action":"execute","dml":${dmlJson}}`;
-      connection.write(command);
-    });
-  }
-  async createAndPoolConnection(args) {
-    const connection = await this.createPooledConnection();
-    if (!connectionPools.has(this.connectionId)) {
-      connectionPools.set(this.connectionId, []);
-    }
-    const result = await this.sendUltraFast(connection, args);
-    const pool = connectionPools.get(this.connectionId);
-    if (pool.length < MAX_POOL_SIZE) {
-      pool.push(connection);
-    } else {
-      connection.destroy();
-    }
-    return result;
-  }
-  async createPooledConnection() {
-    return new Promise((resolve5, reject) => {
-      const client = new net.Socket();
-      client.setNoDelay(true);
-      client.setKeepAlive(true, 0);
-      client.connect(this.tcpPort, "127.0.0.1", () => {
-        resolve5(client);
-      });
-      client.on("error", reject);
-    });
-  }
-  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;
-      }
-      const dmlParsed = this.getCachedDML(dmlJson);
-      let responseBuffer = "";
-      let isResolved = false;
+      }, 100);
       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);
-          connection.off("close", onClose);
           try {
             const response = JSON.parse(match[1]);
             resolve5({
@@ -1218,24 +1096,14 @@ var QueryEngine = class {
       const onError = (error) => {
         if (!isResolved) {
           isResolved = true;
+          clearTimeout(timeout);
           connection.off("data", onData);
           connection.off("error", onError);
-          connection.off("close", onClose);
           reject(error);
         }
       };
-      const onClose = () => {
-        if (!isResolved) {
-          isResolved = true;
-          connection.off("data", onData);
-          connection.off("error", onError);
-          connection.off("close", onClose);
-          reject(new Error("Connection closed unexpectedly"));
-        }
-      };
       connection.on("data", onData);
       connection.on("error", onError);
-      connection.on("close", onClose);
       const command = {
         action: "execute",
         dml: dmlJson