feat: Allow using keyword with pool clients

denodrivers · Feb 27, 2024 · 2af3660 · 2af3660
1 parent 5915574
commit 2af3660
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/client.ts b/client.ts
@@ -515,4 +515,8 @@ export class PoolClient extends QueryClient {
     // Cleanup all session related metadata
     this.resetSessionMetadata();
   }
+
+  [Symbol.dispose]() {
+    this.release();
+  }
 }
diff --git a/tests/pool_test.ts b/tests/pool_test.ts
@@ -140,3 +140,15 @@ Deno.test(
     );
   }),
 );
+
+Deno.test(
+  "Pool client will be released after `using` block",
+  testPool(async (POOL, size) => {
+    const initialPoolAvailable = POOL.available;
+    {
+      using client = await POOL.connect();
+      assertEquals(POOL.available, initialPoolAvailable - 1);
+    }
+    assertEquals(POOL.available, initialPoolAvailable);
+  }),
+);