@dbcube/core 1.0.52 → 1.0.57

package/dist/index.d.mts

@@ -7,19 +7,26 @@ interface ResponseEngine {
 }
 
 declare class Engine {
+    private static instances;
     private name;
     private config;
     private arguments;
     private binary;
     private timeout;
+    private isDaemonConnected;
+    private isConnecting;
     constructor(name: string, timeout?: number);
+    static getInstance(name: string, timeout?: number): Engine;
     initializeBinary(): Promise<void>;
     setArguments(): any[];
     setConfig(name: string): {
         [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

@@ -7,19 +7,26 @@ interface ResponseEngine {
 }
 
 declare class Engine {
+    private static instances;
     private name;
     private config;
     private arguments;
     private binary;
     private timeout;
+    private isDaemonConnected;
+    private isConnecting;
     constructor(name: string, timeout?: number);
+    static getInstance(name: string, timeout?: number): Engine;
     initializeBinary(): Promise<void>;
     setArguments(): any[];
     setConfig(name: string): {
         [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

@@ -549,17 +549,31 @@ var Config = class {
 // src/lib/Engine.ts
 import { spawn } from "child_process";
 import { createRequire } from "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) {
@@ -631,6 +645,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 +668,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 +741,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