Gateway之自定义路由断言工厂

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

第一步：在配置文件中，添加一个Age的断言配置

server:
  port: 7000
spring:
  application:
    name: api-gateway
  cloud:
    nacos:
      discovery:
        server-addr: localhost:8848   # nacos服务端地址
    gateway:
      discovery:
        locator:
          enabled: true # 启用探测器，让gateway可以发现nacos中的微服务
      routes: # 路由数组（路由：就是当指定请求满足什么条件的时候，转发到哪个微服务）
        - id: product_route  # 当前路由的标识，要求唯一。默认uuid
          uri: lb://service-product # lb指的是负载均衡（load balancing）,service-product是nacos中微服务的名称
          order: 1  # 路由的优先级，数字越小级别越高
          predicates: # 断言（就是路由转发要满足的条件）
            - Path=/product-serv/** # 当请求路径满足Path指定的规则时，才进行路由转发
            - Age=18,60 # 限制年龄[18, 60)的人能访问
          filters:  # 过滤器，请求在传递过程中可以通过过滤器对其进行一定的修改
            - StripPrefix=1 # 转发之前去掉1层路径

第二步：自定义一个断言工厂，实现断言方法

package cn.jack.predicates;

import lombok.Data;
import org.springframework.cloud.gateway.handler.predicate.AbstractRoutePredicateFactory;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import org.springframework.web.server.ServerWebExchange;

import java.util.Arrays;
import java.util.List;
import java.util.function.Predicate;

@Component
public class AgeRoutePredicateFactory extends AbstractRoutePredicateFactory<AgeRoutePredicateFactory.Config> {

    public AgeRoutePredicateFactory() {
        super(AgeRoutePredicateFactory.Config.class);

    }

    // 将配置文件中的值按返回集合的顺序，赋值给配置类
    @Override
    public List<String> shortcutFieldOrder() {
        return Arrays.asList(new String[] {"minAge", "maxAge"});
    }

    // 断言
    @Override
    public Predicate<ServerWebExchange> apply(AgeRoutePredicateFactory.Config config) {

        return (ServerWebExchange serverWebExchange)->{
            // TODO 获取请求参数age，判断是否满足[18, 60)
            String age = serverWebExchange.getRequest().getQueryParams().getFirst("age");
            if (StringUtils.isEmpty(age)) {
                return false;
            }

            if (!age.matches("[0-9]+")) {
                return false;
            }

            int iAge = Integer.parseInt(age);

            if (iAge >= config.minAge && iAge < config.maxAge) {
                return true;
            } else {
                return false;
            }
        };
    }

    // 配置类，属性用于接收配置文件中的值
    @Data
    public static class Config {
        private int minAge;
        private int maxAge;
    }
}

第三步：启动测试