Spring Cloud Gateway 限流操作

梦里梦外; 2023-10-03 11:24 114阅读 0赞

### 前言 ###

开发高并发系统时有三把利器用来保护系统：缓存、降级和限流，API网关作为所有请求的入口，请求量大，我们可以通过对并发访问的请求进行限速来保护系统的可用性。

常用的限流算法比如有令牌桶算法，漏桶算法，计数器算法等。

目前限流提供了基于Redis的实现,我们需要增加对应的依赖：

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

可以通过KeyResolver来指定限流的Key,比如我们需要根据用户来做限流，IP来做限流等等。

### IP限流 ###

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

通过exchange对象可以获取到请求信息，这边用了HostName，如果你想根据用户来做限流的话这边可以获取当前请求的用户ID或者用户名就可以了，比如：

### 用户限流 ###

使用这种方式限流，请求路径中必须携带userId参数

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

### 接口限流 ###

获取请求地址的uri作为限流key

@Bean
    KeyResolver apiKeyResolver() {
        
    	return exchange -> Mono.just(exchange.getRequest().getPath().value());
    }

然后配置限流的过滤器信息：

server:
      port: 8084
    spring:
      redis:
        host: 127.0.0.1
        port: 6379
      cloud:
        gateway:
          routes:
          - id: fsh-house
            uri: lb://fsh-house
            predicates:
            - Path=/house/**
            filters:
            - name: RequestRateLimiter
              args:
                redis-rate-limiter.replenishRate: 10
                redis-rate-limiter.burstCapacity: 20
                key-resolver: "#{@ipKeyResolver}"

*  filter名称必须是RequestRateLimiter
 *  redis-rate-limiter.replenishRate：允许用户每秒处理多少个请求
 *  redis-rate-limiter.burstCapacity：令牌桶的容量，允许在一秒钟内完成的最大请求数
 *  key-resolver：使用SpEL按名称引用bean

可以访问接口进行测试，这时候Redis中会有对应的数据：

127.0.0.1:6379> keys *
    1) "request_rate_limiter.{
        localhost}.timestamp"
    2) "request_rate_limiter.{
        localhost}.tokens"

大括号中就是我们的限流Key,这边是IP，本地的就是localhost

*  timestamp:存储的是当前时间的秒数，也就是System.currentTimeMillis() / 1000或 Instant.now().getEpochSecond()
 *  tokens:存储的是当前这秒钟的对应的可用的令牌数量