| 50 | const userRateLimits = new Map<string, RateLimiter>(); |
| 51 | |
| 52 | export const rateLimit = (userId: string, rateLimitCount: number, rateLimitInterval: number): boolean => { |
| 53 | const now = new Date(); |
| 54 | const userLimiter = userRateLimits.get(userId) ?? { timestamps: [] }; |
| 55 | |
| 56 | userLimiter.timestamps = userLimiter.timestamps.filter( |
| 57 | (timestamp) => now.getTime() - timestamp.getTime() < rateLimitInterval |
| 58 | ); |
| 59 | |
| 60 | if (userLimiter.timestamps.length >= rateLimitCount) { |
| 61 | return false; // Rate limit exceeded |
| 62 | } |
| 63 | |
| 64 | userLimiter.timestamps.push(now); |
| 65 | userRateLimits.set(userId, userLimiter); |
| 66 | return true; |
| 67 | }; |