From c71779fcb52565b3228adb6702dc1680f1cc8ae9 Mon Sep 17 00:00:00 2001
From: Karthik KK <karthik@techgeek.co.in>
Date: Tue, 22 Apr 2025 16:44:36 +1200
Subject: [PATCH] Add connection timeout support for PostgreSQL in MCP Database
 Server

Enhanced the PostgreSQL connection handling by introducing a `--connection-timeout` parameter, allowing users to specify a custom timeout in milliseconds. Updated relevant documentation in README and postgresql-setup.md to reflect this new option. Modified the PostgresqlAdapter to utilize the connection timeout setting during database connections.
---
 .DS_Store                    | Bin 8196 -> 8196 bytes
 docs/postgresql-setup.md     |   4 ++++
 readme.md                    |   1 +
 src/.DS_Store                | Bin 6148 -> 6148 bytes
 src/db/postgresql-adapter.ts |  14 +++++++++++++-
 src/index.ts                 |   5 ++++-
 6 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/.DS_Store b/.DS_Store
index 615e03ba58b7bdaca96b2d0a9785bbdc197533a5..2cccfe87c9540f8c1af1eae5f18b0fba6669b55e 100644
GIT binary patch
delta 493
zcmZp1XmQxELx6SPU3Sm7$$JH)A*{`J1-KYv@AWe<Ft9MBFk~_mGnC}!ySOCf<R<~e
zIA#^)Z(4Qxm?Nru3SRkw48!2${M-Vd9tH+RhsiEN#&LQ(peE!qBr_Bv+i;-j$f{g~
z4N!T64cO#SY+!I;n5-x)GWn&@T4v3l?8&*pG9t(p$XTT-uLN2El|Yz`EWwX#B-@3A
z$tohx!w*jdI+__6Bt;C#$R>JOa(KpF0|yFR7GW%=EV9`Qj0&5RL|3xY(C|e}0Es?`
A2><{9

delta 493
zcmZp1XmQxELx8nIs`j(j<h=sY5Z30q0$hx-{YM!X7+4rm7%~}(8A@{VU0jlK@{@pK
z919jM`WA5Qm?Nru3SRkw48!2${M-Vd9w2xy*+s}Wj_(JM39=!dA(^2V*@hMm(-3!r
z4N!T64cO#SZ4h9XtSBrp`K8cWW_gF*lXHb-M360bZCl>u1+)MvfiM|af*;vPwmlCf
ytB5=gUm6Z{G&3+riWriSO>~p_xYY*YWVkHCSWH=Dvl$o}HYbU$WT&Cwi<kfp{)_zp

diff --git a/docs/postgresql-setup.md b/docs/postgresql-setup.md
index fc1691c..db6e322 100644
--- a/docs/postgresql-setup.md
+++ b/docs/postgresql-setup.md
@@ -25,6 +25,9 @@ node dist/src/index.js --postgresql --host localhost --database yourdb --user po
 
 # With SSL enabled
 node dist/src/index.js --postgresql --host localhost --database yourdb --user postgres --password yourpassword --ssl true
+
+# With custom connection timeout (in milliseconds)
+node dist/src/index.js --postgresql --host localhost --database yourdb --user postgres --password yourpassword --connection-timeout 60000
 ```
 
 ## Command Line Arguments
@@ -36,6 +39,7 @@ node dist/src/index.js --postgresql --host localhost --database yourdb --user po
 - `--password`: The password for the PostgreSQL user.
 - `--port`: The port the PostgreSQL server is listening on (default: 5432).
 - `--ssl`: Whether to use SSL for the connection (true/false).
+- `--connection-timeout`: The connection timeout in milliseconds (default: 30000).
 
 ## Usage from MCP Client
 
diff --git a/readme.md b/readme.md
index 6eacbba..2148f14 100644
--- a/readme.md
+++ b/readme.md
@@ -88,6 +88,7 @@ Optional parameters:
 - `--password`: Password for PostgreSQL authentication
 - `--port`: Port number (default: 5432)
 - `--ssl`: Enable SSL connection (true/false)
+- `--connection-timeout`: Connection timeout in milliseconds (default: 30000)
 
 ## Configuring Claude Desktop
 
diff --git a/src/.DS_Store b/src/.DS_Store
index c60d57720499423a19d7e192fc26ca2e89a2221e..59091627ac0be34a4014a7d68d629cbc4e1e62cb 100644
GIT binary patch
delta 127
zcmZoMXffFEj)f_3{p1fUBG$FlLXEk%k2x|hFfcKsFeK&Xr?@2L<R=079O!Zd8HT~h
x`MCu^IUp#QY|k3W=3&I)8Fy{+JXR$HQ*81bRu8uM_Fx$;HW?^$vk%)2K>*xDCguPD

delta 127
zcmZoMXffFEj)loeaPkKh5o<-&g}>dd9dl%0U|?cMVMxl&PjN}g$xj0EInd<_G7N*0
z^K%P;azGF;*`76$?eZVClh>|Jp2w<$V2Vw?!|K82!Wy{{D5J$D17&XZVcQ`H0A-yl
A5dZ)H

diff --git a/src/db/postgresql-adapter.ts b/src/db/postgresql-adapter.ts
index 1b68297..6e95cf1 100644
--- a/src/db/postgresql-adapter.ts
+++ b/src/db/postgresql-adapter.ts
@@ -18,6 +18,7 @@ export class PostgresqlAdapter implements DbAdapter {
     port?: number;
     ssl?: boolean | object;
     options?: any;
+    connectionTimeout?: number;
   }) {
     this.host = connectionInfo.host;
     this.database = connectionInfo.database;
@@ -29,7 +30,9 @@ export class PostgresqlAdapter implements DbAdapter {
       port: connectionInfo.port || 5432,
       user: connectionInfo.user,
       password: connectionInfo.password,
-      ssl: connectionInfo.ssl
+      ssl: connectionInfo.ssl,
+      // Add connection timeout if provided (in milliseconds)
+      connectionTimeoutMillis: connectionInfo.connectionTimeout || 30000,
     };
   }
 
@@ -39,6 +42,15 @@ export class PostgresqlAdapter implements DbAdapter {
   async init(): Promise<void> {
     try {
       console.error(`[INFO] Connecting to PostgreSQL: ${this.host}, Database: ${this.database}`);
+      console.error(`[DEBUG] Connection details:`, {
+        host: this.host, 
+        database: this.database,
+        port: this.config.port,
+        user: this.config.user,
+        connectionTimeoutMillis: this.config.connectionTimeoutMillis,
+        ssl: !!this.config.ssl
+      });
+      
       this.client = new pg.Client(this.config);
       await this.client.connect();
       console.error(`[INFO] PostgreSQL connection established successfully`);
diff --git a/src/index.ts b/src/index.ts
index 5da0fee..6b29b06 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -92,7 +92,8 @@ else if (args.includes('--postgresql') || args.includes('--postgres')) {
     user: undefined,
     password: undefined,
     port: undefined,
-    ssl: undefined
+    ssl: undefined,
+    connectionTimeout: undefined
   };
   
   // Parse PostgreSQL connection parameters
@@ -109,6 +110,8 @@ else if (args.includes('--postgresql') || args.includes('--postgres')) {
       connectionInfo.port = parseInt(args[i + 1], 10);
     } else if (args[i] === '--ssl' && i + 1 < args.length) {
       connectionInfo.ssl = args[i + 1] === 'true';
+    } else if (args[i] === '--connection-timeout' && i + 1 < args.length) {
+      connectionInfo.connectionTimeout = parseInt(args[i + 1], 10);
     }
   }