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