feat: añade limitador de tasa por usuario (15/min) y tests
Co-authored-by: aider (openrouter/openai/gpt-5) <aider@aider.chat>pull/1/head
							parent
							
								
									432409e246
								
							
						
					
					
						commit
						9668802cbe
					
				| @ -0,0 +1,66 @@ | ||||
| /** | ||||
|  * RateLimiter - Token Bucket en memoria por usuario. | ||||
|  * - Límite global por remitente (ID normalizado). | ||||
|  * - Capacidad y tasa configurables por entorno: | ||||
|  *   - RATE_LIMIT_PER_MIN: tokens que se recargan por minuto (default 15) | ||||
|  *   - RATE_LIMIT_BURST: capacidad del bucket (default = RATE_LIMIT_PER_MIN) | ||||
|  * - Desactivación en tests: la integración en server evita usar el limiter con NODE_ENV='test'. | ||||
|  */ | ||||
| export class RateLimiter { | ||||
|   private static readonly buckets = new Map<string, { tokens: number; last: number }>(); | ||||
|   private static readonly lastNotifyAt = new Map<string, number>(); | ||||
|   private static readonly NOTIFY_COOLDOWN_MS = 60_000; // 1 minuto
 | ||||
| 
 | ||||
|   private static get ratePerMin(): number { | ||||
|     const v = Number(process.env.RATE_LIMIT_PER_MIN); | ||||
|     return Number.isFinite(v) && v > 0 ? v : 15; | ||||
|   } | ||||
| 
 | ||||
|   private static get capacity(): number { | ||||
|     const burst = Number(process.env.RATE_LIMIT_BURST); | ||||
|     if (Number.isFinite(burst) && burst > 0) return burst; | ||||
|     return this.ratePerMin; // por defecto, burst = rate
 | ||||
|   } | ||||
| 
 | ||||
|   private static get refillPerMs(): number { | ||||
|     return this.ratePerMin / 60_000; // tokens por ms
 | ||||
|   } | ||||
| 
 | ||||
|   static checkAndConsume(userId: string, now: number = Date.now()): boolean { | ||||
|     if (!userId) return true; // por seguridad, no bloquear IDs vacíos
 | ||||
| 
 | ||||
|     let bucket = this.buckets.get(userId); | ||||
|     if (!bucket) { | ||||
|       bucket = { tokens: this.capacity, last: now }; | ||||
|       this.buckets.set(userId, bucket); | ||||
|     } | ||||
| 
 | ||||
|     // Refill lineal en función del tiempo transcurrido
 | ||||
|     const elapsed = Math.max(0, now - bucket.last); | ||||
|     if (elapsed > 0) { | ||||
|       bucket.tokens = Math.min(this.capacity, bucket.tokens + elapsed * this.refillPerMs); | ||||
|       bucket.last = now; | ||||
|     } | ||||
| 
 | ||||
|     if (bucket.tokens >= 1) { | ||||
|       bucket.tokens -= 1; | ||||
|       return true; | ||||
|     } | ||||
|     return false; | ||||
|   } | ||||
| 
 | ||||
|   static shouldNotify(userId: string, now: number = Date.now()): boolean { | ||||
|     const last = this.lastNotifyAt.get(userId) || 0; | ||||
|     if (now - last >= this.NOTIFY_COOLDOWN_MS) { | ||||
|       this.lastNotifyAt.set(userId, now); | ||||
|       return true; | ||||
|     } | ||||
|     return false; | ||||
|   } | ||||
| 
 | ||||
|   // Solo para tests
 | ||||
|   static reset(): void { | ||||
|     this.buckets.clear(); | ||||
|     this.lastNotifyAt.clear(); | ||||
|   } | ||||
| } | ||||
| @ -0,0 +1,57 @@ | ||||
| import { describe, it, expect, beforeEach } from 'bun:test'; | ||||
| import { RateLimiter } from '../../../src/services/rate-limit'; | ||||
| 
 | ||||
| describe('RateLimiter - token bucket básico', () => { | ||||
|   beforeEach(() => { | ||||
|     RateLimiter.reset(); | ||||
|     // No dependemos de NODE_ENV aquí; testea la clase aislada.
 | ||||
|     process.env.RATE_LIMIT_PER_MIN = '15'; | ||||
|     process.env.RATE_LIMIT_BURST = '15'; | ||||
|   }); | ||||
| 
 | ||||
|   it('permite hasta el burst inicial y bloquea al exceder', () => { | ||||
|     const user = '1234567890'; | ||||
|     const t0 = 0; | ||||
| 
 | ||||
|     // 15 primeros deben pasar
 | ||||
|     for (let i = 0; i < 15; i++) { | ||||
|       const ok = RateLimiter.checkAndConsume(user, t0); | ||||
|       expect(ok).toBe(true); | ||||
|     } | ||||
| 
 | ||||
|     // 16º sin tiempo transcurrido debe bloquear
 | ||||
|     const denied = RateLimiter.checkAndConsume(user, t0); | ||||
|     expect(denied).toBe(false); | ||||
|   }); | ||||
| 
 | ||||
|   it('recupera tokens con el tiempo (refill lineal por minuto)', () => { | ||||
|     const user = '1234567890'; | ||||
|     const t0 = 0; | ||||
| 
 | ||||
|     // Agotar bucket
 | ||||
|     for (let i = 0; i < 15; i++) { | ||||
|       expect(RateLimiter.checkAndConsume(user, t0)).toBe(true); | ||||
|     } | ||||
|     expect(RateLimiter.checkAndConsume(user, t0)).toBe(false); // agotado
 | ||||
| 
 | ||||
|     // Tras 60s debe haberse recuperado capacidad suficiente para permitir de nuevo
 | ||||
|     const t1 = 60_000; | ||||
|     expect(RateLimiter.checkAndConsume(user, t1)).toBe(true); | ||||
| 
 | ||||
|     // Y puede consumir varios más (hasta ~15 por minuto)
 | ||||
|     let count = 1; | ||||
|     while (RateLimiter.checkAndConsume(user, t1)) { | ||||
|       count++; | ||||
|       if (count > 20) break; // seguridad
 | ||||
|     } | ||||
|     expect(count).toBeGreaterThan(1); | ||||
|   }); | ||||
| 
 | ||||
|   it('shouldNotify limita el aviso a como mucho 1/min por usuario', () => { | ||||
|     const user = 'u1'; | ||||
|     const t0 = 0; | ||||
|     expect(RateLimiter.shouldNotify(user, t0)).toBe(true);  // primera vez notifica
 | ||||
|     expect(RateLimiter.shouldNotify(user, t0 + 10_000)).toBe(false); // dentro de la ventana
 | ||||
|     expect(RateLimiter.shouldNotify(user, t0 + 60_000)).toBe(true); // pasado el cooldown
 | ||||
|   }); | ||||
| }); | ||||
					Loading…
					
					
				
		Reference in New Issue