SpringCloud gateway 限流 RequestRateLimiter

配置redis

添加依赖

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-redis-reactive</artifactId>
</dependency>

要添加上面的redis依赖，否则启动会报错：

Caused by: java.lang.IllegalArgumentException: Unable to find GatewayFilterFactory with name RequestRateLimiter
	at org.springframework.cloud.gateway.route.RouteDefinitionRouteLocator.loadGatewayFilters(RouteDefinitionRouteLocator.java:187)
	at org.springframework.cloud.gateway.route.RouteDefinitionRouteLocator.getFilters(RouteDefinitionRouteLocator.java:228)
	at org.springframework.cloud.gateway.route.RouteDefinitionRouteLocator.convertToRoute(RouteDefinitionRouteLocator.java:170)
	at reactor.core.publisher.FluxMap$MapSubscriber.onNext(FluxMap.java:100)
	at reactor.core.publisher.FluxFlatMap$FlatMapMain.tryEmitScalar(FluxFlatMap.java:480)
	at reactor.core.publisher.FluxFlatMap$FlatMapMain.onNext(FluxFlatMap.java:413)

配置redis地址

spring:
  redis:
    database: 0
    password: 123456
    sentinel:
      master: mymaster
      nodes: 192.168.1.20:26379,192.168.1.20:27379,192.168.1.20:28379
    lettuce:
      pool:
        max-active: 4
        max-idle: 4
        min-idle: 0

如果不配置redis数据库地址，会默认连接localhost。

RequestRateLimiter

Redis的实现基于 Stripe。它需要使用 spring-boot-starter-data-redis-reactive Spring Boot starter。

使用的算法是 令牌桶算法 , 点击查看。

redis-rate-limiter.replenishRate是你允许用户每秒执行多少请求，而不丢弃任何请求。这是令牌桶的填充速率。

redis-rate-limiter.burstCapacity是允许用户在一秒钟内执行的最大请求数。这是令牌桶可以保存的令牌数。将此值设置为零将阻止所有请求。

redis-rate-limiter.requestedTokens 是每个请求消耗多少个令牌，默认是1.

稳定速率是通过在replenishRate（补充令牌速度） 和 burstCapacity（令牌桶容量）中设置相同的值来实现的。可通过设置burstCapacity高于replenishRate来允许临时突发流浪。在这种情况下，限流器需要在两次突发之间留出一段时间（根据replenishRate），因为连续两次突发将导致请求丢失 (HTTP 429 - Too Many Requests).。

要限制每秒一个请求，可以将replenishRate设置为目标请求数，requestedTokens设置目标的时间秒数，burstCapacity为replenishRate * requestedTokens。如：设置replenishRate=1, requestedTokens=60 和 burstCapacity=60，就是限制每分钟1个请求。

spring:
  cloud:
    gateway:
      default-filters:
        - name: RequestRateLimiter
          args:
            redis-rate-limiter.replenishRate: 1
            redis-rate-limiter.burstCapacity: 1
            redis-rate-limiter.requestedTokens: 1
            key-resolver: "#{@userKeyResolver}" #这个必须要配置，否则返回403

这里配置的是全局限流，也可以单独为某个路由配置限流。

KeyResolver

@Bean
KeyResolver userKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getQueryParams().getFirst("user"));
}