yuken 3 лет назад
Родитель
Сommit
2eeb7b529c
1 измененных файлов с 7 добавлено и 8 удалено
  1. 7 8
      packages/app/src/server/middlewares/api-rate-limiter.ts

+ 7 - 8
packages/app/src/server/middlewares/api-rate-limiter.ts

@@ -37,15 +37,14 @@ module.exports = () => {
     const endpoint = req.path;
     const endpoint = req.path;
     const key = req.ip + endpoint;
     const key = req.ip + endpoint;
 
 
-    let points = defaultConsumePoints;
-    Object.keys(apiRateLimitConfig).forEach((endpointInConfig) => {
-      if (endpointInConfig === endpoint) {
-        const consumePointsInConfig = apiRateLimitConfig[endpointInConfig].consumePoints;
-        points = consumePointsInConfig;
-      }
-    });
+    const customizedConfig = apiRateLimitConfig[endpoint];
+
+    if (customizedConfig === undefined) {
+      await consumePoints(rateLimiter, key, defaultConsumePoints, next);
+      return;
+    }
 
 
-    await consumePoints(rateLimiter, key, points, next);
+    await consumePoints(rateLimiter, key, customizedConfig.consumePoints, next);
     return;
     return;
   };
   };
 };
 };