自定义路由断言工厂

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

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

spring:
application:
 name: api-gateway
cloud:
 nacos:
  discovery:
   server-addr: 127.0.0.1:8848
 gateway:
  discovery:
   locator:
    enabled: true
  routes:
   - id: product-route
    uri: lb://service-product
    predicates:
     - Path=/product-serv/**
     - Age=18,60 # 限制年龄只有在18到60岁之间的人能访问
    filters:
     - StripPrefix=1

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

package com.itheima.predicates;
//泛型 用于接收一个配置类,配置类用于接收中配置文件中的配置
@Component
public class AgeRoutePredicateFactory
    extends AbstractRoutePredicateFactory<AgeRoutePredicateFactory.Config> {
  public AgeRoutePredicateFactory() {
    super(AgeRoutePredicateFactory.Config.class);
 }
  //用于从配置文件中获取参数值赋值到配置类中的属性上
  @Override
  public List<String> shortcutFieldOrder() {
    //这里的顺序要跟配置文件中的参数顺序一致
    return Arrays.asList("minAge", "maxAge");
 }
  //断言
  @Override
  public Predicate<ServerWebExchange> apply(AgeRoutePredicateFactory.Config
config) {
    return new Predicate<ServerWebExchange>() {
      @Override
      public boolean test(ServerWebExchange serverWebExchange) {
        //从serverWebExchange获取传入的参数
         String ageStr =
        
serverWebExchange.getRequest().getQueryParams().getFirst("age");
        if (StringUtils.isNotEmpty(ageStr)) {
          int age = Integer.parseInt(ageStr);
          return age > config.getMinAge() && age < config.getMaxAge();
       }
        return true;
     }
   };
 }
}
//自定义一个配置类, 用于接收配置文件中的参数
@Data
class Config {
  private int minAge;
  private int maxAge;
}

第4步：启动测试

#测试发现当age在(20,60)可以访问,其它范围不能访问
http://localhost:7000/product-serv/product/1?age=30
http://localhost:7000/product-serv/product/1?age=10