@dbcube/core 3.0.1 → 3.0.3

package/dist/index.cjs

@@ -600,11 +600,17 @@ var Engine = class {
   arguments;
   binary = null;
   timeout;
+  connectionId;
+  isConnected = false;
   constructor(name, timeout = 3e4) {
     this.name = name;
     this.config = this.setConfig(name);
     this.arguments = this.setArguments();
     this.timeout = timeout;
+    this.connectionId = this.generateConnectionId();
+  }
+  generateConnectionId() {
+    return `${this.name}_${this.config.type}_${this.config.config.DATABASE}_${this.config.config.HOST || "localhost"}`;
   }
   async initializeBinary() {
     if (!this.binary) {
@@ -671,7 +677,35 @@ var Engine = class {
   getConfig() {
     return this.config;
   }
+  async ensurePersistentConnection() {
+    if (!this.isConnected) {
+      console.log("\u{1F50C} Creating persistent connection...");
+      const response = await this.createProcess("query_engine", [
+        "--action",
+        "connect"
+      ]);
+      if (response.status === 200) {
+        this.isConnected = true;
+        console.log("\u2705 Persistent connection established");
+      } else {
+        throw new Error(`Failed to create persistent connection: ${response.message}`);
+      }
+    }
+  }
   async run(binary, args) {
+    const actionIndex = args.indexOf("--action");
+    const isExecuteAction = actionIndex !== -1 && args[actionIndex + 1] === "execute";
+    if (isExecuteAction) {
+      await this.ensurePersistentConnection();
+      return this.executeWithPersistentConnection(args);
+    } else {
+      return this.createProcess(binary, args);
+    }
+  }
+  async executeWithPersistentConnection(args) {
+    return this.createProcess("query_engine", args);
+  }
+  async createProcess(binary, args) {
     await this.initializeBinary();
     if (!this.binary) {
       throw new Error("Binary not initialized");
@@ -758,6 +792,25 @@ var Engine = class {
       child.unref();
     });
   }
+  async disconnect() {
+    if (this.isConnected) {
+      console.log("\u{1F50C} Disconnecting persistent connection...");
+      const response = await this.createProcess("query_engine", [
+        "--action",
+        "disconnect"
+      ]);
+      if (response.status === 200) {
+        this.isConnected = false;
+        console.log("\u2705 Persistent connection closed");
+      }
+      return response;
+    }
+    return {
+      status: 200,
+      message: "No persistent connection to disconnect",
+      data: null
+    };
+  }
 };
 
 // src/lib/SqliteExecutor.ts