在Zuul中设置服务访问限流

最新推荐文章于 2024-05-03 02:05:58 发布

weixin_33915554

最新推荐文章于 2024-05-03 02:05:58 发布

阅读量164

点赞数

原文链接：http://www.cnblogs.com/a-small-lyf/p/10857148.html

版权

　　使用Zuul中的过滤器结合谷歌提供的工具类RateLimiter达到限流

在网关服务中创建一个过滤器,并添加到容器中

@Component
public class RateLimitFilter extends ZuulFilter {

    private static final RateLimiter RATE_LIMITER = RateLimiter.create(1);//设置每秒访问的次数，这里是每秒只能被访问一次


    @Override
    public String filterType() {
        //设置为前置过滤器
        return PRE_TYPE;
    }

    @Override
    public int filterOrder() {
          //设置优先级,这里设置的优先级是最高级
        return SERVLET_DETECTION_FILTER_ORDER-1;
    }

    @Override
    public boolean shouldFilter() {
        return true;
    }

    @Override
    public Object run() throws ZuulException {
        if(!RATE_LIMITER.tryAcquire()){
            //这里我们可以直接抛出异常，提示友好信息给用户
            System.out.println("限流");
        }
        return null;
    }
}