滑动窗口限流

大家好，我是神韵，是一个技术&生活博主。出文章目的主要是两个，一是好记忆不如烂笔头，记录总结中提高自己。二是希望我的文章可以帮到大家。欢迎大家留言讨论，你们的行动将是我无限的动力。
本篇主题是：滑动窗口限流

滑动窗口限流，一分钟限流10次请求，代码带注释。

// 限流 一分钟 10次
    int countPerSecond = 10;
    // 1分钟分成6个窗口 每10s一个窗口（整数 0 10 20 30 40 50） 越细越准 需要有序
    TreeMap<Long, Integer> timeMapCount = new TreeMap<>();

    public boolean slideWindow() {
        // 当前的情求次数
        int reqCount = 0;
        // 获取当前窗口时间 秒 北京时区 东八区是 ZoneOffset.UTC
        // /10 * 10的含义是，比如当前是 10:42:26 ,26秒，我们就获取 20秒，因为存的窗口节点是20秒
        long currentWindowTime = LocalDateTime.now().toEpochSecond(ZoneOffset.of("+8")) / 10 * 10;
        // 获取当前时间的开始时间窗口，-50s就可以 当前时间永远是时间段的最前面一段子窗口时间
        long startWindowTime = currentWindowTime - 50;
        // 获取当前时间的情求次数
        Iterator<Map.Entry<Long, Integer>> iterator = timeMapCount.entrySet().iterator();
        while (iterator.hasNext()) {
            Map.Entry<Long, Integer> item = iterator.next();
            if (item.getKey() < startWindowTime) {
                // 淘汰窗口时间
                iterator.remove();
            } else {
                reqCount += item.getValue();
                if (reqCount >= countPerSecond) {
                    return false;
                }
            }
        }
        timeMapCount.merge(currentWindowTime, 1, Integer::sum);
        // 如果该时间段已经有过其它请求，则在原基础上叠加1 并覆盖原来的时间窗口map
//        int newReqCount = timeMapCount.get(currentWindowTime) != null ? timeMapCount.get(currentWindowTime) + 1: 1;
//        timeMapCount.put(currentWindowTime, newReqCount);
        return true;
    }

测试代码

        public static void main(String[] args) throws InterruptedException {
            while (true) {
                System.out.println(slideWindow() + "," + LocalDateTime.now());
                Thread.sleep(2000);
            }
        }

测试结果
可以看到限流在38分46秒时开始，在39分04秒时达到了10个请求，所以后面都是被限流了false，直到39分40多秒才有新的请求进来，所以可以看到39分40秒是true。

true,2023-10-27T14:38:46.138
true,2023-10-27T14:38:48.141
true,2023-10-27T14:38:50.145
true,2023-10-27T14:38:52.147
true,2023-10-27T14:38:54.153
true,2023-10-27T14:38:56.158
true,2023-10-27T14:38:58.163
true,2023-10-27T14:39:00.169
true,2023-10-27T14:39:02.174
true,2023-10-27T14:39:04.179
false,2023-10-27T14:39:06.184
false,2023-10-27T14:39:08.187
false,2023-10-27T14:39:10.192
false,2023-10-27T14:39:12.194
false,2023-10-27T14:39:14.196
false,2023-10-27T14:39:16.201
false,2023-10-27T14:39:18.206
false,2023-10-27T14:39:20.212
false,2023-10-27T14:39:22.217
false,2023-10-27T14:39:24.222
false,2023-10-27T14:39:26.228
false,2023-10-27T14:39:28.233
false,2023-10-27T14:39:30.238
false,2023-10-27T14:39:32.243
false,2023-10-27T14:39:34.249
false,2023-10-27T14:39:36.250
false,2023-10-27T14:39:38.254
true,2023-10-27T14:39:40.259
true,2023-10-27T14:39:42.265
false,2023-10-27T14:39:44.270