Skip to content

Commit ce8262c

Browse files
Rework rate-limiter
1 parent b71c616 commit ce8262c

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
@@ -346,12 +346,13 @@ export default class UploadPlugin extends AdminForthPlugin {
346346
const { prompt, recordId } = body;
347347
if (this.options.generation.rateLimit?.limit) {
348348
// rate limit
349-
const { error } = RateLimiter.checkRateLimit(
350-
this.pluginInstanceId,
351-
this.options.generation.rateLimit?.limit,
352-
this.adminforth.auth.getClientIp(headers),
353-
);
354-
if (error) {
349+
// const { error } = RateLimiter.checkRateLimit(
350+
// this.pluginInstanceId,
351+
// this.options.generation.rateLimit?.limit,
352+
// this.adminforth.auth.getClientIp(headers),
353+
// );
354+
const rateLimiter = new RateLimiter(this.options.generation.rateLimit?.limit);
355+
if (!rateLimiter.consume(`${this.pluginInstanceId}-${this.adminforth.auth.getClientIp(headers)}`)) {
355356
return { error: this.options.generation.rateLimit.errorMessage };
356357
}
357358
}

0 commit comments

Comments
 (0)