SpringCloud Alibaba(三) - GateWay网关

<!--  Gatway 网关会和springMvc冲突，不能添加web依赖      -->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter</artifactId>
</dependency>

<!--   gateway 依赖     -->
<dependency>
    <groupId>org.springframework.cloud</groupId>
    <artifactId>spring-cloud-starter-gateway</artifactId>
</dependency>

# 端口
server:
  port: 9606

# 服务名
spring:
  application:
    name: kgcmall-gatway

  cloud:
    #nacos 配置
    nacos:
      discovery:
        server-addr: 127.0.0.1:8848

    # 网关配置
    gateway:
      routes: # 路由，是list集合，可以配置多个路由
      	#product模块
        - id: kh96_route_first # 当前route路由的唯一标识，不能重复
          #uri: http://localhost:9602 # 路由转发的目标资源地址，不支持多负载调用，不利于扩展，不推荐
          uri: lb://kgcmall96-prod # lb 从nacos注册中心的服务列表中，根据指定的服务名，调用服务，推荐用法
          predicates: # 指定路由断言配置，支持多个断言，只要断言成功（满足路由转发条件），才会执行转发到目标资源地址访问
            - Path=/prod-gateway/** # 指定path路径断言，必须满足请求地址是/prod-gateway开始，才会执行路由转发
          filters: # 指定路由过滤配置，支持多个过滤器，在断言成功，执行路由转发时，对请求和响应数据进行过滤处理
            - StripPrefix=1 # 在请求断言成功后，执行路由转发时，自动去除第一层的访问路径/prod-gateway
        #user模块
        - id: kh96_route_second
          uri: lb://kgcmall96-user
          predicates:
            - Path=/user-gateway/**
          filters:
            - StripPrefix=1

/**
 * Created On : 28/11/2022.
 * <p>
 * Author : huayu
 * <p>
 * Description: 自定义网关断言工厂-权限断言
 */
@Component //自定义断言工厂，必须是一个组件放入容器才可以生效
public class MyAuthRoutePredicateFactory
        extends AbstractRoutePredicateFactory<MyAuthRoutePredicateFactory.Config> {

    /*
        配置项名MyAuth的配置参数值，映射到断言工厂内部类的属性名
     */
    public static final String MYAUTH_KEY = "myAuth";

    /*
        通过空参构造方法，指定静态内部类，用于接收配置文件中的配置项的内容，即断言(- myAuth=xxx)
     */
    public MyAuthRoutePredicateFactory() {
        super(MyAuthRoutePredicateFactory.Config.class);
    }

    /*
        价格核心配置文件中的自定义配置项的内容，映射到当前配置类的属性中，即Collections.singletonList(DATETIME_KEY)；指定的内部属性
     */
    public List<String> shortcutFieldOrder() {
        return Collections.singletonList(MYAUTH_KEY);
    }

    @Override
    public Predicate<ServerWebExchange> apply(MyAuthRoutePredicateFactory.Config config) {
        return new GatewayPredicate() {
            @Override
            public boolean test(ServerWebExchange serverWebExchange) {
                //自定义权限断言业务逻辑
                if (StringUtils.isNotBlank(config.getMyAuth())) {
                    //判断自定义权限配置参数值，是否和当前指定的值一直，如果一致则断言成功，否则失败
                    return config.getMyAuth().equals("KH96");
                }
                //断言失败
                return false;

            }
        };
    }

    @Data
    public static class Config {
        private String myAuth;
    }

}

predicates:
	- MyAuth=KH96  # 自定义 权限断言 配置 (注意首字母要大写)

predicates:
	- MyAuth=KHxx  # 配置错误信息

filters: 
	- AddRequestHeader=X-Request-token,token_kh96  # 添加请求头参数，两个参数，第一个是添加请求头参数名，第二个参数值

@GetMapping("/mallProduct")
public KgcMallProduct mallProduct(@RequestParam Integer pid,
                                  @RequestHeader(value = "X-Request-token", required = false) String gateWayFilterToken) {
    log.info("------ 根据商品编号：{}， 查询商品详情 ------", pid);

    //通过GateWay网关滤器，增加请求头参数
    log.info("------ 通过GateWay网关滤器，增加请求头参数 X-Request-token：{} ------", gateWayFilterToken);
    // 调用业务接口，查询商品详情
    return kgcMallProductService.getMallProductById(pid);
}

filters: 
	- AddRequestParameter=queryName,param_kh96 # 添加普通 请求参数 ,两个参数，第一个是添加请求头参数名，第二个参数值

@GetMapping("/mallProduct")
public KgcMallProduct mallProduct(@RequestParam Integer pid
                                  @RequestParam(value = "queryName", required = false) String gateWayFilterParamQueryName) {
    log.info("------ 根据商品编号：{}， 查询商品详情 ------", pid);
    
    //通过GateWay网关滤器，增加请求参数
    log.info("------ 通过GateWay网关滤器，增加请求头参数 queryName：{} ------", gateWayFilterParamQueryName);

    // 调用业务接口，查询商品详情
    return kgcMallProductService.getMallProductById(pid);
}

/**
 * Created On : 28/11/2022.
 * <p>
 * Author : huayu
 * <p>
 * Description: 自定义网关过滤工厂 -校验过滤器
 */
@Component //自定义网关过滤工厂，必须是一个组件放入容器才可以生效
public class MyCheckGatewayFilterFactory extends AbstractGatewayFilterFactory<MyCheckGatewayFilterFactory.Config> {
    /*
        配置项名MyAuth的配置参数值，映射到断言工厂内部类的属性名
     */
    public static final String MYCHECK_KEY = "myCheck";

    public MyCheckGatewayFilterFactory() {
        super(MyCheckGatewayFilterFactory.Config.class);
    }

    public List<String> shortcutFieldOrder() {
        return Arrays.asList(MYCHECK_KEY);
    }

    public GatewayFilter apply(MyCheckGatewayFilterFactory.Config config) {
        return new GatewayFilter() {
            public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
                //自定义网关过滤器实现过滤业务逻辑
                //模拟发送到网关的请求，如果请求携带参数和网关过滤器配置参数一致，放行，如果不一致，直接404
                String checkParam = exchange.getRequest().getQueryParams().getFirst("checkParam");

                //如果携带了checkParam参数  判断获取的请求参数是否和网关过滤器中配置的参数一致
                if (StringUtils.isNotBlank(checkParam)) {
                    //判断是否一致
                    if (checkParam.equals(config.getMyCheck())) {
                        //过滤器放行到目标请求
                        return chain.filter(exchange);
                    }

                    //返回404
                    exchange.getResponse().setStatusCode(HttpStatus.NOT_FOUND);

                    return exchange.getResponse().setComplete();

                }
                //没有携带参数，直接放行
                return chain.filter(exchange);
            }
        };
    }

    @Data
    public static class Config {
        private String myCheck;
    }

}

/**
 * Created On : 28/11/2022.
 * <p>
 * Author : huayu
 * <p>
 * Description: 自定义网关全局过滤，实现token鉴权，实现GlobalFilter 和 Ordered接口
 */
@Component  //必须是spring组件
public class MyTokenGlobalFilter implements GlobalFilter, Ordered {
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        //自定义全局过滤逻辑，判断请求头中们是否携带了token参数，如果带了就放行，非欧洲就就拒绝
        if (StringUtils.isBlank(exchange.getRequest().getHeaders().getFirst("token"))) {
            //没有携带token,直接重定向到登录页
            //模拟重定向到单点登录网址，临时用百度
            String redirectUrl = "https://www.baidu.com";

            ServerHttpResponse response = exchange.getResponse();
            response.setStatusCode(HttpStatus.SEE_OTHER); //303 错误，重定向
            response.getHeaders().set(HttpHeaders.LOCATION, redirectUrl);

            //结束响应
            return response.setComplete();
        }
        //代了token,
        //TODO token校验
        return chain.filter(exchange);
    }

    @Override
    public int getOrder() {
        //指定全局过滤器的优先级，值越小，优先级越高
        return 0;
    }
}