一、前言

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

聊了以下问题:

本文接着聊如何自定义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”;

【云原生&amp;微服务>SCG网关篇五】Spring Cloud Gateway自定义网关路由Predicate-LMLPHP

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

【云原生&amp;微服务>SCG网关篇五】Spring Cloud Gateway自定义网关路由Predicate-LMLPHP

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

四、总结

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

07-15 07:46