【云原生&微服务＞SCG网关篇五】Spring Cloud Gateway自定义网关路由Predicate

文章目录

• 一、前言
• 二、自定义PredicateFactory
• 三、验证
• 四、总结

一、前言

至此微服务网关系列文章已出：

聊了以下问题：

本文接着聊如何自定义PredicateFactory，实现网关早九晚六之间才提供路由服务。

<properties>
    <spring-boot.version>2.4.2</spring-boot.version>
    <spring-cloud.version>2020.0.1</spring-cloud.version>
    <spring-cloud-alibaba.version>2021.1</spring-cloud-alibaba.version>
</properties>

<dependencyManagement>
    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-dependencies</artifactId>
            <version>${spring-boot.version}</version>
            <type>pom</type>
            <scope>import</scope>
        </dependency>
        <!--整合spring cloud-->
        <dependency>
            <groupId>org.springframework.cloud</groupId>
            <artifactId>spring-cloud-dependencies</artifactId>
            <version>${spring-cloud.version}</version>
            <type>pom</type>
            <scope>import</scope>
        </dependency>
        <!--整合spring cloud alibaba-->
        <dependency>
            <groupId>com.alibaba.cloud</groupId>
            <artifactId>spring-cloud-alibaba-dependencies</artifactId>
            <version>${spring-cloud-alibaba.version}</version>
            <type>pom</type>
            <scope>import</scope>
        </dependency>
    </dependencies>
</dependencyManagement>

二、自定义PredicateFactory

路由谓词工厂的命令要求一定要以RoutePredicateFactory结尾。

这里谓词工厂要实现的效果是：限制每天的09:00 – 18:00才能访问路由；想要实现这一效果，需要一个实现自AbstractRoutePredicateFactory的工厂类、一个接收时间参数的配置类，并且需要在application.yml文件中配置这个谓词工厂。

1、自定义参数配置类（TimeBetweenConfig）

package com.saint.gateway.predicate;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.time.LocalTime;

/**
 * time parameter config
 *
 * @author Saint
 */
@Data
@NoArgsConstructor
@AllArgsConstructor
public class TimeBetweenConfig {

    private LocalTime start;

    private LocalTime end;
}

2、自定义谓词工厂（TimeBetweenRoutePredicateFactory）

自定义谓词工厂需要继承自AbstractRoutePredicateFactory<C>，并需要将泛型替换为自己需要接收Yaml文件中参数的配置类；

package com.saint.gateway.predicate;

import org.springframework.cloud.gateway.handler.predicate.AbstractRoutePredicateFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;

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

/**
 * Customized time between route predicate factory
 *
 * @author Saint
 */
@Component
public class TimeBetweenRoutePredicateFactory extends AbstractRoutePredicateFactory<TimeBetweenConfig> {

    public TimeBetweenRoutePredicateFactory() {
        super(TimeBetweenConfig.class);
    }

    /**
     * 控制路由的条件
     *
     * @param config
     * @return
     */
    @Override
    public Predicate<ServerWebExchange> apply(TimeBetweenConfig config) {
        LocalTime start = config.getStart();
        LocalTime end = config.getEnd();
        return serverWebExchange -> {
            // 当前时间再start 和 end之间，则允许匹配路由
            LocalTime now = LocalTime.now();
            return now.isAfter(start) && now.isBefore(end);
        };
    }

    /**
     * 用来配置控制类和yaml配置文件的映射关系
     *
     * @return
     */
    @Override
    public List<String> shortcutFieldOrder() {
        return Arrays.asList("start", "end");
    }
}

3、修改application.yml配置文件

在application.yml文件中添加TimeBetween 谓词；

server:
  port: 9999
spring:
  cloud:
    gateway:
      routes:
        - id: time_between_route
          uri: http://127.0.0.1:9001
          predicates:
            # 当且仅当请求时的时间在Between配置的时间范围内时，才会匹配到路由进行转发
            - TimeBetween=上午9:00, 下午6:00

这里表情请求仅在早上9点到下午6点之间提供服务，过时不给路由做匹配，全部休息。

三、验证

当前时间：早上7点；随便访问一个地址时，会报错“404 not found”；

假如我们把TimeBetween的时间放开成：上午7:00, 下午6:00，便可以进行路由匹配。

做个验证就行，赶紧改回去，要求是9点之前不提供服务；

四、总结

至此，Spring Cloud Gateway的 Predicate也就告一段落了，后面我们继续看Filter。