网关限流，防止冲垮网关

前言

有没有遇到过某一刻时间内的流量很大，导致服务出现线程资源耗尽的情况，导致后续的用户再查询相关接口直接超时，体验非常不好，所以可以在网关中加限流，直接在网关拒绝掉，不需要在转发到下层服务。

直接上代码

package com.junyu.loancloud.filter.zrs;

import cn.hutool.json.JSONUtil;
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 com.zcckj.common.mvc.result.Result;
import org.springframework.cloud.netflix.zuul.filters.support.FilterConstants;
import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Component;

/**
 * @author lwh
 * @date 2022/7/6
 **/
@Component
public class LimitFilter extends ZuulFilter {


    private static final RateLimiter LIMIT_PUTI = RateLimiter.create(100);

    private static final RateLimiter LIMIT_PANGU = RateLimiter.create(100);

    @Override
    public String filterType() {
        return FilterConstants.PRE_TYPE;
    }

    @Override
    public int filterOrder() {
        return -100;
    }

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

    @Override
    public Object run() throws ZuulException {
        RequestContext ctx = RequestContext.getCurrentContext();
        String requestURI = ctx.getRequest().getRequestURI();
        if (requestURI.startsWith("/api/puti") ) {
            if (!LIMIT_PUTI.tryAcquire()) {
                setResp(ctx);
            }
        } else if (requestURI.startsWith("/api/pangu")) {
            if (!LIMIT_PANGU.tryAcquire()) {
                setResp(ctx);
            }
        }
        return null;
    }

    private void setResp(RequestContext ctx) {
        ctx.setSendZuulResponse(false);
        ctx.setResponseBody(JSONUtil.toJsonStr(Result.error("人数过多，请稍后再试")));
        ctx.setResponseStatusCode(HttpStatus.OK.value());
        ctx.getResponse().setContentType("application/json; charset=utf-8");
    }
}

解释

上面的代码的意思就是创建2个令牌桶，只要请求是/api/puti开头或者/api/pangu/开头的就会命中RateLimiter的令牌桶，其他的请求直接放过，因为我们的直播和下单都是在这2个前缀请求中，所以只对这2个高并发的限制，RateLimiter.create(100) 代表每秒释放100个令牌，没有令牌直接拒绝掉，不会再转发到下层服务，从而在网关解决掉资源的问题。

当然也可以使用Sentinel，我们下层服务使用Sentinel做了流控。