java单机滑动窗口限流器的实现

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentLinkedDeque;
import java.util.concurrent.TimeUnit;
/**
 * 基于 Map 的滑动窗口限流器（10秒限频）
 */
public class LocalSlidingWindowRateLimiter {
    // 存储每个用户的请求时间戳队列
    private final Map<String, ConcurrentLinkedDeque<Long>> userQueueMap = new ConcurrentHashMap<>();
    private final int windowSeconds;   // 窗口大小（秒）
    private final int maxRequests;     // 窗口内最大请求数
    public LocalSlidingWindowRateLimiter(int windowSeconds, int maxRequests) {
        this.windowSeconds = windowSeconds;
        this.maxRequests = maxRequests;
    }
    /**
     * 检查是否允许请求
     * @param key 限流标识（如 userId、ip）
     * @return true 允许，false 被限流
     */
    public boolean allowRequest(String key) {
        long now = System.currentTimeMillis();
        long windowStart = now - TimeUnit.SECONDS.toMillis(windowSeconds);
        // 获取或创建该用户的队列
        ConcurrentLinkedDeque<Long> deque = userQueueMap.computeIfAbsent(key,
                k -> new ConcurrentLinkedDeque<>());
        // 对队列加锁，保证清理和添加的原子性
        synchronized (deque) {
            // 1. 清理过期记录（窗口外的旧时间戳）
            while (!deque.isEmpty() && deque.peekFirst() < windowStart) {
                deque.pollFirst();
            }
            // 2. 判断是否超过阈值
            if (deque.size() < maxRequests) {
                // 允许请求，记录当前时间戳
                deque.addLast(now);
                return true;
            } else {
                return false;
            }
        }
    }
    // 可选：定期清理无数据的用户 Key，避免内存泄漏（可以用 ScheduledExecutorService 定时清理）
    public void cleanEmptyKeys() {
        userQueueMap.entrySet().removeIf(entry -> entry.getValue().isEmpty());
    }
}

public class Main {
    public static void main(String[] args) throws InterruptedException {
        // 限流规则：10秒内最多5次请求
        LocalSlidingWindowRateLimiter limiter = new LocalSlidingWindowRateLimiter(10, 5);
        String userId = "user_123";
        for (int i = 0; i < 10; i++) {
            boolean allowed = limiter.allowRequest(userId);
            System.out.println("请求 " + (i + 1) + "：" + (allowed ? "通过" : "限流"));
            Thread.sleep(1000); // 模拟间隔 1 秒
        }
    }
}