Spring Cloud Zuul自定义过滤器的实现

<dependency>
  <groupId>org.springframework.cloud</groupId>
  <artifactId>spring-cloud-starter-netflix-zuul</artifactId>
</dependency>

import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.exception.ZuulException;
import org.springframework.stereotype.Component;


@Component
public class MyFilter extends ZuulFilter {

  
  @Override
  public String filterType() {
    return "pre";
  }

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

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

  
  @Override
  public Object run() throws ZuulException {

    System.out.println("per zuul filter...");

    return null;
  }
}

import com.imooc.grace.result.GraceJSONResult;
import com.imooc.grace.result.ResponseStatusEnum;
import com.imooc.utils.IPUtil;
import com.imooc.utils.jsonUtils;
import com.imooc.utils.RedisOperator;
import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;
import com.netflix.zuul.exception.ZuulException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.Http.MediaType;
import org.springframework.stereotype.Component;

import javax.servlet.http.httpservletRequest;


@Component
public class BlackIpFilter extends ZuulFilter {

  
  private static final int CONTINUE_COUNTS = 10;
  
  private static final int TIME_INTERVAL = 10;
  
  private static final int LIMIT_TIMES = 15;

  @Autowired
  private RedisOperator redisOperator;

  @Override
  public String filterType() {
    return "pre";
  }


  @Override
  public int filterOrder() {
    // 这里设置为2，上面那个过滤器设置为1，则执行顺序为 1->2，大家可以测试一下
    return 2;
  }

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

  @Override
  public Object run() throws ZuulException {
    // 获取上下文对象
    RequestContext currentContext = RequestContext.getCurrentContext();
    HttpServletRequest request = currentContext.getRequest();
    // 获取ip
    String requestIp = IPUtil.getRequestIp(request);

    // 判断该ip在10秒内请求次数是否超过10次，超过则限制该ip15秒内不能访问，15秒后再放行
    final String ipRedisKey = "zuul-ip:" + requestIp;
    final String ipRedisLimiTKEy = "zuul-ip-limit:" + requestIp;

    // 获取当前ip这个key的剩余时间
    long limitLeftTime = redisOperator.ttl(ipRedisLimitKey);
    // 判断该ip是否还有剩余时间
    if (limitLeftTime > 0) {
      stopRequest(currentContext);
      return null;
    }

    // 在redis中累加ip的请求次数
    long requestCounts = redisOperator.increment(ipRedisKey, 1);
    if (requestCounts == 1) {
      redisOperator.expire(ipRedisKey, TIME_INTERVAL);
    }

    if (requestCounts > CONTINUE_COUNTS) {
      // 限制ip访问
      redisOperator.set(ipRedisLimitKey, ipRedisLimitKey, LIMIT_TIMES);
      stopRequest(currentContext);
    }

    return null;
  }

  private void stopRequest(RequestContext context) {
    // 停止zuul继续向下路由，禁止请求通信
    context.setSendZuulResponse(false);
    // 返回响应码200
    context.setResponseStatusCode(200);

    // TODO 要返回提示的json内容（可以自定义任何响应内容）
    // 例如 {"status":544,"msg":"请求过于频繁，请稍后再试","success":false,"data":null}
    String result = "json内容";

    // 设置返回内容
    context.setResponseBody(result);
    // 设置编码
    context.getResponse().setCharacterEncoding("utf-8");
    // 设置返回内容格式为json
    context.getResponse().setContentType(MediaType.APPLICATION_JSON_VALUE);
  }
}

import javax.servlet.http.HttpServletRequest;


public class IPUtil {

  
  public static String getRequestIp(HttpServletRequest request) {
    String ip = request.getHeader("x-forwarded-for");
    if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
      ip = request.getHeader("Proxy-Client-IP");
    }
    if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
      ip = request.getHeader("WL-Proxy-Client-IP");
    }
    if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
      ip = request.getHeader("HTTP_CLIENT_IP");
    }
    if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
      ip = request.getHeader("HTTP_X_FORWARDED_FOR");
    }
    if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
      ip = request.getRemoteAddr();
    }
    return ip;
  }
}