This commit is contained in:
Nicolas
2024-07-16 22:41:13 -04:00
parent 92202de12b
commit c9073a747c
5 changed files with 127 additions and 61 deletions
+6 -7
View File
@@ -1,6 +1,6 @@
import { RateLimiterRedis } from "rate-limiter-flexible";
import * as redis from "redis";
import { RateLimiterMode } from "../../src/types";
import Redis from "ioredis";
const RATE_LIMITS = {
crawl: {
@@ -57,14 +57,13 @@ const RATE_LIMITS = {
},
};
export const redisClient = redis.createClient({
url: process.env.REDIS_RATE_LIMIT_URL,
legacyMode: true,
});
export const redisRateLimitClient = new Redis(
process.env.REDIS_RATE_LIMIT_URL
)
const createRateLimiter = (keyPrefix, points) =>
new RateLimiterRedis({
storeClient: redisClient,
storeClient: redisRateLimitClient,
keyPrefix,
points,
duration: 60, // Duration in seconds
@@ -76,7 +75,7 @@ export const serverRateLimiter = createRateLimiter(
);
export const testSuiteRateLimiter = new RateLimiterRedis({
storeClient: redisClient,
storeClient: redisRateLimitClient,
keyPrefix: "test-suite",
points: 10000,
duration: 60, // Duration in seconds