mcp-db-analyzer 0.2.0 → 0.2.2

package/build/db-mysql.js

@@ -1,4 +1,5 @@
 import mysql from "mysql2/promise";
+import { getConnectionTimeoutMs } from "./db.js";
 function wrapConnectionError(err) {
     const msg = err instanceof Error ? err.message : String(err);
     const sanitized = msg.replace(/\/\/[^@]+@/g, "//****:****@");
@@ -11,9 +12,10 @@ export function createMysqlAdapter() {
     let pool = null;
     function getPool() {
         if (!pool) {
+            const timeoutMs = getConnectionTimeoutMs();
             const uri = process.env.DATABASE_URL;
             if (uri) {
-                pool = mysql.createPool(uri);
+                pool = mysql.createPool({ uri, connectTimeout: timeoutMs });
             }
             else {
                 pool = mysql.createPool({
@@ -22,6 +24,7 @@ export function createMysqlAdapter() {
                     database: process.env.MYSQL_DATABASE || process.env.DB_NAME,
                     user: process.env.MYSQL_USER || process.env.DB_USER,
                     password: process.env.MYSQL_PASSWORD || process.env.DB_PASSWORD,
+                    connectTimeout: timeoutMs,
                 });
             }
         }

package/build/db-postgres.js

@@ -1,4 +1,5 @@
 import pg from "pg";
+import { getConnectionTimeoutMs } from "./db.js";
 const { Pool } = pg;
 function wrapConnectionError(err) {
     const msg = err instanceof Error ? err.message : String(err);
@@ -12,9 +13,14 @@ export function createPostgresAdapter() {
     let pool = null;
     function getPool() {
         if (!pool) {
+            const timeoutMs = getConnectionTimeoutMs();
             const connectionString = process.env.DATABASE_URL;
             if (connectionString) {
-                pool = new Pool({ connectionString });
+                pool = new Pool({
+                    connectionString,
+                    connectionTimeoutMillis: timeoutMs,
+                    query_timeout: timeoutMs,
+                });
             }
             else {
                 pool = new Pool({
@@ -23,6 +29,8 @@ export function createPostgresAdapter() {
                     database: process.env.PGDATABASE || "postgres",
                     user: process.env.PGUSER,
                     password: process.env.PGPASSWORD,
+                    connectionTimeoutMillis: timeoutMs,
+                    query_timeout: timeoutMs,
                 });
             }
         }

package/build/db.js

@@ -1,8 +1,15 @@
 let adapter = null;
 let driverType = "postgres";
+let connectionTimeoutMs = 30000;
 export function getDriverType() {
     return driverType;
 }
+export function getConnectionTimeoutMs() {
+    return connectionTimeoutMs;
+}
+export function setConnectionTimeoutMs(ms) {
+    connectionTimeoutMs = ms;
+}
 export function setAdapter(a) {
     adapter = a;
     driverType = a.driver;

package/build/index.js

@@ -11,7 +11,7 @@ import { analyzeSlowQueries } from "./analyzers/slow-queries.js";
 import { analyzeConnections } from "./analyzers/connections.js";
 import { analyzeTableRelationships } from "./analyzers/relationships.js";
 import { analyzeVacuum } from "./analyzers/vacuum.js";
-import { closePool, initDriver } from "./db.js";
+import { closePool, initDriver, setConnectionTimeoutMs } from "./db.js";
 import { formatToolError } from "./errors.js";
 import { validateLicense, formatUpgradePrompt } from "./license.js";
 // License check (reads MCP_LICENSE_KEY env var once at startup)
@@ -73,6 +73,15 @@ const server = new McpServer({
     name: "mcp-db-analyzer",
     version: "0.1.0",
 });
+// Shared Zod parameter for connection timeout
+const timeoutParam = z
+    .number()
+    .optional()
+    .default(30000)
+    .describe("Connection timeout in milliseconds (default: 30000). Increase for slow or remote databases.");
+function applyTimeout(timeout_ms) {
+    setConnectionTimeoutMs(timeout_ms);
+}
 // --- Tool: inspect_schema ---
 server.tool("inspect_schema", "List all tables in a schema with row counts and sizes, or inspect a specific table's columns, types, constraints, and foreign keys.", {
     table: z
@@ -83,7 +92,9 @@ server.tool("inspect_schema", "List all tables in a schema with row counts and s
         .string()
         .default("public")
         .describe("Database schema to inspect (default: public)"),
-}, async ({ table, schema }) => {
+    timeout_ms: timeoutParam,
+}, async ({ table, schema, timeout_ms }) => {
+    applyTimeout(timeout_ms);
     try {
         const result = table
             ? await inspectTable(table, schema)
@@ -111,7 +122,9 @@ server.tool("analyze_indexes", "Analyze index usage statistics to find unused in
         .enum(["usage", "missing", "all"])
         .default("all")
         .describe("Analysis mode: 'usage' for unused index detection, 'missing' for missing index suggestions, 'all' for both"),
-}, async ({ schema, mode }) => {
+    timeout_ms: timeoutParam,
+}, async ({ schema, mode, timeout_ms }) => {
+    applyTimeout(timeout_ms);
     try {
         const parts = [];
         if (mode === "usage" || mode === "all") {
@@ -140,7 +153,9 @@ server.tool("explain_query", "Run EXPLAIN on a SQL query and return a formatted
         .boolean()
         .default(false)
         .describe("Run EXPLAIN ANALYZE to get actual execution times (executes the query). Only allowed for SELECT queries."),
-}, async ({ sql, analyze }) => {
+    timeout_ms: timeoutParam,
+}, async ({ sql, analyze, timeout_ms }) => {
+    applyTimeout(timeout_ms);
     try {
         const result = await explainQuery(sql, analyze);
         return { content: [{ type: "text", text: result }] };
@@ -162,7 +177,9 @@ server.tool("analyze_table_bloat", "Analyze table bloat by checking dead tuple r
         .string()
         .default("public")
         .describe("Database schema to analyze (default: public)"),
-}, async ({ schema }) => {
+    timeout_ms: timeoutParam,
+}, async ({ schema, timeout_ms }) => {
+    applyTimeout(timeout_ms);
     try {
         const result = await analyzeTableBloat(schema);
         return { content: [{ type: "text", text: result }] };
@@ -184,7 +201,9 @@ server.tool("suggest_missing_indexes", "Find tables with high sequential scan co
         .string()
         .default("public")
         .describe("Database schema to analyze (default: public)"),
-}, async ({ schema }) => {
+    timeout_ms: timeoutParam,
+}, async ({ schema, timeout_ms }) => {
+    applyTimeout(timeout_ms);
     if (!license.isPro) {
         return {
             content: [{
@@ -221,7 +240,9 @@ server.tool("analyze_slow_queries", "Find the slowest queries using pg_stat_stat
         .number()
         .default(10)
         .describe("Number of slow queries to return (default: 10)"),
-}, async ({ schema, limit }) => {
+    timeout_ms: timeoutParam,
+}, async ({ schema, limit, timeout_ms }) => {
+    applyTimeout(timeout_ms);
     if (!license.isPro) {
         return {
             content: [{
@@ -249,7 +270,10 @@ server.tool("analyze_slow_queries", "Find the slowest queries using pg_stat_stat
     }
 });
 // Tool 7: analyze_connections
-server.tool("analyze_connections", "Analyze active database connections. Detects idle-in-transaction sessions, long-running queries, lock contention, and connection pool utilization. PostgreSQL and MySQL only.", {}, async () => {
+server.tool("analyze_connections", "Analyze active database connections. Detects idle-in-transaction sessions, long-running queries, lock contention, and connection pool utilization. PostgreSQL and MySQL only.", {
+    timeout_ms: timeoutParam,
+}, async ({ timeout_ms }) => {
+    applyTimeout(timeout_ms);
     if (!license.isPro) {
         return {
             content: [{
@@ -283,7 +307,9 @@ server.tool("analyze_table_relationships", "Analyze foreign key relationships be
         .string()
         .default("public")
         .describe("Database schema to analyze (default: public)"),
-}, async ({ schema }) => {
+    timeout_ms: timeoutParam,
+}, async ({ schema, timeout_ms }) => {
+    applyTimeout(timeout_ms);
     if (!license.isPro) {
         return {
             content: [{
@@ -317,7 +343,9 @@ server.tool("analyze_vacuum", "Analyze PostgreSQL VACUUM maintenance status. Che
         .string()
         .default("public")
         .describe("Database schema to analyze (default: public)"),
-}, async ({ schema }) => {
+    timeout_ms: timeoutParam,
+}, async ({ schema, timeout_ms }) => {
+    applyTimeout(timeout_ms);
     try {
         const result = await analyzeVacuum(schema);
         return { content: [{ type: "text", text: result }] };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mcp-db-analyzer",
-  "version": "0.2.0",
+  "version": "0.2.2",
   "description": "MCP server for PostgreSQL, MySQL, and SQLite schema analysis, index optimization, and query plan inspection",
   "author": "Dmytro Lisnichenko",
   "type": "module",
@@ -24,16 +24,22 @@
   ],
   "keywords": [
     "mcp",
+    "mcp-server",
     "model-context-protocol",
+    "ai",
+    "claude",
+    "anthropic",
+    "database",
+    "sql",
+    "schema",
+    "database-analyzer",
+    "database-diagnostics",
     "postgresql",
     "mysql",
     "sqlite",
-    "database",
     "schema-analyzer",
     "index-optimization",
-    "query-plan",
-    "ai",
-    "claude"
+    "query-plan"
   ],
   "license": "MIT",
   "engines": {