高并发解决方案限流技术-----计数器

Posted 2022-05-18 pickknow

1.它是限流算法中最简单最容易的一种算法

计数器实现限流 每分钟只允许10个请求 第一个请求进去的时间为startTime,在startTime + 60s内只允许10个请求

当60s内超过十个请求后，拒绝，不超过，到第60s 重新设置时间

package com.aiyuesheng.utils;

import java.util.concurrent.atomic.AtomicInteger;
import lombok.Getter;
import lombok.Setter;

/**
 * @author chris
 * @see 计数器实现限流 每分钟只允许10个请求 第一个请求进去的时间为startTime,在startTime + 60s内只允许10个请求
 *      当60s内超过十个请求后，拒绝，不超过，到第60s 重新设置时间
 */
@Setter
@Getter
public class LimitService 
    // 限流的个数
    private int maxCount = 10;
    // 指定的时间内
    private long interval = 60;
    // 原子类计数器
    private AtomicInteger atomicInteger = new AtomicInteger(0);
    // 起始时间
    private long startTime = System.currentTimeMillis();

    public boolean limit(int maxCount, int interval) 
        atomicInteger.addAndGet(1);
        if (atomicInteger.get() == 1) 
            startTime = System.currentTimeMillis();
            atomicInteger.addAndGet(1);
            return true;
        
        // 超过了间隔时间，直接重新开始计数
        if (System.currentTimeMillis() - startTime > interval * 1000) 
            startTime = System.currentTimeMillis();
            atomicInteger.set(1);
            return true;
        
        // 还在间隔时间内,check有没有超过限流的个数
        if (atomicInteger.get() > maxCount) 
            return false;
        
        return true;
    

 

它是限流算法中最简单最容易的一种算法，比如我们要求某一个接口，1分钟内的请求不能超过10次，我们可以在开始时设置一个计数器，每次请求，该计数器+1；如果该计数器的值大于10并且与第一次请求的时间间隔在1分钟内，那么说明请求过多，如果该请求与第一次请求的时间间隔大于1分钟，并且该计数器的值还在限流范围内，那么重置该计数器