「 从0到1学习微服务SpringCloud 」12 Zuul的综合使用

@Component
public class TokenFilter extends ZuulFilter {

    /**
     * 过滤器类型
     * @return
     */
    @Override
    public String filterType() {
        return PRE_TYPE;
    }

    /**
     *  优先级，数字越大，优先级越低
     *  只做一般过滤,没有特殊要求的,可以放在内置过滤器后面
     * @return
     */
    @Override
    public int filterOrder() {
        //最后一个内置PRE过滤器后面
        return PRE_DECORATION_FILTER_ORDER+1;
    }

    /**
     * 是否执行该过滤器，true代表需要过滤
     * 这里可以忽略掉一些不需要过滤请求
     * @return
     */
    @Override
    public boolean shouldFilter() {
        return true;
    }

    @Override
    public Object run() throws ZuulException {
        RequestContext ctx = RequestContext.getCurrentContext();
        HttpServletRequest request = ctx.getRequest();
        String token = request.getParameter("token");
        //如果没有token,则返回401
        //这里就不做校验了,只判断有无
        if(StringUtils.isEmpty(token)){
            ctx.setSendZuulResponse(false);
            ctx.setResponseStatusCode(HttpStatus.SC_UNAUTHORIZED);
            ctx.setResponseBody("未登陆!");
             //设置返回体的编码为UTF-8
            HttpServletResponse response = ctx.getResponse();
            response.setHeader("Content-type", "text/html;charset=UTF-8");
            response.setCharacterEncoding("UTF-8");
        }
        return null;
    }
}

@Component
public class AddResponseHeaderFilter extends ZuulFilter {
    @Override
    public String filterType() {
        return POST_TYPE;
    }

    @Override
    public int filterOrder() {
        //放在返回过滤器前
        return SEND_RESPONSE_FILTER_ORDER - 1;
    }

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

    @Override
    public Object run() throws ZuulException {
        RequestContext context = RequestContext.getCurrentContext();
        HttpServletResponse  response = context.getResponse();
        response.setHeader("my-response","hello");
        return null;
    }
}

@Component
public class RateFilter extends ZuulFilter {
    /**
     * 创建令牌桶 容量为1(为了方便测试,这里设置小一点)
     */
    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()){
            //TODO 跳转到错误页面或友好提示
            throw new RuntimeException("访问人数过多,请稍后再试!");
        }
        return null;
    }
}

@Configuration
public class CorsConfig {
    @Bean
    public CorsFilter corsFilter(){
        final UrlBasedCorsConfigurationSource source = new UrlBasedCorsConfigurationSource();
        final CorsConfiguration config = new CorsConfiguration();
        //是否支持Cookie跨域
        config.setAllowCredentials(true);
        //支持的原始域 *表示所有
        config.setAllowedOrigins(Arrays.asList("*"));
        //允许的头 *表示所有
        config.setAllowedHeaders(Arrays.asList("*"));
        //允许的请求方法 POST,GET等 *表示所有
        config.setAllowedMethods(Arrays.asList("*"));
        //缓存时间 表示300秒内对相同请求不需要再判断
        config.setMaxAge(300L);

        //path参数为对哪些域名进行设置 /**表示所有
        source.registerCorsConfiguration("/**",config);
        return  new CorsFilter(source);
    }
}