常用的限流办法是使用令牌桶进行限流
1)所有的请求在处理之前都需要拿到一个可用的令牌才会被处理;
2)根据限流大小,设置按照一定的速率往桶里添加令牌;
3)桶设置最大的放置令牌限制,当桶满时、新添加的令牌就被丢弃或者拒绝;
4)请求达到后首先要获取令牌桶中的令牌,拿着令牌才可以进行其他的业务逻辑,处理完业务逻辑之后,将令牌直接删除;
5)令牌桶有最低限额,当桶中的令牌达到最低限额的时候,请求处理完之后将不会删除令牌,以此保证足够的限流
这个算法的实现,有很多技术, Guava(读音: 瓜哇)是其中之一,redis客户端也有其实现。
代码实现
- 1导入坐标:spring cloud gateway 默认使用redis的RateLimter限流算法来实现。所以我们要
在这里插入代码片
使用首先需要引入redis的依赖
<!‐‐redis‐‐>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring‐boot‐starter‐data‐redis‐reactive</artifactId>
<version>2.1.3.RELEASE</version>
</dependency>
2.定义KeyResolver,在GatewayApplicatioin引导类中添加如下代码KeyResolve用于计算某一个类型的限流的KEY也就是说,可以通过KeyResolver来指定限流的Key。
//定义一个KeyResolver
@Bean
public KeyResolver ipKeyResolver() {
return new KeyResolver() {
@Override
public Mono<String> resolve(ServerWebExchange exchange) {
return
Mono.just(exchange.getRequest().getRemoteAddress().getHostName());
}
};
}
3.修改application.yml中配置项,指定限制流量的配置以及REDIS的配置,修改后最终配置如下:
spring:
application:
name: sysgateway
cloud:
gateway:
routes:
- id: goods
uri: lb://goods
predicates:
- Path=/goods/**
filters:
- StripPrefix= 1
- name: RequestRateLimiter #请求数限流 名字不能随便写
args:
key-resolver: "#{@ipKeyResolver}"
redis-rate-limiter.replenishRate: 1 #令牌桶每秒填充平均速率
redis-rate-limiter.burstCapacity: 1 #令牌桶总容量
解释:
burstCapacity :令牌桶总容量。
replenishRate :令牌桶每秒填充平均速率。
key-resolver :用于限流的键的解析器的 Bean 对象的名字。它使用 SpEL 表达式根
据#{@beanName}从 Spring 容器中获取 Bean 对象。
通过在 replenishRate 和中设置相同的值来实现稳定的速率 burstCapacity 。设
置 burstCapacity 高于时,可以允许临时突发 replenishRate 。在这种情况下,需要在
突发之间允许速率限制器一段时间(根据 replenishRate ),因为2次连续突发将导致请
求被丢弃( HTTP 429 ‐ Too Many Requests )
key-resolver: “#{@userKeyResolver}” 用于通过SPEL表达式来指定使用哪一个
KeyResolver.
如上配置:
表示 一秒内,允许 一个请求通过,令牌桶的填充速率也是一秒钟添加一个令牌。
最大突发状况 也只允许 一秒内有一次请求,可以根据业务来调整 。