错误使用
在实现限流时,网上的各种文章基本都会提到Guava的RateLimiter,用于实现单机的限流,并给出类似的代码:
public void method() {
RateLimiter rateLimiter = RateLimiter.create(10);
if(rateLimiter.tryAcquire()){
// do business
......
}
}
可是上面的代码真的能限流吗?
首先,从代码逻辑角度来讲,方法在每次被调用是都new一个RateLimiter,不同请求之间毫无关联,怎么能起到限流的作用呢?
其次,经过本人实际验证,上面的方法运行结果表明,根本没有限流的作用。
正确使用
在SpringMVC项目中,controller、service等对应的bean都是单例,因此将RateLimiter作为bean的属性并初始化,再加上RateLimiter的注释中表示RateLimiter是并发安全的:
RateLimiter is safe for concurrent use: It will restrict the total rate of calls from all threads. Note, however, that it does not guarantee fairness.
因此,正确的写法如下:
private RateLimiter rateLimiter = RateLimiter.create(10);
public void method() {
if(rateLimiter.tryAcquire()){
// do business
......
}
}