@dbcube/core 1.0.52 → 1.0.57

package/dist/index.cjs

@@ -593,17 +593,31 @@ var Config = class {
 // src/lib/Engine.ts
 var import_child_process = require("child_process");
 var import_module = require("module");
-var Engine = class {
-  name;
+var Engine = class _Engine {
+  static instances = /* @__PURE__ */ new Map();
+  name = "";
   config;
   arguments;
   binary = null;
-  timeout;
+  timeout = 3e4;
+  isDaemonConnected = false;
+  isConnecting = false;
   constructor(name, timeout = 3e4) {
+    if (_Engine.instances.has(name)) {
+      return _Engine.instances.get(name);
+    }
     this.name = name;
     this.config = this.setConfig(name);
     this.arguments = this.setArguments();
     this.timeout = timeout;
+    _Engine.instances.set(name, this);
+    this.setupProcessExitHandlers();
+  }
+  static getInstance(name, timeout = 3e4) {
+    if (_Engine.instances.has(name)) {
+      return _Engine.instances.get(name);
+    }
+    return new _Engine(name, timeout);
   }
   async initializeBinary() {
     if (!this.binary) {
@@ -675,6 +689,12 @@ var Engine = class {
     if (!this.binary) {
       throw new Error("Binary not initialized");
     }
+    const isConnectAction = args.includes("connect");
+    const isDisconnectAction = args.includes("disconnect");
+    const isExecuteAction = args.includes("execute");
+    if (isExecuteAction) {
+      await this.ensureDaemonConnected();
+    }
     return new Promise((resolve5, reject) => {
       const child = (0, import_child_process.spawn)(this.binary[binary], [...this.arguments, ...args]);
       let stdoutBuffer = "";
@@ -692,6 +712,12 @@ var Engine = class {
           isResolved = true;
           clearTimeout(timeoutId);
           resolve5(response);
+          if (isConnectAction && response.status === 200) {
+            this.isDaemonConnected = true;
+            this.isConnecting = false;
+          } else if (isDisconnectAction) {
+            this.isDaemonConnected = false;
+          }
         }
       };
       child.stdout.on("data", (data) => {
@@ -759,6 +785,67 @@ var Engine = class {
       child.unref();
     });
   }
+  async ensureDaemonConnected() {
+    if (this.isDaemonConnected) {
+      return;
+    }
+    if (this.isConnecting) {
+      while (this.isConnecting) {
+        await new Promise((resolve5) => setTimeout(resolve5, 50));
+      }
+      return;
+    }
+    this.isConnecting = true;
+    try {
+      const response = await this.run("query_engine", ["--action", "connect"]);
+      if (response.status !== 200) {
+        throw new Error(`Failed to connect daemon: ${response.message}`);
+      }
+    } catch (error) {
+      this.isConnecting = false;
+      throw error;
+    }
+  }
+  async disconnect() {
+    if (this.isDaemonConnected) {
+      await this.run("query_engine", ["--action", "disconnect"]);
+    }
+  }
+  setupProcessExitHandlers() {
+    const gracefulShutdown = async () => {
+      if (this.isDaemonConnected) {
+        console.log("\u{1F504} Disconnecting daemon before exit...");
+        try {
+          await this.disconnect();
+          console.log("\u2705 Daemon disconnected successfully");
+        } catch (error) {
+          console.error("\u274C Error disconnecting daemon:", error);
+        }
+      }
+    };
+    process.on("exit", () => {
+    });
+    process.on("SIGINT", async () => {
+      console.log("\n\u{1F6D1} Received SIGINT (Ctrl+C)");
+      await gracefulShutdown();
+      process.exit(0);
+    });
+    process.on("SIGTERM", async () => {
+      console.log("\u{1F6D1} Received SIGTERM");
+      await gracefulShutdown();
+      process.exit(0);
+    });
+    process.on("uncaughtException", async (error) => {
+      console.error("\u{1F4A5} Uncaught Exception:", error);
+      await gracefulShutdown();
+      process.exit(1);
+    });
+    process.on("unhandledRejection", async (reason, promise) => {
+      console.error("\u{1F4A5} Unhandled Rejection at:", promise, "reason:", reason);
+      await gracefulShutdown();
+      process.exit(1);
+    });
+  }
 };
 
 // src/lib/SqliteExecutor.ts