@zz1996/dbhub-dameng 0.1.1 → 0.1.2

package/dbhub.dameng.toml.example

@@ -5,6 +5,7 @@
 id = "dm8_dev"
 description = "Dameng DM8 development database. Read-only access for agent schema discovery and SQL queries."
 dsn = "dameng://${DM8_USER}:${DM8_PASSWORD}@${DM8_HOST}:5236/${DM8_SCHEMA}"
+connection_timeout = 5
 query_timeout = 30
 lazy = true
 

package/dist/{dameng-DC7OP4VV.js → dameng-UJELBZ5R.js}

@@ -65,6 +65,7 @@ var DamengConnector = class _DamengConnector {
     this.pool = null;
     this.sourceId = "default";
     this.defaultSchema = null;
+    this.poolAlias = null;
   }
   getId() {
     return this.sourceId;
@@ -73,13 +74,20 @@ var DamengConnector = class _DamengConnector {
     return new _DamengConnector();
   }
   async connect(dsn, initScript, config) {
+    let createdPool = null;
     try {
       const connectionConfig = await this.dsnParser.parse(dsn, config);
+      connectionConfig.poolAlias = this.buildPoolAlias();
       this.defaultSchema = connectionConfig.schema ?? null;
+      this.poolAlias = connectionConfig.poolAlias;
+      this.queryTimeoutMs = void 0;
       if (config?.queryTimeoutSeconds !== void 0) {
         this.queryTimeoutMs = config.queryTimeoutSeconds * 1e3;
       }
-      this.pool = await dmdb.createPool(connectionConfig);
+      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());
         if (!this.defaultSchema) {
@@ -97,15 +105,26 @@ var DamengConnector = class _DamengConnector {
         }
       });
     } catch (error) {
+      if (createdPool) {
+        await this.closePoolQuietly(createdPool);
+      } else if (this.poolAlias) {
+        await this.closeRegisteredPool(this.poolAlias);
+      }
+      this.pool = null;
+      this.poolAlias = null;
       console.error("Failed to connect to Dameng database:", error);
       throw error;
     }
   }
   async disconnect() {
     if (this.pool) {
-      await this.pool.close(0);
+      await this.closePoolQuietly(this.pool);
       this.pool = null;
     }
+    if (this.poolAlias) {
+      await this.closeRegisteredPool(this.poolAlias);
+      this.poolAlias = null;
+    }
   }
   async getSchemas() {
     const rows = await this.queryRows(`
@@ -366,6 +385,38 @@ var DamengConnector = class _DamengConnector {
       ...extra
     };
   }
+  async validateDirectConnection(config) {
+    let conn = null;
+    try {
+      conn = await dmdb.getConnection(config.connectString);
+      await conn.execute("SELECT 1 AS OK", [], this.executeOptions());
+    } finally {
+      if (conn) {
+        await conn.close();
+      }
+    }
+  }
+  buildPoolAlias() {
+    const safeSourceId = this.sourceId.replace(/[^a-zA-Z0-9_-]/g, "_") || "default";
+    return `dbhub_dameng_${safeSourceId}`;
+  }
+  async closeRegisteredPool(poolAlias) {
+    if (!dmdb.pools?.has?.(poolAlias)) {
+      return;
+    }
+    const pool = dmdb.pools.get(poolAlias);
+    await this.closePoolQuietly(pool);
+    dmdb.pools?.delete?.(poolAlias);
+  }
+  async closePoolQuietly(pool) {
+    try {
+      await pool.close(0);
+    } catch {
+      if (pool.poolAlias) {
+        dmdb.pools?.delete?.(pool.poolAlias);
+      }
+    }
+  }
   async resolveSchema(schema) {
     const resolved = schema || this.defaultSchema || await this.getDefaultSchema();
     if (!resolved) {

package/dist/index.js

@@ -1833,7 +1833,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-DC7OP4VV.js"), name: "Dameng", driver: "dmdb" }
+  { load: () => import("./dameng-UJELBZ5R.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.1",
+  "version": "0.1.2",
   "mcpName": "io.github.zuozh11/dbhub-dameng",
   "description": "Local fork of DBHub with Dameng/DM8 database connector support",
   "repository": {