Mizah/mizzkey
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: return rate limit instead of throwing an object

Browse source
This commit is contained in:
anatawa12 2024-06-22 13:34:52 +09:00
parent fce656b5e9
commit f9dd3aca78
No known key found for this signature in database
GPG key ID: 9CA909848B8E4EA6
3 changed files with 26 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

20
packages/backend/src/server/api/RateLimiterService.ts
View file

@ -12,6 +12,14 @@ import { LoggerService } from '@/core/LoggerService.js';
import { bindThis } from '@/decorators.js';
import type { IEndpointMeta } from './endpoints.js';
type RateLimitInfo = {
code: 'BRIEF_REQUEST_INTERVAL',
info: Limiter.LimiterInfo,
} | {
code: 'RATE_LIMIT_EXCEEDED',
info: Limiter.LimiterInfo,
}
@Injectable()
export class RateLimiterService {
private logger: Logger;
@ -35,7 +43,7 @@ export class RateLimiterService {
return new Promise<Limiter.LimiterInfo>((resolve, reject) => {
new Limiter(options).get((err, info) => {
if (err) {
return reject({ code: 'ERR', info });
return reject(err);
}
resolve(info);
});
@ -43,9 +51,9 @@ export class RateLimiterService {
}
@bindThis
public async limit(limitation: IEndpointMeta['limit'] & { key: NonNullable<string> }, actor: string, factor = 1) {
public async limit(limitation: IEndpointMeta['limit'] & { key: NonNullable<string> }, actor: string, factor = 1): Promise<RateLimitInfo | null> {
if (this.disabled) {
return;
return null;
}
// Short-term limit
@ -61,7 +69,7 @@ export class RateLimiterService {
if (info.remaining === 0) {
// eslint-disable-next-line no-throw-literal
throw { code: 'BRIEF_REQUEST_INTERVAL', info };
return { code: 'BRIEF_REQUEST_INTERVAL', info };
}
}
@ -78,8 +86,10 @@ export class RateLimiterService {
if (info.remaining === 0) {
// eslint-disable-next-line no-throw-literal
throw { code: 'RATE_LIMIT_EXCEEDED', info };
return { code: 'RATE_LIMIT_EXCEEDED', info };
}
}
return null
}
}