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实现接口限流的主要内容,如果未能解决你的问题,请参考以下文章