【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/**

            # 自定義Age斷言工廠
            # Age：自動找到 "Age" + "RoutePredicateFactory" 斷言工廠
            # 18,60: 斷言工廠參數
            # 限制年齡[18, 60)的人能訪問
            - Age=18,60


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

第二步：自定義一個斷言工廠，實現斷言方法

package com.test.springcloud.predicate;

import org.springframework.cloud.gateway.handler.predicate.AbstractRoutePredicateFactory;
import org.springframework.stereotype.Component;
import org.springframework.util.MultiValueMap;
import org.springframework.util.StringUtils;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.server.ServerWebExchange;

import java.util.Arrays;
import java.util.List;
import java.util.function.Consumer;
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(Consumer<Config> consumer) {
        return super.apply(consumer);
    }

    @Override
    public Predicate<ServerWebExchange> apply(Config config) {
        // 創建網關斷言對象
        return new Predicate<ServerWebExchange>() {
            // 檢查
            @Override
            public boolean test(ServerWebExchange serverWebExchange) {
                // TODO 獲取請求參數age，判斷是否滿足[18, 60)
                MultiValueMap<String, String> queryParams = serverWebExchange.getRequest().getQueryParams();
                String age = queryParams.getFirst("age");
                if (!StringUtils.isEmpty(age) && age.matches("[0-9]+")) {
                    int iAge = Integer.parseInt(age);
                    if (iAge >= config.minAge && iAge < config.maxAge) {
                        return true;
                    }
                }
                return false;
            }
        };
    }


    // 配置類，屬性用於接收配置文件中的值
    @Validated
    public static class Config {
        private int minAge;
        private int maxAge;

        public int getMinAge() {
            return minAge;
        }

        public void setMinAge(int minAge) {
            this.minAge = minAge;
        }

        public int getMaxAge() {
            return maxAge;
        }

        public void setMaxAge(int maxAge) {
            this.maxAge = maxAge;
        }
    }
}

第三步：啟動測試

　　

 

參考：https://www.freesion.com/article/36441306839/