feat(ai): simpler In-Memory-Ratelimiter pro IP
This commit is contained in:
21
src/lib/server/ai/rate-limit.ts
Normal file
21
src/lib/server/ai/rate-limit.ts
Normal file
@@ -0,0 +1,21 @@
|
||||
export type RateLimiter = { check: (key: string) => boolean };
|
||||
|
||||
export function createRateLimiter(opts: {
|
||||
windowMs: number;
|
||||
max: number;
|
||||
}): RateLimiter {
|
||||
const store = new Map<string, { count: number; resetAt: number }>();
|
||||
return {
|
||||
check(key: string): boolean {
|
||||
const now = Date.now();
|
||||
const entry = store.get(key);
|
||||
if (!entry || entry.resetAt <= now) {
|
||||
store.set(key, { count: 1, resetAt: now + opts.windowMs });
|
||||
return true;
|
||||
}
|
||||
if (entry.count >= opts.max) return false;
|
||||
entry.count += 1;
|
||||
return true;
|
||||
}
|
||||
};
|
||||
}
|
||||
29
tests/unit/rate-limit.test.ts
Normal file
29
tests/unit/rate-limit.test.ts
Normal file
@@ -0,0 +1,29 @@
|
||||
import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
|
||||
import { createRateLimiter } from '../../src/lib/server/ai/rate-limit';
|
||||
|
||||
describe('rate-limit', () => {
|
||||
beforeEach(() => vi.useFakeTimers());
|
||||
afterEach(() => vi.useRealTimers());
|
||||
|
||||
it('allows first 10 requests, rejects 11th', () => {
|
||||
const limiter = createRateLimiter({ windowMs: 60_000, max: 10 });
|
||||
for (let i = 0; i < 10; i++) expect(limiter.check('1.2.3.4')).toBe(true);
|
||||
expect(limiter.check('1.2.3.4')).toBe(false);
|
||||
});
|
||||
|
||||
it('tracks per-IP independently', () => {
|
||||
const limiter = createRateLimiter({ windowMs: 60_000, max: 2 });
|
||||
expect(limiter.check('a')).toBe(true);
|
||||
expect(limiter.check('a')).toBe(true);
|
||||
expect(limiter.check('a')).toBe(false);
|
||||
expect(limiter.check('b')).toBe(true);
|
||||
});
|
||||
|
||||
it('resets after window elapses', () => {
|
||||
const limiter = createRateLimiter({ windowMs: 1000, max: 1 });
|
||||
expect(limiter.check('x')).toBe(true);
|
||||
expect(limiter.check('x')).toBe(false);
|
||||
vi.advanceTimersByTime(1001);
|
||||
expect(limiter.check('x')).toBe(true);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user