feat: add idleCount, totalCount, waitingCount

Author: gajus

src/bridge.ts

@@ -79,11 +79,14 @@ export const bridge = (postgres: typeof Postgres, poolConfiguration: PgPool) =>
     min: poolConfiguration.min,
   });
 
-  return {
+  const compatiblePool = {
     connect: async () => {
       const connection = await pool.acquire();
 
       const compatibleConnection = {
+        end: async () => {
+          await pool.destroy(connection);
+        },
         off: connection.events.off.bind(connection.events),
         on: connection.events.on.bind(connection.events),
         query: async (sql: string): Promise<QueryResult> => {
@@ -102,13 +105,27 @@ export const bridge = (postgres: typeof Postgres, poolConfiguration: PgPool) =>
             rows: Array.from(resultArray),
           };
         },
+        release: async () => {
+          await pool.release(connection);
+        },
       };
 
       poolEvents.emit('connect', compatibleConnection);
 
       return compatibleConnection;
     },
+    get idleCount () {
+      return pool.available;
+    },
     off: poolEvents.off.bind(poolEvents),
     on: poolEvents.on.bind(poolEvents),
+    get totalCount () {
+      return pool.size;
+    },
+    get waitingCount () {
+      return pool.pending;
+    },
   };
+
+  return compatiblePool;
 };

test/postgres-bridge/bridge.ts

@@ -79,6 +79,36 @@ for (const client of clients) {
     });
   });
 
+  test(client + ': pool.connect() 2x creates two connections', async (t) => {
+    const pool = createPool(client, {
+      user: 'postgres',
+    });
+
+    t.is(pool.totalCount, 0);
+
+    const connection1 = await pool.connect();
+    const connection2 = await pool.connect();
+
+    t.is(pool.totalCount, 2);
+
+    await connection1.end();
+    await connection2.end();
+  });
+
+  test(client + ': connection.release() releases connection back to the pool', async (t) => {
+    const pool = createPool(client, {
+      user: 'postgres',
+    });
+
+    const connection = await pool.connect();
+
+    t.is(pool.idleCount, 0);
+
+    await connection.release();
+
+    t.is(pool.idleCount, 1);
+  });
+
   test(client + ': query method', async (t) => {
     const pool = createPool(client, {
       user: 'postgres',