SpringCloud gateway 限流 RequestRateLimiter

谁借莪１个温暖的怀抱￠ 2022-11-21 09:55 167阅读 0赞

# 配置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。

使用的算法是 [令牌桶算法][Link 1] , 点击查看。

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"));
    }

[Link 1]: https://blog.csdn.net/xgw1010/article/details/107595141