java guava限流,Guava的RateLimiter实现接口限流

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

首先添加guava依赖:

com.google.guava

guava

23.0

然后封装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 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一下。

作者：堕落门徒