Я использовал алгоритм токенового ведра для реализации ограничения скорости 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
