@bytebase/dbhub 1.2.2 → 1.2.4

package/dist/index.js

@@ -2068,11 +2068,11 @@ See documentation for more details on configuring database connections.
 
 // src/index.ts
 var connectorModules = [
-  { load: () => import("./postgres-2FHS64EO.js"), name: "PostgreSQL", driver: "pg" },
+  { load: () => import("./postgres-M3W3Y6F5.js"), name: "PostgreSQL", driver: "pg" },
   { load: () => import("./sqlserver-WNVSBFJ3.js"), name: "SQL Server", driver: "mssql" },
   { load: () => import("./sqlite-ASYMU5BI.js"), name: "SQLite", driver: "node:sqlite" },
   { load: () => import("./mysql-GXDUNLJ6.js"), name: "MySQL", driver: "mysql2" },
-  { load: () => import("./mariadb-DQ6P57IX.js"), name: "MariaDB", driver: "mariadb" }
+  { load: () => import("./mariadb-EVJ5HLYH.js"), name: "MariaDB", driver: "mariadb" }
 ];
 loadConnectors(connectorModules).then(() => main()).catch((error) => {
   console.error("Fatal error:", error);

package/dist/{mariadb-DQ6P57IX.js → mariadb-EVJ5HLYH.js}

@@ -125,6 +125,12 @@ var MariaDBConnector = class _MariaDBConnector {
     try {
       const connectionConfig = await this.dsnParser.parse(dsn, config);
       this.pool = mariadb.createPool(connectionConfig);
+      this.pool.on("error", (err) => {
+        console.error(
+          `MariaDB pool (source "${this.sourceId}"): background connection error, pool will retry:`,
+          err.message
+        );
+      });
       const rows = await this.pool.query("SELECT VERSION() AS version");
       this.supportsReadOnlyTransaction = !isTiDBVersion(rows?.[0]?.version);
     } catch (err) {

package/dist/{postgres-2FHS64EO.js → postgres-M3W3Y6F5.js}

@@ -19,7 +19,7 @@ import {
 import fs from "fs";
 import os from "os";
 import path from "path";
-import pg from "pg";
+import pg2 from "pg";
 
 // src/connectors/postgres/failed-to-read-certificate.ts
 var FailedToReadCertificate = class extends Error {
@@ -29,8 +29,35 @@ var FailedToReadCertificate = class extends Error {
   }
 };
 
+// src/connectors/postgres/type-parsers.ts
+import pg from "pg";
+var VERBATIM_DATE_TIME_OIDS = /* @__PURE__ */ new Set([
+  1082,
+  // date
+  1114,
+  // timestamp without time zone
+  1184,
+  // timestamp with time zone
+  1182,
+  // date[]
+  1115,
+  // timestamp[]
+  1185
+  // timestamptz[]
+]);
+var passthrough = (value) => value;
+var postgresTypeParsers = {
+  getTypeParser(oid, format) {
+    if (VERBATIM_DATE_TIME_OIDS.has(oid) && (format === void 0 || format === "text")) {
+      return passthrough;
+    }
+    return pg.types.getTypeParser(oid, format);
+  }
+};
+
 // src/connectors/postgres/index.ts
-var { Pool } = pg;
+var { Pool } = pg2;
+var POSTGRES_CLIENT_QUERY_TIMEOUT_GRACE_MS = 5e3;
 var PostgresDSNParser = class {
   async parse(dsn, config) {
     const connectionTimeoutSeconds = config?.connectionTimeoutSeconds;
@@ -90,7 +117,9 @@ Expected: ${expectedFormat}`
         poolConfig.connectionTimeoutMillis = connectionTimeoutSeconds * 1e3;
       }
       if (queryTimeoutSeconds !== void 0) {
-        poolConfig.query_timeout = queryTimeoutSeconds * 1e3;
+        const queryTimeoutMs = queryTimeoutSeconds * 1e3;
+        poolConfig.statement_timeout = queryTimeoutMs;
+        poolConfig.query_timeout = queryTimeoutMs + POSTGRES_CLIENT_QUERY_TIMEOUT_GRACE_MS;
       }
       return poolConfig;
     } catch (error) {
@@ -136,6 +165,7 @@ var PostgresConnector = class _PostgresConnector {
     this.defaultSchema = "public";
     try {
       const poolConfig = await this.dsnParser.parse(dsn, config);
+      poolConfig.types = postgresTypeParsers;
       if (config?.readonly) {
         poolConfig.options = (poolConfig.options || "") + " -c default_transaction_read_only=on";
       }
@@ -149,6 +179,12 @@ var PostgresConnector = class _PostgresConnector {
         }
       }
       this.pool = new Pool(poolConfig);
+      this.pool.on("error", (err) => {
+        console.error(
+          `PostgreSQL pool (source "${this.sourceId}"): idle connection dropped, will reconnect on next query:`,
+          err.message
+        );
+      });
       const client = await this.pool.connect();
       client.release();
     } catch (err) {
@@ -196,7 +232,7 @@ var PostgresConnector = class _PostgresConnector {
         SELECT table_name
         FROM information_schema.tables
         WHERE table_schema = $1
-        AND table_type = 'BASE TABLE'
+        AND table_type IN ('BASE TABLE', 'FOREIGN')
         ORDER BY table_name
       `,
         [schemaToUse]
@@ -275,7 +311,7 @@ var PostgresConnector = class _PostgresConnector {
           AND i.oid = ix.indexrelid
           AND a.attrelid = t.oid
           AND a.attnum = ANY(ix.indkey)
-          AND t.relkind = 'r'
+          AND t.relkind IN ('r','p')
           AND t.relname = $1
           AND ns.oid = t.relnamespace
           AND ns.nspname = $2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bytebase/dbhub",
-  "version": "1.2.2",
+  "version": "1.2.4",
   "mcpName": "io.github.bytebase/dbhub",
   "description": "Minimal, token-efficient Database MCP Server for PostgreSQL, MySQL, SQL Server, SQLite, MariaDB",
   "repository": {