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