Max/zkldi_Tachi
1
0
Fork 0
mirror of https://github.com/zkldi/Tachi.git synced 2026-10-06 13:58:05 +03:00
Code Issues Packages Projects Releases Wiki Activity

Use different stores for the rate limiter

Browse Source
This commit is contained in:
zkldi
2021-11-23 06:04:00 +00:00
parent 844ff12736
commit 40802322de
1 changed files with 17 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/src/server/middleware/rate-limiter.ts
+17 -8
View File
@@ -8,10 +8,11 @@ import { integer } from "tachi-common";
const logger = CreateLogCtx(__filename);
const store =
Environment.nodeEnv === "production" || Environment.nodeEnv === "staging"
? new RateLimitRedis({ prefix: `${TachiConfig.NAME}-RL:`, client: RedisClient })
function CreateStore(name: string) {
return Environment.nodeEnv === "production" || Environment.nodeEnv === "staging"
? new RateLimitRedis({ prefix: `${TachiConfig.NAME}-RL:${name}`, client: RedisClient })
: undefined; // undefined forces a default to an in-memory store
}
export function ClearTestingRateLimitCache() {
// ???
@@ -20,7 +21,11 @@ export function ClearTestingRateLimitCache() {
HyperAggressiveRateLimitMiddleware.resetKey(`::ffff:127.0.0.1`);
}
const CreateRateLimitOptions = (max: integer, windowMs?: number): rateLimit.Options => ({
const CreateRateLimitOptions = (
max: integer,
name: string,
windowMs?: number
): rateLimit.Options => ({
max,
onLimitReached: (req) => {
logger.warn(`User ${req.ip} hit rate limit.`, {
@@ -29,7 +34,7 @@ const CreateRateLimitOptions = (max: integer, windowMs?: number): rateLimit.Opti
hideFromConsole: ["req"],
});
},
store,
store: CreateStore(name),
message: {
success: false,
description: `You have exceeded ${max} requests per ${
@@ -42,12 +47,16 @@ const CreateRateLimitOptions = (max: integer, windowMs?: number): rateLimit.Opti
});
// 100 requests / minute is the current cap
export const NormalRateLimitMiddleware = rateLimit(CreateRateLimitOptions(ServerConfig.RATE_LIMIT));
export const NormalRateLimitMiddleware = rateLimit(
CreateRateLimitOptions(ServerConfig.RATE_LIMIT, "Normal")
);
// 15 requests every 10 minutes.
export const AggressiveRateLimitMiddleware = rateLimit(CreateRateLimitOptions(10, ONE_MINUTE * 10));
export const AggressiveRateLimitMiddleware = rateLimit(
CreateRateLimitOptions(10, "Aggressive", ONE_MINUTE * 10)
);
// 2 requests every 20 minutes.
export const HyperAggressiveRateLimitMiddleware = rateLimit(
CreateRateLimitOptions(2, ONE_MINUTE * 5)
CreateRateLimitOptions(2, "HyAgressive", ONE_MINUTE * 5)
);