@canmertinyo/rate-limiter-redis/dist/redis.storage.js

A simple rate-limiting middleware for Express.js with support for in-memory, Redis, and MongoDB storage

"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.RedisStorage = void 0;
const ioredis_1 = require("ioredis");
class RedisStorage {
    constructor(options) {
        this.options = options;
    }
    initialize() {
        return __awaiter(this, void 0, void 0, function* () {
            this.redis = new ioredis_1.default(this.options);
            return this;
        });
    }
    getRateLimitRecord(key) {
        return __awaiter(this, void 0, void 0, function* () {
            const data = yield this.redis.get(key);
            return data ? JSON.parse(data) : undefined;
        });
    }
    createRateLimitRecord(record) {
        return __awaiter(this, void 0, void 0, function* () {
            yield this.redis.set(record.key, JSON.stringify(record));
            return record;
        });
    }
    updateRateLimitRecord(key, timestamp, count) {
        return __awaiter(this, void 0, void 0, function* () {
            const record = yield this.getRateLimitRecord(key);
            if (!record)
                throw new Error("Record not found");
            const updatedRecord = Object.assign(Object.assign({}, record), { count,
                timestamp });
            yield this.redis.set(key, JSON.stringify(updatedRecord));
            return updatedRecord;
        });
    }
    increment(key) {
        return __awaiter(this, void 0, void 0, function* () {
            const record = yield this.getRateLimitRecord(key);
            if (!record)
                throw new Error("Record not found");
            const updatedRecord = Object.assign(Object.assign({}, record), { count: record.count + 1 });
            yield this.redis.set(key, JSON.stringify(updatedRecord));
            return updatedRecord;
        });
    }
}
exports.RedisStorage = RedisStorage;