Spring Cloud Gateway 提供了内置的限流功能，可以通过使用 RequestRateLimiter 过滤器来实现。下面是使用 RequestRateLimiter 的一些步骤和示例，说明如何在 Spring Cloud Gateway 中实现限流。
1. 添加依赖
首先，确保在你的 Spring Cloud Gateway 项目中添加了以下依赖：
org.springframework.cloud
spring-cloud-starter-gateway
org.springframework.boot
spring-boot-starter-data-redis-reactive
2. 配置 Redis
你需要配置 Redis，因为 RequestRateLimiter 使用 Redis 来存储和共享限流信息。
spring:
redis:
host: localhost
port: 6379
3. 配置 Gateway
在 application.yml 或 application.properties 中配置你的路由和过滤器：
spring:
cloud:
gateway:
routes:
- id: my-route
uri: lb://SERVICE-NAME
predicates:
- Path=/api/**
filters:
- name: RequestRateLimiter
args:
redis-rate-limiter.replenishRate: 10 # 每秒允许处理的请求数
redis-rate-limiter.burstCapacity: 20 # 允许的最大并发请求数
redis-rate-limiter.key-resolver: "#{@myKeyResolver}" # 用于解析请求的 KeyResolver
4. 实现 KeyResolver
你需要定义一个 KeyResolver，它决定了如何根据请求来识别限流的键。以下是一个简单的例子，它使用请求的 IP 地址作为限流键：
import org.springframework.cloud.gateway.filter.ratelimit.KeyResolver;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;
public class IpKeyResolver implements KeyResolver {
@Override
public Mono resolve(ServerWebExchange exchange) {
return Mono.just(exchange.getRequest().getRemoteAddress().getAddress().getHostAddress());
}
}
然后在 Spring 配置中注册这个 KeyResolver：
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
@Configuration
public class GatewayConfig {
@Bean
public KeyResolver myKeyResolver() {
return new IpKeyResolver();
}
}
5. 启动 Spring Cloud Gateway
启动你的 Spring Cloud Gateway 应用程序，现在它将根据配置的限流规则来限制请求。
这样，你的 Spring Cloud Gateway 应用程序就可以通过 RequestRateLimiter 过滤器实现限流了。每个请求都会根据配置的速率和容量进行评估，如果超过了配置的速率，请求将被拒绝。
注意：上述配置中的 replenishRate 和 burstCapacity 参数需要根据你的应用场景来调整。replenishRate 是每秒允许处理的请求数，而 burstCapacity 是允许的突发请求的最大数量。