Skip to content

Commit 88368b8

Browse files
Rework rate-limiter
1 parent ff9e2e6 commit 88368b8

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
@@ -273,12 +273,13 @@ export default class RichEditorPlugin extends AdminForthPlugin {
273273

274274
if (this.options.completion.rateLimit?.limit) {
275275
// rate limit
276-
const { error } = RateLimiter.checkRateLimit(
277-
this.pluginInstanceId,
278-
this.options.completion.rateLimit?.limit,
279-
this.adminforth.auth.getClientIp(headers),
280-
);
281-
if (error) {
276+
// const { error } = RateLimiter.checkRateLimit(
277+
// this.pluginInstanceId,
278+
// this.options.completion.rateLimit?.limit,
279+
// this.adminforth.auth.getClientIp(headers),
280+
// );
281+
const rateLimiter = new RateLimiter(this.options.completion.rateLimit?.limit);
282+
if (!rateLimiter.consume(`${this.pluginInstanceId}-${this.adminforth.auth.getClientIp(headers)}`)) {
282283
return {
283284
completion: [],
284285
}

0 commit comments

Comments
 (0)