feat(decorator): add setThrottlerMetadata() function back

nestjs · Jun 12, 2020 · ea31a9c · ea31a9c
1 parent 4be7fab
commit ea31a9c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/src/throttle.decorator.ts b/src/throttle.decorator.ts
@@ -1,5 +1,10 @@
 import { THROTTLER_LIMIT, THROTTLER_SKIP, THROTTLER_TTL } from './throttler.constants';
 
+function setThrottlerMetadata(target: any, limit: number, ttl: number): void {
+  Reflect.defineMetadata(THROTTLER_TTL, ttl, target);
+  Reflect.defineMetadata(THROTTLER_LIMIT, limit, target);
+}
+
 /**
  * Adds metadata to the target which will be handled by the ThrottlerGuard to
  * handle incoming requests based on the given metadata.
@@ -12,12 +17,10 @@ export const Throttle = (limit = 20, ttl = 60): MethodDecorator & ClassDecorator
     descriptor?: TypedPropertyDescriptor<any>,
   ) => {
     if (descriptor) {
-      Reflect.defineMetadata(THROTTLER_LIMIT, limit, descriptor.value);
-      Reflect.defineMetadata(THROTTLER_TTL, ttl, descriptor.value);
+      setThrottlerMetadata(descriptor.value, limit, ttl);
       return descriptor;
     }
-    Reflect.defineMetadata(THROTTLER_LIMIT, limit, target);
-    Reflect.defineMetadata(THROTTLER_TTL, ttl, target);
+    setThrottlerMetadata(target, limit, ttl);
     return target;
   };
 };

diff --git a/src/throttler.guard.ts b/src/throttler.guard.ts
@@ -118,8 +118,8 @@ export class ThrottlerGuard implements CanActivate {
   ): Promise<boolean> {
     const client = context.switchToWs().getClient();
     const ip = ['conn', '_socket']
-      .map(key => client[key])
-      .filter(obj => obj)
+      .map((key) => client[key])
+      .filter((obj) => obj)
       .shift().remoteAddress;
     const key = this.generateKey(context, ip);
     const ttls = await this.storageService.getRecord(key);