nestjs-throttler-storage-mongo
Version:
Mongo storage provider for nestjs/throttler
67 lines • 2.77 kB
JavaScript
;
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 __metadata = (this && this.__metadata) || function (k, v) {
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.ThrottlerStorageMongoService = void 0;
const common_1 = require("@nestjs/common");
const mongodb_1 = require("mongodb");
let ThrottlerStorageMongoService = exports.ThrottlerStorageMongoService = class ThrottlerStorageMongoService {
get storage() {
return this._storage;
}
constructor(url, mongoOptions) {
this._storage = {};
this.mongoClient = new mongodb_1.MongoClient(url, mongoOptions);
(async () => {
try {
await this.mongoClient.connect();
await this.createTTLIndex();
}
catch (error) {
console.error('Error connecting to MongoDB:', error);
}
})();
}
async createTTLIndex() {
await this.mongoClient
.db()
.collection('throttler')
.createIndex({ expireAt: 1 }, { expireAfterSeconds: 0 });
}
async increment(key, ttl) {
const ttlMilliseconds = ttl * 1000;
const result = await this.mongoClient
.db()
.collection('throttler')
.findOneAndUpdate({ key }, {
$inc: { totalHits: 1 },
$set: { expireAt: new Date(Date.now() + ttlMilliseconds) },
}, {
upsert: true,
returnDocument: 'after',
});
const { totalHits, expireAt } = result.value;
const timeToExpire = Math.max(0, Math.floor((expireAt.getTime() - Date.now()) / 1000));
return {
totalHits,
timeToExpire,
};
}
async onApplicationShutdown() {
if (this.mongoClient) {
await this.mongoClient.close();
}
}
};
exports.ThrottlerStorageMongoService = ThrottlerStorageMongoService = __decorate([
(0, common_1.Injectable)(),
__metadata("design:paramtypes", [String, Object])
], ThrottlerStorageMongoService);
//# sourceMappingURL=throttler-storage-mongo.service.js.map