package com.swapping.springcloud.ms.gateway.filter;

import com.alibaba.fastjson.JSON;

import com.netflix.zuul.ZuulFilter;

import com.netflix.zuul.context.RequestContext;

import com.netflix.zuul.exception.ZuulException;

import com.swapping.springcloud.ms.core.response.UniVerResponse;

import org.apache.commons.lang3.StringUtils;

import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;

/**

 * 路由拦截

 *

 *

 * >>>>>zuul的filter过滤器的生命周期有一下四个：

 *

 *  PRE： 这种过滤器在请求被路由之前调用。我们可利用这种过滤器实现身份验证、在集群中选择请求的微服务、记录调试信息等。

 *  ROUTING：这种过滤器将请求路由到微服务。这种过滤器用于构建发送给微服务的请求，并使用Apache HttpClient或Netfilx Ribbon请求微服务。

 *  POST：这种过滤器在路由到微服务以后执行。这种过滤器可用来为响应添加标准的HTTP Header、收集统计信息和指标、将响应从微服务发送给客户端等。

 *  ERROR：在其他阶段发生错误时执行该过滤器。 除了默认的过滤器类型，Zuul还允许我们创建自定义的过滤器类型。例如，我们可以定制一种STATIC类型的过滤器，直接在Zuul中生成响应，而不将请求转发到后端的微服务。

 *

 *

 *  Zuul中默认实现了很多Filter，也可以自己自定义过滤器

 *

 *  下面是自己自定义过滤器

 *  实际使用中我们可以结合shiro、oauth2.0等技术去做鉴权、验证

 *

 */

@Component

public class AuthFilter extends ZuulFilter{

    @Override

    public String filterType() {

        return "pre";//可以在请求被路由之前调用

    }

    @Override

    public int filterOrder() {

        return 0;//filter执行顺序，通过数字指定 ,优先级为0，数字越大，优先级越低

    }

    @Override

    public boolean shouldFilter() {

        return true;// 是否执行该过滤器，此处为true，说明需要过滤

    }

    /**

     *  filter需要执行的具体操作

     *

     * 例如：本filter实际执行的逻辑 是验证所有的访问请求中，是否包含安全信息auth

     * @return

     * @throws ZuulException

     */

    @Override

    public Object run() throws ZuulException {

        RequestContext ctx = RequestContext.getCurrentContext();

        HttpServletRequest request = ctx.getRequest();

        String auth = request.getParameter("auth");

        //TODO 此处可以做日志记录

        System.out.println("zuul拦截--请求前验证---auth："+auth);

        //成功的情况

        if (StringUtils.isNotBlank(auth)){

            ctx.setSendZuulResponse(true); //对请求进行路由

            ctx.setResponseStatusCode(200);

            ctx.set("isSuccess", true);

        }else {

            //失败的情况

            UniVerResponse res = new UniVerResponse();

            res.beFalse3("zuul拦截--请求前验证---没有auth登录验证",UniVerResponse.ERROR_BUSINESS);

            ctx.setSendZuulResponse(false); //不对请求进行路由

            ctx.setResponseStatusCode(res.getCode());//设置返回状态码

            ctx.setResponseBody(JSON.toJSONString(res));//设置返回响应体

            ctx.set("isSuccess", false);

            ctx.getResponse().setContentType("application/json;charset=UTF-8");//设置返回响应体格式，可能会乱码

        }

        return null;

    }

}

@RestController

@RequestMapping("/member")

public class MemberController {

    @Autowired

    MemberService memberService;

    @RequestMapping(value = "/showMember")

    public String showMember(){

        return "会员服务正常,会员服务是服务消费者,也就是服务调用者,会调用商品服务进行商品购买,减少库存,增加销量;\n\r同时调用积分服务,增加积分";

    }

package com.swapping.springcloud.ms.gateway.fallback;

import com.alibaba.fastjson.JSON;

import com.swapping.springcloud.ms.core.response.UniVerResponse;

import org.springframework.cloud.netflix.zuul.filters.route.FallbackProvider;

import org.springframework.http.HttpHeaders;

import org.springframework.http.HttpStatus;

import org.springframework.http.MediaType;

import org.springframework.http.client.ClientHttpResponse;

import org.springframework.stereotype.Component;

import java.io.ByteArrayInputStream;

import java.io.IOException;

import java.io.InputStream;

/**

 * 路由熔断

 *

 * 类似与服务之间的feign调用熔断器设定，网关路由各个服务的请求也可以做路由熔断，

 * 在不能成功路由到具体服务上的请求时，进行降级处理，定制返回内容【也就是设置统一响应体】

 *

 * 目前路由熔断仅能 支持服务级别的熔断，不支持具体到某个URL进行熔断

 *

 */

@Component

public class UniVerFallback implements FallbackProvider {

    /**

     * 指定 可以熔断拦截 哪些服务

     * @return

     */

    @Override

    public String getRoute() {

//        return "springcloud-ms-member";//指定 可熔断某个服务 服务名是配置文件中配置的各服务的serviceId

        return "*"; //指定    可熔断所有服务

    }

    /**

     * 指定  熔断后返回的定制化内容

     * @param route

     * @param cause

     * @return

     */

    @Override

    public ClientHttpResponse fallbackResponse(String route, Throwable cause) {

        return new ClientHttpResponse() {

            @Override

            public HttpStatus getStatusCode() throws IOException {

                return HttpStatus.OK;

            }

            @Override

            public int getRawStatusCode() throws IOException {

                return 200;

            }

            @Override

            public String getStatusText() throws IOException {

                return "OK";

            }

            @Override

            public void close() {

            }

            /**

             * 设置响应体

             * @return

             * @throws IOException

             */

            @Override

            public InputStream getBody() throws IOException {

                //TODO 此处可以做日志记录

                UniVerResponse uniVerResponse = new UniVerResponse();

                uniVerResponse.beFalse3(route+"服务凉凉了",UniVerResponse.ERROR_BUSINESS);

                return new ByteArrayInputStream(JSON.toJSONString(uniVerResponse).getBytes());

            }

            /**

             * 设置响应头信息

             * @return

             */

            @Override

            public HttpHeaders getHeaders() {

                HttpHeaders headers = new HttpHeaders();

                headers.setContentType(MediaType.APPLICATION_JSON_UTF8);//指定响应体 格式+编码方式

                return headers;

            }

        };

    }

}