【SpringCloud】Gateway自定义过滤器工厂

　　我们来设定一个场景：假设我们的应用仅仅让age在（min, max）之间的人来访问。

第一步：在配置文件中，添加一个Age的过滤工厂配置

# 端口
server:
  port: 9527

spring:
  application:
    name: cloud-gateway-gateway
  cloud:
    gateway:
      discovery:
        locator:
          # 开启从注册中心动态创建路由的功能，利用微服务名进行路由
          enabled: true
          # 忽略大小写匹配，默认为 false。
          # 当 eureka 自动大写 serviceId 时很有用。 所以 MYSERIVCE，会匹配 /myservice/**
          lowerCaseServiceId: true
      routes:
          # 路由id
        - id: payment_routh
          # 匹配后提供服务路由地址
          uri: lb://cloud-payment-service
          # 断言
          predicates:
            # curl http://localhost:9527/payment/get/1
            - Path=/payment/get/**


          filters:
            # 自定义Age过滤器工厂
            # Age：自动找到 "Age" + "GatewayFilterFactory" 过滤器工厂
            # 18,60: 断言工厂参数
            # 限制年龄[18, 60)的人能访问
            - name: Age
              args:
                name: minAge,maxAge
                value: 18,60

eureka:
  client:
    register-with-eureka: true
    fetch-registry: true
    service-url:
      defaultZone: http://localhost:8761/eureka

第二步：自定义一个过滤器工厂，实现过滤方法

　　自定义过滤器工厂 继承AbstractNameValueGatewayFilterFactory且我们的自定义名称必须要以GatewayFilterFactory结尾并交给spring管理。

package com.test.springcloud.filter;

import org.springframework.cloud.gateway.filter.GatewayFilter;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.factory.AbstractNameValueGatewayFilterFactory;
import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.stereotype.Component;
import org.springframework.util.MultiValueMap;
import org.springframework.util.StringUtils;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

import java.nio.charset.StandardCharsets;

// 继承AbstractNameValueGatewayFilterFactory且
// 我们的自定义名称必须要以GatewayFilterFactory结尾并交给spring管理。
// 自定义过滤器工厂
@Component
public class AgeGatewayFilterFactory extends AbstractNameValueGatewayFilterFactory {
    @Override
    public GatewayFilter apply(NameValueConfig config) {
        System.out.println("config.getName() = " + config.getName());
        System.out.println("config.getValue() = " + config.getValue());
//        int minAge = Integer.parseInt(config.getName());
//        int maxAge = Integer.parseInt(config.getValue());
        String[] arr = config.getValue().split(",");
        int minAge = Integer.parseInt(arr[0]);
        int maxAge = Integer.parseInt(arr[1]);
        return new GatewayFilter() {
            @Override
            public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
                // TODO 获取请求参数age，判断是否满足[18, 60)
                MultiValueMap<String, String> queryParams = exchange.getRequest().getQueryParams();
                String age = queryParams.getFirst("age");
                if (!StringUtils.isEmpty(age) && age.matches("[0-9]+")) {
                    int iAge = Integer.parseInt(age);
                    if (iAge >= minAge && iAge < maxAge) {
                         return  chain.filter(exchange);
                    }
                }

                ServerHttpResponse response = exchange.getResponse();
                byte[] bits = "请求参数 age 不满足[18, 60)".getBytes(StandardCharsets.UTF_8);
                DataBuffer buffer = response.bufferFactory().wrap(bits);
                //指定编码，否则在浏览器中会中文乱码
                response.getHeaders().add("Content-Type", "text/plain;charset=UTF-8");
                return response.writeWith(Mono.just(buffer));
            }
        };
    }
}

第三步：启动测试