Add query as final argument to resultListener (addresses #181)

jawj · Oct 2, 2024 · 2128797 · 2128797
1 parent 505c7b9
commit 2128797
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/db/config.ts b/src/db/config.ts
@@ -16,7 +16,7 @@ export interface Config {
   castArrayParamsToJson: boolean;   // see https://github.com/brianc/node-postgres/issues/2012
   castObjectParamsToJson: boolean;  // useful if json will be cast onward differently from text
   queryListener?(query: SQLQuery, txnId?: number): void;
-  resultListener?(result: any, txnId?: number, elapsedMs?: number): void;
+  resultListener?(result: any, txnId?: number, elapsedMs?: number, query?: SQLQuery): void;
   transactionListener?(message: string, txnId?: number): void;
 }
 export type NewConfig = Partial<Config>;

diff --git a/src/db/core.ts b/src/db/core.ts
@@ -280,7 +280,7 @@ export class SQLFragment<RunResult = pg.QueryResult['rows'], Constraint = never>
       result = this.noopResult;
     }
 
-    if (resultListener) resultListener(result, txnId, timing() - startMs!);
+    if (resultListener) resultListener(result, txnId, timing() - startMs!, query);
     return result;
   };