[Feature] Flow control for Netflix Zuul 1.x about sentinel HOT 6 CLOSED

Tracer.trace(e3);

@Override
public int filterOrder() {
    return 1;
}

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

@Override
public Object run() {
    RequestContext ctx = RequestContext.getCurrentContext();
   // 统计API访问时间
    ctx.set("startTime", System.currentTimeMillis());
  // IP限流
    if (!sentinelIp(ctx)) {
        return null;
    }
  // 访问频率限流
    if (!sentinelApi(ctx)) {
        return null;
    }
    return null;
}

/**
 * @Author: ---
 * @Date: 2018/9/10 17:32
 * @Description: 限流IP
 **/
private boolean sentinelIp(RequestContext ctx) {
    HttpServletRequest request = ctx.getRequest();
    // 获取用户IP地址，将用户IP地址当做限流key
    String ip = IpTool.getIp(request);
    Entry entry = null;
    try {
        entry = SphU.entry(ip);
    } catch (BlockException e1) {
        log.info("API -> 限流ip：" + ip);
        // 过滤该请求，不往下级服务去转发请求，到此结束
        ctx.setSendZuulResponse(false);
        ctx.setResponseStatusCode(429);
        ctx.setResponseBody("{\"result\":\"IP访问次数超限!\"}");
        ctx.getResponse().setContentType("application/json;charset=UTF-8");
        return false;
    } finally {
        if (entry != null) {
            entry.exit();
        }
    }
    return true;
}

/**
 * @Author: ---
 * @Date: 2018/9/10 17:32
 * @Description: 限流
 **/
private boolean sentinelApi(RequestContext ctx) {
    HttpServletRequest request = ctx.getRequest();
    //获取用户请求地址，将用户请求地址当做限流key
    String url = request.getServletPath();
    Entry entry = null;
    try {
        entry = SphU.entry(url);
    } catch (BlockException e1) {
        log.info("API -> api限流地址：" + url);
        ctx.setSendZuulResponse(false);
        ctx.setResponseStatusCode(429);
        ctx.setResponseBody("{\"result\":\"API访问次数超限!\"}");
        ctx.getResponse().setContentType("application/json;charset=UTF-8");
        return false;
    } finally {
        if (entry != null) {
            entry.exit();
        }
    }
    return true;
}

@Resource(name = "redisTemplate")
private RedisTemplate redisTemplate;

/**
 * @Author: --------
 * @Date: 2018/9/11 13:15
 * @Description: 初始化限流规则
 **/
public void init() {
    // 从redis读取限流规则
    if (!RedisTool.exists(redisTemplate, ZuulLimitEnum.ZUUL_LIMIT.getKey())) {
        log.info("API -> 限流规则为空,直接返回");
        return;
    }
    // 把限流规则配置在redis
    List<String> list = RedisTool.hget(redisTemplate, ZuulLimitEnum.ZUUL_LIMIT.getKey());
    List<FlowRule> rules = new ArrayList<>();
    list.forEach(s -> {
        JSONObject jsonObject = JSONObject.parseObject(s);
        // 初始化限流规则
        FlowRule rule = new FlowRule();
        rule.setResource(jsonObject.getString("resource"));
        rule.setGrade(RuleConstant.FLOW_GRADE_QPS);
        // 每秒钟多少个QPS
        rule.setCount(jsonObject.getInteger("count"));
        rules.add(rule);
    });
    log.info("API -> 初始化限流规则：{}", rules);
    //加载限流规则
    FlowRuleManager.loadRules(rules);
}

@Autowired
private ZuulLimitInit zuulLimitInit;

public static void main(String[] args) {
    SpringApplication.run(CommonGatewayApplication.class, args);
}

@Override
public void run(String... strings) throws Exception {
    zuulLimitInit.init();
}