原理:使用令牌桶。
固定时间内产生一定数量的令牌,比如设置1秒产生50个令牌,但是1秒内出现了100个用户并发访问,此时只有50个用户能拿到令牌,剩余50直接阻挡,被限流。
核心代码,zuu编写PRE过滤器。
@Component
public class LimitFilter extends ZuulFilter {
private static final RateLimiter RATE_LIMITER = RateLimiter.create(1);
@Override
public String filterType() {
return FilterConstants.PRE_TYPE;
}
@Override
public int filterOrder() {
return 0;
}
@Override
public boolean shouldFilter() {
return true;
}
@Override
public Object run() throws ZuulException {
RequestContext currentContext = RequestContext.getCurrentContext();
if(RATE_LIMITER.tryAcquire()){
System.out.println("通过");
}else{
System.out.println("限流了");
// currentContext.setSendZuulResponse(false);
//currentContext.setResponseStatusCode(HttpStatus.TOO_MANY_REQUESTS.value());
}
return null;
}
}
使用谷歌guava包在固定时间内产生固定数据的令牌。比如:
RateLimiter RATE_LIMITER = RateLimiter.create(2);
就是在1秒内产生2个令牌。
使用Jmeter性能测试:
先创建测试计划,选择Threds(uder)->Thread Group
然后在Thread Group 中增加Sampler->HTTP Request
设置每秒用户并发数量。
点击上面的绿色Start按钮即可执行并发请求。