Spring Cloud学习：如何实现Gateway 服务网关限流

在Spring Cloud中，可以使用Spring Cloud Gateway内置的限流功能。以下是一个简单的例子，展示如何在Spring Cloud Gateway中实现限流。

1. 添加依赖：

   确保在pom.xml中添加了Spring Cloud Gateway和Spring Cloud Circuit Breaker依赖。

<dependencies>
    <dependency>
        <groupId>org.springframework.cloud</groupId>
        <artifactId>spring-cloud-starter-gateway</artifactId>
    </dependency>
    <dependency>
        <groupId>org.springframework.cloud</groupId>
        <artifactId>spring-cloud-starter-circuitbreaker-reactor-resilience4j</artifactId>
    </dependency>
</dependencies>

2. 配置限流规则：

   在application.yml中配置限流规则。以下示例配置了每秒允许通过1个请求。

spring:
  cloud:
    gateway:
      routes:
        - id: limited_route
          uri: http://localhost:8081
          filters:
            - name: RequestRateLimiter
              args:
                key-resolver: '#{@userKeyResolver}'
                redis-rate-limiter:
                  slices: 1
                  limit: 1
                  replenishRate: 1
                  burstCapacity: 1
 
    circuitbreaker:
      config:
        default:
          slidingWindowSize: 10
          permittedNumberOfCallsInSlidingWindow: 1

3. 定义KeyResolver：

   创建一个KeyResolver实例，用于确定限流的键。

@Component
public class UserKeyResolver implements KeyResolver {
    @Override
    public Mono<String> resolve(ServerWebExchange exchange) {
        return Mono.just(exchange.getRequest().getQueryParams().getFirst("user"));
    }
}

在上述配置中，RequestRateLimiter过滤器使用了@userKeyResolver Bean来决定使用哪个键来限流。这里假设请求中带有一个名为user的查询参数用作限流键。

以上配置了每秒只允许一个带有user查询参数的请求通过。如果没有user参数或者参数值相同，这些请求将会被限流。

请注意，实际使用时需要配置Redis连接以存储限流状态。以上代码片段仅展示了核心配置，并假设已经有一个Redis服务可用。