feat: add default tx options

Papooch · Apr 5, 2024 · 9799711 · 9799711
1 parent 753b592
commit 9799711
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 4 deletions.
diff --git a/.../docs/06_plugins/01_available-plugins/01-transactional/04-pg-promise-adapter.md b/.../docs/06_plugins/01_available-plugins/01-transactional/04-pg-promise-adapter.md
@@ -32,6 +32,8 @@ pnpm add @nestjs-cls/transactional-adapter-pg-promise
 ## Registration
 
 ```ts
+import { txMode } from 'pg-promise'
+
 ClsModule.forRoot({
     plugins: [
         new ClsPluginTransactional({
@@ -42,6 +44,9 @@ ClsModule.forRoot({
             adapter: new TransactionalAdapterPgPromise({
                 // the injection token of the database instance
                 dbInstanceToken: DB,
+
+                // default transaction options (optional)
+                defaultTxOptions: { mode: new txMode.TransactionMode({ tiLevel: txMode.isolationLevel.serializable }) }
             }),
         }),
     ],

diff --git a/...nal-adapters/transactional-adapter-pg-promise/src/lib/transactional-adapter-pg-promise.ts b/...nal-adapters/transactional-adapter-pg-promise/src/lib/transactional-adapter-pg-promise.ts
@@ -10,15 +10,20 @@ export interface PgPromiseTransactionalAdapterOptions {
      * The injection token for the pg-promise instance.
      */
     dbInstanceToken: any;
+
+    defaultTxOptions?: TxOptions;
 }
 
 export class TransactionalAdapterPgPromise
     implements TransactionalAdapter<Database, Database, any>
 {
     connectionToken: any;
 
+    defaultTxOptions?: TxOptions;
+
     constructor(options: PgPromiseTransactionalAdapterOptions) {
         this.connectionToken = options.dbInstanceToken;
+        this.defaultTxOptions = options.defaultTxOptions;
     }
 
     optionsFactory = (pgPromiseDbInstance: Database) => ({
@@ -27,10 +32,13 @@ export class TransactionalAdapterPgPromise
             fn: (...args: any[]) => Promise<any>,
             setClient: (client?: Database) => void,
         ) => {
-            return pgPromiseDbInstance.tx(options ?? {}, (tx) => {
-                setClient(tx as unknown as Database);
-                return fn();
-            });
+            return pgPromiseDbInstance.tx(
+                options ?? this.defaultTxOptions ?? {},
+                (tx) => {
+                    setClient(tx as unknown as Database);
+                    return fn();
+                },
+            );
         },
         getFallbackInstance: () => pgPromiseDbInstance,
     });