spring cloud zuul实现限流

上一篇文章《spring cloud zuul实践》介绍了zuul做网关的demo，再扩展一下过滤器就可以实现限流。限流有很多做法，这里介绍令牌实现限流

一、 令牌

通过第三方组件按照固定速率投放令牌到令牌桶里，当令牌桶装满，则不再接收令牌。当请求过来时，先申请拿令牌，拿到令牌则pass请求，并且消耗掉一个令牌，拿不到令牌则拒绝请求，概要流程如下图所示

 

二、在网关zuul中过滤请求时，加入令牌逻辑，从而实现限流

import com.google.common.util.concurrent.RateLimiter;
import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;
import com.netflix.zuul.exception.ZuulException;
import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Component;

import java.util.concurrent.TimeUnit;

import static org.springframework.cloud.netflix.zuul.filters.support.FilterConstants.PRE_TYPE;
import static org.springframework.cloud.netflix.zuul.filters.support.FilterConstants.SERVLET_DETECTION_FILTER_ORDER;

/**
 * @Classname RateLimiter
 * @Created by Michael
 * @Date 2023/7/13
 * @Description 限流过滤
 */
@Component
public class MyRateLimiter extends ZuulFilter {

  //  令牌桶
  static RateLimiter rateLimiter 
          = RateLimiter.create(8,2, TimeUnit.SECONDS);

  @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 {
    RequestContext context = RequestContext.getCurrentContext();
    boolean allowed = rateLimiter.tryAcquire();
    if(!allowed){
      context.setSendZuulResponse(false);
      context.setResponseStatusCode(501);
    }
    return null;
  }
}

以上代码中RateLimiter.create(8,2, TimeUnit.SECONDS)

第一个参数，是总令牌数

第二个参数，简单理解为多久放令牌

第三个参数，是时间单位，即多久投放依次令牌

三、测试

每200ms发起一次请求，连续请求15次，可以看到请求结果结果如下：