漏桶限流算法实现

package com.xxxx.seckill.limit;

/**
 * @author WangHeng
 * @version V1.0
 * @date 2023/1/9- 16:48
 * @desc 漏桶限流算法实现-(先通过速率预估出当前的水量，进而判断取舍)
 **/
public class LeakBucketLimit {
    //第一次初始化系统的当前的时间
    private Long timeStamp = System.currentTimeMillis();
    //设置桶的容量大小为100
    private Long capacity = 100L;
    //水漏出的速度（每秒系统能处理的请求数目）
    private Long rate = 10L;
    //当前的水量（当前累计的请求数目）
    private Long water = 20L;

    /**
     * 方法限流方法
     *
     * @return true:代表限流 false:代表通过
     */
    public synchronized Boolean limit() {
        //获取当前的时间
        Long now = System.currentTimeMillis();
        //先执行漏水，计算剩余的水量 如果时间太多的话会为负数，设置最小值0
        water = Math.max(0, water - (now - timeStamp) / 1000 * rate);
        //更新这次访问的时间
        timeStamp = now;
        //如果加上这个请求后仍小于100的话
        if ((water + 1) <= capacity) {
            //当前水量++
            water++;
            return false;
        }
        //水满了
        else {
            return true;
        }
    }
}