@dbcube/core 1.0.52 → 1.0.55

package/dist/index.cjs

@@ -599,11 +599,14 @@ var Engine = class {
   arguments;
   binary = null;
   timeout;
+  isDaemonConnected = false;
+  isConnecting = false;
   constructor(name, timeout = 3e4) {
     this.name = name;
     this.config = this.setConfig(name);
     this.arguments = this.setArguments();
     this.timeout = timeout;
+    this.setupProcessExitHandlers();
   }
   async initializeBinary() {
     if (!this.binary) {
@@ -675,6 +678,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 +701,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 +774,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