gateway(网关)基于Filter限流

前言:

SpringCloudGateway官方就提供了基于令牌桶的限流支持。基于其内置的过滤器工厂 RequestRateLimiterGatewayFilterFactory 实现。在过滤器工厂中是通过Redis和lua脚本结合的方式进行流量控制

简说:说白了,就是n秒内快速访问n次给你拦截住,类型网站的防cc攻击

使用

1.在gateway(网关)加入依赖

代码语言:javascript
复制
<!--监控依赖-->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-actuator</artifactId>
</dependency>
<!--redis的依赖-->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-redis-reactive</artifactId>
</dependency>

2.修改application.yml配置文件,添加redis,然后启动Redis 在spring下

代码语言:javascript
复制
  redis:
    port: 6379
    host: 127.0.0.1
image-1678094033530

3.修改application.yml配置文件,在spring-cloud-gateway-routes下 参数讲解,请看其它

代码语言:javascript
复制
        filters:
        - RewritePath=/api/(?<segment>.*), /$\{segment}
        - name: RequestRateLimiter  #限流
          args:
            key-resolver: '#{@pathKeyResolver}' # 使用SpEL从容器中获取对象
            redis-rate-limiter.replenishRate: 1 # 令牌桶每秒填充平均速率
            redis-rate-limiter.burstCapacity: 3 # 令牌桶的总容量
image-1678093523239

4.创建config包,建MyConfig类,代码如下

代码语言:javascript
复制
package com.zb.config;

import org.springframework.cloud.gateway.filter.ratelimit.KeyResolver;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

@Configuration
public class MyConfig {

/**
 * 路径限流
 *
 * @return
 */
@Bean
public KeyResolver pathKeyResolver() {
    return new KeyResolver() {
        @Override
        public Mono&lt;String&gt; resolve(ServerWebExchange exchange) {
            
            return Mono.just(exchange.getRequest().getPath().toString());
        }
    };
}


/**
 * 用户限流
 * @return
 */

/* @Bean
public KeyResolver userKeyResolver() {
return new KeyResolver() {
@Override
public Mono<String> resolve(ServerWebExchange exchange) {

            return Mono.just(exchange.getRequest().getQueryParams().getFirst(&#34;token&#34;));
        }
    };
}*/

}

其它

1.路径线路和用户限流只能有一个,使用一个时,另外一个必须注释掉
2.复制方法名,在application.yml粘贴方法名,即可启用/切换

key-resolver

redis-rate-limiter.replenishRate

redis-rate-limiter.burstCapacity

填写限流的方法名

就是几秒内

多少次会被拦截