@dbcube/core 1.0.52 → 1.0.55

package/dist/index.d.mts

@@ -12,6 +12,8 @@ declare class Engine {
     private arguments;
     private binary;
     private timeout;
+    private isDaemonConnected;
+    private isConnecting;
     constructor(name: string, timeout?: number);
     initializeBinary(): Promise<void>;
     setArguments(): any[];
@@ -19,7 +21,10 @@ declare class Engine {
         [x: string]: any;
     } | null;
     getConfig(): any;
-    run(binary: string, args: []): Promise<ResponseEngine>;
+    run(binary: string, args: string[]): Promise<ResponseEngine>;
+    ensureDaemonConnected(): Promise<void>;
+    disconnect(): Promise<void>;
+    private setupProcessExitHandlers;
 }
 
 interface SystemInfo {

package/dist/index.d.ts

@@ -12,6 +12,8 @@ declare class Engine {
     private arguments;
     private binary;
     private timeout;
+    private isDaemonConnected;
+    private isConnecting;
     constructor(name: string, timeout?: number);
     initializeBinary(): Promise<void>;
     setArguments(): any[];
@@ -19,7 +21,10 @@ declare class Engine {
         [x: string]: any;
     } | null;
     getConfig(): any;
-    run(binary: string, args: []): Promise<ResponseEngine>;
+    run(binary: string, args: string[]): Promise<ResponseEngine>;
+    ensureDaemonConnected(): Promise<void>;
+    disconnect(): Promise<void>;
+    private setupProcessExitHandlers;
 }
 
 interface SystemInfo {

package/dist/index.js

@@ -555,11 +555,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) {
@@ -631,6 +634,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 = spawn(this.binary[binary], [...this.arguments, ...args]);
       let stdoutBuffer = "";
@@ -648,6 +657,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) => {
@@ -715,6 +730,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