f

Author: killagu

src/operator.ts

@@ -77,7 +77,7 @@ export abstract class Operator {
     // query(sql, values)
     if (this.beforeQueryHandlers.length > 0) {
       for (const beforeQueryHandler of this.beforeQueryHandlers) {
-        const newSql = beforeQueryHandler(sql);
+        const newSql = beforeQueryHandler(sql, values);
         if (newSql) {
           sql = newSql;
         }
@@ -119,7 +119,7 @@ export abstract class Operator {
       } as QueryEndMessage);
       if (this.afterQueryHandlers.length > 0) {
         for (const afterQueryHandler of this.afterQueryHandlers) {
-          afterQueryHandler(sql, rows, duration, lastError);
+          afterQueryHandler(sql, rows, duration, lastError, values);
         }
       }
     }

src/types.ts

@@ -63,8 +63,8 @@ export type LockTableOption = {
   tableAlias: string;
 };
 
-export type BeforeQueryHandler = (sql: string) => string | undefined | void;
-export type AfterQueryHandler = (sql: string, result: any, execDuration: number, err?: Error) => void;
+export type BeforeQueryHandler = (sql: string, values?: object | any[]) => string | undefined | void;
+export type AfterQueryHandler = (sql: string, result: any, execDuration: number, err?: Error, values?: object | any[]) => void;
 
 export type TransactionContext = Record<PropertyKey, RDSTransaction | null>;
 export type TransactionScope = (transaction: RDSTransaction) => Promise<any>;