blob: db90dcff4fa8a70491ae21b731744f0f2d9d9df1 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
// lib/rate-limit.ts
import { NextRequest } from 'next/server';
import { getClientIp } from './network/get-client-ip';
interface RateLimitResult {
success: boolean;
remaining?: number;
}
// 메모리 기반 간단한 rate limiter (프로덕션에서는 Redis 사용 권장)
const requestCounts = new Map<string, { count: number; resetTime: number }>();
export default async function rateLimit(
request: NextRequest,
maxRequests: number = 100,
windowMs: number = 60 * 60 * 1000 // 1시간
): Promise<RateLimitResult> {
// 클라이언트 IP 가져오기
const clientIP = getClientIp(request);
const now = Date.now();
const key = `rate_limit:${clientIP}`;
// 기존 요청 정보 가져오기
let requestInfo = requestCounts.get(key);
// 윈도우가 리셋된 경우 또는 첫 요청인 경우
if (!requestInfo || now > requestInfo.resetTime) {
requestInfo = {
count: 1,
resetTime: now + windowMs
};
requestCounts.set(key, requestInfo);
return { success: true, remaining: maxRequests - 1 };
}
// 요청 한도 초과 확인
if (requestInfo.count >= maxRequests) {
return { success: false, remaining: 0 };
}
// 요청 카운트 증가
requestInfo.count++;
requestCounts.set(key, requestInfo);
return {
success: true,
remaining: maxRequests - requestInfo.count
};
}
// 주기적으로 만료된 항목 정리 (메모리 누수 방지)
setInterval(() => {
const now = Date.now();
for (const [key, value] of requestCounts.entries()) {
if (now > value.resetTime) {
requestCounts.delete(key);
}
}
}, 60 * 60 * 1000); // 1시간마다 정리
|
