直接上代码

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
package demo;

import java.time.LocalDateTime;
import java.util.LinkedList;
import java.util.Random;

/**
 * @author weylan
 */
public class SlideWindow {

    static LinkedList<long> timeWindow = new LinkedList<>();

    /**
     * 滑动时间窗口限流算法
     * 在指定时间窗口，指定限制次数内，是否允许通过
     *
     * @param windowSize 时间窗口大小,单位 秒
     * @param count      允许通过的次数
     * @return 是否允许通过
     */
    public static boolean process(long windowSize, int count) {
        long nowTime = System.currentTimeMillis();
        // 未达到限制次数，则允许通过，并添加当前时间戳到队列开始位置
        if (timeWindow.size() < count) {
            timeWindow.addFirst(nowTime);
            return Boolean.TRUE;
        }
        // 队列已满（达到限制次数），则获取队列中最早(队尾)添加的时间戳
        Long farTime = timeWindow.getLast();
        // 用当前时间戳 减去 最早添加的时间戳
        if (nowTime - farTime <= windowSize) {
            // 若结果小于等于windowSize，则说明在windowSize内，通过的次数大于count
            return false;
        }
        // 若结果大于windowSize，则说明在windowSize内，通过的次数小于等于count
        // 允许通过，并删除最早添加的时间戳，将当前时间添加到队首
        timeWindow.removeLast();
        timeWindow.addFirst(nowTime);
        return Boolean.TRUE;
    }

    public static void main(String[] args) throws InterruptedException {
        Random random = new Random();
        while (true) {
            // 10秒只允许2次通过
            System.out.println(LocalDateTime.now() + " : " + process(10000, 2));
            // 睡眠 0~10秒
            Thread.sleep(1000L * random.nextInt(10));
        }
    }


}

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

2021-10-19T16:30:56.654 : true
2021-10-19T16:30:56.655 : true
2021-10-19T16:31:03.659 : false
2021-10-19T16:31:09.663 : true
2021-10-19T16:31:12.663 : true
2021-10-19T16:31:20.667 : true
2021-10-19T16:31:21.671 : false
2021-10-19T16:31:21.671 : false
2021-10-19T16:31:26.675 : true
2021-10-19T16:31:27.679 : false
2021-10-19T16:31:29.680 : false
2021-10-19T16:31:38.682 : true
2021-10-19T16:31:47.686 : true
2021-10-19T16:31:56.690 : true
2021-10-19T16:31:58.695 : true
2021-10-19T16:32:04.696 : false
2021-10-19T16:32:11.698 : true
2021-10-19T16:32:11.698 : true
2021-10-19T16:32:11.698 : false
2021-10-19T16:32:15.702 : false
2021-10-19T16:32:21.705 : true
2021-10-19T16:32:23.709 : true
2021-10-19T16:32:32.708 : true
2021-10-19T16:32:40.711 : true
2021-10-19T16:32:45.715 : true
2021-10-19T16:32:50.719 : true
2021-10-19T16:32:57.720 : true
2021-10-19T16:32:59.722 : false
2021-10-19T16:33:03.723 : true
2021-10-19T16:33:11.727 : true
2021-10-19T16:33:11.727 : false
2021-10-19T16:33:16.731 : true
2021-10-19T16:33:16.732 : false
2021-10-19T16:33:17.735 : false
2021-10-19T16:33:22.738 : true
2021-10-19T16:33:28.741 : true
2021-10-19T16:33:29.741 : false
2021-10-19T16:33:30.744 : false