Guava的RateLimiter实现接口限流

Posted cord

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Guava的RateLimiter实现接口限流相关的知识,希望对你有一定的参考价值。

最近开发需求中有需要对后台接口进行限流处理,整理了一下基本使用方法。

首先添加guava依赖:

    <dependency>
      <groupId>com.google.guava</groupId>
      <artifactId>guava</artifactId>
      <version>23.0</version>
    </dependency>

然后封装RateLimiter适用对多接口的限制:

import com.google.common.util.concurrent.RateLimiter;
import org.springframework.util.StringUtils;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.TimeUnit;

public class RateLimiterFactory {

    private static ConcurrentHashMap<String, RateLimiter> limiterMap = new ConcurrentHashMap<>();

    //每秒许可证数量
    private static double permitsPerSecond = 1000;

    //等待超时时间
    private static long timeout = 4L;

    /**
     *@param apiId 对应接口的id
     *@description 获取许可
     */
    public static boolean tryAcquire(String apiId) {
        //如果传入apiId为空则返回true
        if (StringUtils.isEmpty(apiId)) {
            return true;
        }
        limiterMap.putIfAbsent(apiId, RateLimiter.create(permitsPerSecond));
        return limiterMap.get(apiId).tryAcquire(timeout, TimeUnit.SECONDS);
    }

}

使用RateLimiterFactory实现对接口的限制:

@Controller
public class TestController {

    ...

    @RequestMapping("/test")
    @ResponseBody
    public String test(){
        if(!RateLimiterFactory.tryAcquire(ApiInfo.TEST)) {
            return "当前请求数过高!";
        }
        ...
    }
}

以上就是RateLimiter实现接口限流的常见使用,头一次发文,mark一下。

以上是关于Guava的RateLimiter实现接口限流的主要内容,如果未能解决你的问题,请参考以下文章

Guava的RateLimiter在单机限流中的正确用法

Guava RateLimiter限流器使用示例

高并发学习之使用RateLimiter实现令牌桶限流

Guava RateLimiter限流问题

Guava RateLimiter 实现 API 限流,这才是正确的姿势!

coding++:高并发解决方案限流技术-使用RateLimiter实现令牌桶限流-Demo