Exposed option to inject pg.PoolClient

src/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@event-driven-io/pongo-core",
-  "version": "0.6.1",
+  "version": "0.7.0",
   "description": "Pongo - Mongo with strong consistency on top of Postgres",
   "type": "module",
   "engines": {

src/packages/pongo/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@event-driven-io/pongo",
-  "version": "0.6.1",
+  "version": "0.7.0",
   "description": "Pongo - Mongo with strong consistency on top of Postgres",
   "type": "module",
   "scripts": {

src/packages/pongo/src/main/pongoClient.ts

@@ -1,12 +1,16 @@
 import { getDatabaseNameOrDefault } from '@event-driven-io/dumbo';
+import pg from 'pg';
 import { getDbClient, type DbClient } from './dbClient';
 import type { PongoClient, PongoDb } from './typing/operations';
 
-export const pongoClient = (connectionString: string): PongoClient => {
+export const pongoClient = (
+  connectionString: string,
+  options: { client?: pg.PoolClient } = {},
+): PongoClient => {
   const defaultDbName = getDatabaseNameOrDefault(connectionString);
   const dbClients: Map<string, DbClient> = new Map();
 
-  const dbClient = getDbClient({ connectionString });
+  const dbClient = getDbClient({ connectionString, client: options.client });
   dbClients.set(defaultDbName, dbClient);
 
   const pongoClient: PongoClient = {
@@ -25,7 +29,14 @@ export const pongoClient = (connectionString: string): PongoClient => {
       return (
         dbClients.get(dbName) ??
         dbClients
-          .set(dbName, getDbClient({ connectionString, dbName: dbName }))
+          .set(
+            dbName,
+            getDbClient({
+              connectionString,
+              dbName: dbName,
+              client: options.client,
+            }),
+          )
           .get(dbName)!
       );
     },

src/packages/pongo/src/mongo/mongoClient.ts

@@ -1,12 +1,16 @@
 // src/MongoClientShim.ts
+import pg from 'pg';
 import { pongoClient, type PongoClient } from '../main';
 import { Db } from './mongoDb';
 
 export class MongoClient {
   private pongoClient: PongoClient;
 
-  constructor(connectionString: string) {
-    this.pongoClient = pongoClient(connectionString);
+  constructor(
+    connectionString: string,
+    options: { client?: pg.PoolClient } = {},
+  ) {
+    this.pongoClient = pongoClient(connectionString, options);
   }
 
   async connect() {

src/packages/pongo/src/postgres/client.ts

@@ -10,7 +10,7 @@ import { postgresCollection } from './postgresCollection';
 export type PongoClientOptions = {
   connectionString: string;
   dbName?: string | undefined;
-  client?: pg.PoolClient;
+  client?: pg.PoolClient | undefined;
 };
 
 export const postgresClient = (options: PongoClientOptions): DbClient => {