Spring Cloud Gateway 限流

一：Spring Cloud Gateway 自帶限流實作

  1. 添加依賴

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

  2.指定KeyResolver的限流key

其中 exchange.getRequest().getRemoteAddress().getHostName(), 可以根據自己的需求去切換限流的key，我這裡是根據hostName進行切換的。
           

@Bean
    KeyResolver ipResolver() {
        return exchange -> Mono.just(exchange.getRequest().getRemoteAddress().getHostName());
    }
           

3. 配置檔案修改

 3.1根據官方文檔設定的配置檔案格式

spring:
  cloud:
    gateway:
      routes:
      - id: requestratelimiter_route
        uri: http://example.org
        filters:
        - name: RequestRateLimiter
          args:
            redis-rate-limiter.replenishRate: 10
            redis-rate-limiter.burstCapacity: 20
            key-resolver: "#{@ipResolver}"
           

上面是官網的配置，但是我設定之後并不起作用，并且不會傳回任何資訊。

檢視源碼後修改配置

改造後的配置為

spring:
  cloud:
    gateway:
      routes:
      - id: requestratelimiter_route
        uri: http://example.org
        filters:
        - name: RequestRateLimiter
          args:
            replenishRate: 10
            burstCapacity: 20
            key-resolver: "#{@ipResolver}"
           

4.運作測試

jemeter 測試之後，檢視redis存儲資訊得：

超出限流請求時 Response code: 429