Обучение API -ограничения с использованием ведра токена ⇐ Javascript

[Anonymous]

Я использовал алгоритм токенового ведра для реализации ограничения скорости API. Кажется, это работает, но можете ли вы помочь мне улучшить это? Я планирую использовать Redis позже, а не в хранении данных памяти. < /p>
Спасибо. < /p>
import { Request, Response } from "express";

interface Iuser {
userName: string;
expireDate: number;
count: number;
}

type Iusers = Record;

const users: Iusers = {};

export class TokenBucket {
count = 0;
time = Date.now();
maxLimit: number = 3;
constructor() {
setInterval(() => {
this.expireRequest(users);
}, 5000);
}

addrequest(req: Request, res: Response, successFn: Function) {
let user = req.headers.authorization as string;

if (!users[user]) {
users[user] = {
count: 1,
expireDate: Date.now() + 5000,
userName: user,
};
this.count = users[user].count;
} else {
this.count = ++users[user].count;
}

if (this.count > this.maxLimit) {
console.log("rate limit excetion for ", users[user].userName);
return res.status(429).json({
message: "rate limit exceeded. please try again",
});
}
return successFn();
}

expireRequest(users: Iusers) {
for (let key in users) {
console.log("inside loop");
const user = users[key];
if (Date.now() > user.expireDate && user.count !== 0) {
console.log("inisdide reset for ", user.userName);
user.expireDate = Date.now() + 5000;
user.count = 0;
}
}
}
}

< /code>
Объяснение:
всякий раз, когда приходит запрос, если он существует в пользователях obj. Он увеличивает свой номер на 1, и если не существует, он добавляет имя пользователя в качестве свойства и инициализируется с 1 как значение.
Существует также SetInterval, который выполняет функцию expirErequest < /code>, который сбрасывает счетчик для пользователя, если текущее время больше, чем истек срок действия этого пользователя. меньше, чем текущая дата.

Подробнее здесь: https://stackoverflow.com/questions/796 ... ken-bucket