feat: pass context to get tracker fn as second arg

nestjs · Dec 10, 2024 · a3d602e · a3d602e
1 parent 52608f3
commit a3d602e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/throttler-module-options.interface.ts b/src/throttler-module-options.interface.ts
@@ -125,7 +125,10 @@ export interface ThrottlerAsyncOptions extends Pick<ModuleMetadata, 'imports'> {
 /**
  * @publicApi
  */
-export type ThrottlerGetTrackerFunction = (req: Record<string, any>) => Promise<string> | string;
+export type ThrottlerGetTrackerFunction = (
+  req: Record<string, any>,
+  context: ExecutionContext,
+) => Promise<string> | string;
 
 /**
  * @publicApi

diff --git a/src/throttler.guard.ts b/src/throttler.guard.ts
@@ -165,7 +165,7 @@ export class ThrottlerGuard implements CanActivate {
         }
       }
     }
-    const tracker = await getTracker(req);
+    const tracker = await getTracker(req, context);
     const key = generateKey(context, tracker, throttler.name);
     const { totalHits, timeToExpire, isBlocked, timeToBlockExpire } =
       await this.storageService.increment(key, ttl, limit, blockDuration, throttler.name);