From 5744206a7aa0bd70001100ab6cc2376cca5d32c7 Mon Sep 17 00:00:00 2001
From: avallete <andrew.valleteau@supabase.io>
Date: Wed, 2 Jul 2025 17:22:32 +0200
Subject: [PATCH] feat(query): add idle_session_timeout for idle session
 auto-close

---
 src/lib/db.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/lib/db.ts b/src/lib/db.ts
index 6f7e906a..263be4d8 100644
--- a/src/lib/db.ts
+++ b/src/lib/db.ts
@@ -118,8 +118,10 @@ export const init: (config: PoolConfig) => {
             attributes: { sql: trackQueryInSentry ? sql : 'custom' },
           },
           async () => {
+            // Use statement_timeout AND idle_session_timeout to ensure the connection will be killed even if idle after
+            // timeout time.
             const statementTimeoutQueryPrefix = statementQueryTimeout
-              ? `SET statement_timeout='${statementQueryTimeout}s';`
+              ? `SET statement_timeout='${statementQueryTimeout}s'; SET idle_session_timeout='${statementQueryTimeout}s';`
               : ''
             // node-postgres need a statement_timeout to kill the connection when timeout is reached
             // otherwise the query will keep running on the database even if query timeout was reached