37 lines
1.9 KiB
JavaScript
37 lines
1.9 KiB
JavaScript
"use strict";
|
|
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
|
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
};
|
|
var TenantThrottlerGuard_1;
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.TenantThrottlerGuard = void 0;
|
|
const common_1 = require("@nestjs/common");
|
|
const throttler_1 = require("@nestjs/throttler");
|
|
let TenantThrottlerGuard = TenantThrottlerGuard_1 = class TenantThrottlerGuard extends throttler_1.ThrottlerGuard {
|
|
logger = new common_1.Logger(TenantThrottlerGuard_1.name);
|
|
async getTracker(req) {
|
|
const apiKeyId = req['apiKey']?.id || req.ip;
|
|
return `throttler:key:${apiKeyId}`;
|
|
}
|
|
async handleRequest(requestProps) {
|
|
const { context, limit, ttl } = requestProps;
|
|
const request = context.switchToHttp().getRequest();
|
|
const dynamicLimit = request['rateLimit'];
|
|
if (dynamicLimit) {
|
|
this.logger.debug(`Applying dynamic rate limit: ${dynamicLimit} req/min for key ${request['apiKey']?.id}`);
|
|
requestProps.limit = dynamicLimit;
|
|
}
|
|
else {
|
|
this.logger.debug(`Applying default rate limit: ${limit} req/min`);
|
|
}
|
|
return super.handleRequest(requestProps);
|
|
}
|
|
};
|
|
exports.TenantThrottlerGuard = TenantThrottlerGuard;
|
|
exports.TenantThrottlerGuard = TenantThrottlerGuard = TenantThrottlerGuard_1 = __decorate([
|
|
(0, common_1.Injectable)()
|
|
], TenantThrottlerGuard);
|
|
//# sourceMappingURL=rate-limiter.guard.js.map
|