Skip to content

Commit f112667

Browse files
Rework rate-limiter
1 parent f130871 commit f112667

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

index.ts

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -57,12 +57,13 @@ export default class BulkAiFlowPlugin extends AdminForthPlugin {
5757
private checkRateLimit(field: string,fieldNameRateLimit: string | undefined, headers: Record<string, string | string[] | undefined>): { error?: string } | void {
5858
if (fieldNameRateLimit) {
5959
// rate limit
60-
const { error } = RateLimiter.checkRateLimit(
61-
field,
62-
fieldNameRateLimit,
63-
this.adminforth.auth.getClientIp(headers),
64-
);
65-
if (error) {
60+
// const { error } = RateLimiter.checkRateLimit(
61+
// field,
62+
// fieldNameRateLimit,
63+
// this.adminforth.auth.getClientIp(headers),
64+
// );
65+
const rateLimiter = new RateLimiter(fieldNameRateLimit);
66+
if (!rateLimiter.consume(`${field}-${this.adminforth.auth.getClientIp(headers)}`)) {
6667
return { error: "Rate limit exceeded" };
6768
}
6869
}

0 commit comments

Comments
 (0)