@zz1996/dbhub-dameng 0.1.3 → 0.1.4

package/dist/{dameng-JYMTC4HE.js → dameng-SOGGZUEW.js}

@@ -8,6 +8,14 @@ import {
 
 // src/connectors/dameng/index.ts
 import dmdb from "dmdb";
+var DEFAULT_OPERATION_TIMEOUT_MS = 11e4;
+var RESOURCE_CLEANUP_TIMEOUT_MS = 2e3;
+var DamengOperationTimeoutError = class extends Error {
+  constructor(message) {
+    super(message);
+    this.name = "DamengOperationTimeoutError";
+  }
+};
 var DamengDSNParser = class {
   async parse(dsn, config) {
     if (!this.isValidDSN(dsn)) {
@@ -74,6 +82,10 @@ var DamengConnector = class _DamengConnector {
     this.sourceId = "default";
     this.defaultSchema = null;
     this.poolAlias = null;
+    this.connectionConfig = null;
+    this.connectionTimeoutMs = 5e3;
+    this.operationTimeoutMs = DEFAULT_OPERATION_TIMEOUT_MS;
+    this.reconnectingPool = null;
   }
   getId() {
     return this.sourceId;
@@ -88,14 +100,22 @@ var DamengConnector = class _DamengConnector {
       connectionConfig.poolAlias = this.buildPoolAlias();
       this.defaultSchema = connectionConfig.schema ?? null;
       this.poolAlias = connectionConfig.poolAlias;
+      this.connectionTimeoutMs = config?.connectionTimeoutSeconds !== void 0 ? config.connectionTimeoutSeconds * 1e3 : 5e3;
+      this.operationTimeoutMs = config?.queryTimeoutSeconds !== void 0 ? config.queryTimeoutSeconds * 1e3 : DEFAULT_OPERATION_TIMEOUT_MS;
       await this.closeRegisteredPool(connectionConfig.poolAlias);
       await this.validateDirectConnection(connectionConfig);
       createdPool = await dmdb.createPool(connectionConfig);
       this.pool = createdPool;
       await this.withConnection(async (conn) => {
-        await conn.execute("SELECT 1 AS OK", [], this.executeOptions());
+        await this.executeWithTimeout(
+          conn,
+          "SELECT 1 AS OK",
+          [],
+          this.executeOptions()
+        );
         if (!this.defaultSchema) {
-          const result = await conn.execute(
+          const result = await this.executeWithTimeout(
+            conn,
             "SELECT USER AS SCHEMA_NAME FROM DUAL",
             [],
             this.executeOptions()
@@ -104,10 +124,17 @@ var DamengConnector = class _DamengConnector {
         }
         if (initScript) {
           for (const statement of splitSQLStatements(initScript, "dameng")) {
-            await conn.execute(statement, [], this.executeOptions({ autoCommit: true }));
+            await this.executeWithTimeout(
+              conn,
+              statement,
+              [],
+              this.executeOptions({ autoCommit: true })
+            );
           }
         }
       });
+      this.connectionConfig = connectionConfig;
+      this.initScript = initScript;
     } catch (error) {
       if (createdPool) {
         await this.closePoolQuietly(createdPool);
@@ -116,6 +143,8 @@ var DamengConnector = class _DamengConnector {
       }
       this.pool = null;
       this.poolAlias = null;
+      this.connectionConfig = null;
+      this.initScript = void 0;
       console.error("Failed to connect to Dameng database:", error);
       throw error;
     }
@@ -129,6 +158,9 @@ var DamengConnector = class _DamengConnector {
       await this.closeRegisteredPool(this.poolAlias);
       this.poolAlias = null;
     }
+    this.connectionConfig = null;
+    this.initScript = void 0;
+    this.reconnectingPool = null;
   }
   async getSchemas() {
     const rows = await this.queryRows(`
@@ -379,7 +411,8 @@ var DamengConnector = class _DamengConnector {
           processedSQL,
           index === 0 ? parameters ?? [] : []
         );
-        const result = await conn.execute(
+        const result = await this.executeWithTimeout(
+          conn,
           boundSQL,
           this.toBindParams(bindValues),
           this.executeOptions({ autoCommit: true, maxRows: options.maxRows })
@@ -393,25 +426,136 @@ var DamengConnector = class _DamengConnector {
   }
   async queryRows(sql, bindValues = []) {
     return this.withConnection(async (conn) => {
-      const result = await conn.execute(sql, this.toBindParams(bindValues), this.executeOptions());
+      const result = await this.executeWithTimeout(
+        conn,
+        sql,
+        this.toBindParams(bindValues),
+        this.executeOptions()
+      );
       return this.normalizeRows(result.rows ?? []);
     });
   }
   async withConnection(fn) {
-    if (!this.pool) {
+    await this.ensurePool();
+    const pool = this.pool;
+    if (!pool) {
       throw new Error("Not connected to Dameng database");
     }
-    const conn = await this.pool.getConnection();
+    let conn = null;
+    let shouldRelease = true;
     try {
+      conn = await this.withTimeout(
+        pool.getConnection(),
+        this.connectionTimeoutMs,
+        "Dameng connection acquisition"
+      );
       return await fn(conn);
+    } catch (error) {
+      if (error instanceof DamengOperationTimeoutError) {
+        shouldRelease = false;
+        this.markPoolUnhealthy(error.message);
+      }
+      throw error;
+    } finally {
+      if (conn && shouldRelease) {
+        await this.releaseConnectionQuietly(conn);
+      }
+    }
+  }
+  async ensurePool() {
+    if (this.pool) {
+      return;
+    }
+    if (!this.connectionConfig) {
+      throw new Error("Not connected to Dameng database");
+    }
+    if (!this.reconnectingPool) {
+      this.reconnectingPool = this.reconnectPool();
+    }
+    try {
+      await this.reconnectingPool;
+    } finally {
+      this.reconnectingPool = null;
+    }
+  }
+  async reconnectPool() {
+    const config = this.connectionConfig;
+    if (!config) {
+      throw new Error("Not connected to Dameng database");
+    }
+    console.error(`Reconnecting Dameng source '${this.sourceId}' after pool reset...`);
+    await this.closeRegisteredPool(config.poolAlias ?? this.buildPoolAlias());
+    await this.validateDirectConnection(config);
+    const createdPool = await dmdb.createPool(config);
+    this.pool = createdPool;
+    this.poolAlias = config.poolAlias ?? null;
+    try {
+      await this.withConnection(async (conn) => {
+        await this.executeWithTimeout(conn, "SELECT 1 AS OK", [], this.executeOptions());
+        if (this.initScript) {
+          for (const statement of splitSQLStatements(this.initScript, "dameng")) {
+            await this.executeWithTimeout(
+              conn,
+              statement,
+              [],
+              this.executeOptions({ autoCommit: true })
+            );
+          }
+        }
+      });
+    } catch (error) {
+      await this.closePoolQuietly(createdPool);
+      this.pool = null;
+      throw error;
+    }
+  }
+  executeWithTimeout(conn, sql, bindParams, options) {
+    return this.withTimeout(
+      conn.execute(sql, bindParams, options),
+      this.operationTimeoutMs,
+      "Dameng SQL execution"
+    );
+  }
+  async withTimeout(promise, timeoutMs, label) {
+    if (!Number.isFinite(timeoutMs) || timeoutMs <= 0) {
+      return promise;
+    }
+    let timer;
+    const timeout = new Promise((_, reject) => {
+      timer = setTimeout(() => {
+        reject(new DamengOperationTimeoutError(`${label} timed out after ${timeoutMs}ms`));
+      }, timeoutMs);
+    });
+    try {
+      return await Promise.race([promise, timeout]);
     } finally {
-      if (conn.release) {
-        await conn.release();
-      } else {
-        await conn.close();
+      if (timer) {
+        clearTimeout(timer);
       }
     }
   }
+  markPoolUnhealthy(reason) {
+    const pool = this.pool;
+    this.pool = null;
+    console.error(`Resetting Dameng source '${this.sourceId}' pool: ${reason}`);
+    if (pool) {
+      void this.closePoolQuietly(pool);
+    }
+  }
+  async releaseConnectionQuietly(conn) {
+    try {
+      const release = conn.release ? conn.release() : conn.close();
+      await this.withTimeout(
+        release,
+        RESOURCE_CLEANUP_TIMEOUT_MS,
+        "Dameng connection release"
+      );
+    } catch (error) {
+      this.markPoolUnhealthy(
+        `failed to release connection: ${error instanceof Error ? error.message : String(error)}`
+      );
+    }
+  }
   executeOptions(extra = {}) {
     return {
       outFormat: dmdb.OUT_FORMAT_OBJECT,
@@ -430,15 +574,19 @@ var DamengConnector = class _DamengConnector {
         queueTimeout,
         ...directConfig
       } = config;
-      const directConn = await dmdb.getConnection({
-        ...directConfig,
-        connectString: directConnectString ?? config.connectString
-      });
+      const directConn = await this.withTimeout(
+        dmdb.getConnection({
+          ...directConfig,
+          connectString: directConnectString ?? config.connectString
+        }),
+        this.connectionTimeoutMs,
+        "Dameng direct connection"
+      );
       conn = directConn;
-      await conn.execute("SELECT 1 AS OK", [], this.executeOptions());
+      await this.executeWithTimeout(conn, "SELECT 1 AS OK", [], this.executeOptions());
     } finally {
       if (conn) {
-        await conn.close();
+        await this.releaseConnectionQuietly(conn);
       }
     }
   }
@@ -456,7 +604,7 @@ var DamengConnector = class _DamengConnector {
   }
   async closePoolQuietly(pool) {
     try {
-      await pool.close(0);
+      await this.withTimeout(pool.close(0), RESOURCE_CLEANUP_TIMEOUT_MS, "Dameng pool close");
     } catch {
       if (pool.poolAlias) {
         dmdb.pools?.delete?.(pool.poolAlias);

package/dist/index.js

@@ -1835,7 +1835,7 @@ var connectorModules = [
   { load: () => import("./sqlite-IOUAYHGE.js"), name: "SQLite", driver: "node:sqlite" },
   { load: () => import("./mysql-A43SL7UM.js"), name: "MySQL", driver: "mysql2" },
   { load: () => import("./mariadb-7F72IRB4.js"), name: "MariaDB", driver: "mariadb" },
-  { load: () => import("./dameng-JYMTC4HE.js"), name: "Dameng", driver: "dmdb" }
+  { load: () => import("./dameng-SOGGZUEW.js"), name: "Dameng", driver: "dmdb" }
 ];
 loadConnectors(connectorModules).then(() => main()).catch((error) => {
   console.error("Fatal error:", error);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zz1996/dbhub-dameng",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "mcpName": "io.github.zuozh11/dbhub-dameng",
   "description": "Local fork of DBHub with Dameng/DM8 database connector support",
   "repository": {