【Java】快速配置spring cloud gateway-redis令牌桶访问限制功能

导入依赖

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

在启动类中配置KeyResolver

可以另外定义一个配置类存放

@Bean
    public KeyResolver myKeyResolver() {
        return new KeyResolver() {
            @Override
            public Mono<String> resolve(ServerWebExchange exchange) {
            	//通过获取它的主机名进行访问限制（还有很多种方式）
                return Mono.just(exchange.getRequest().getRemoteAddress().getHostName());
            }
        };
    }

在配置文件添加参数

application.yml
全局访问限制

default-filters:
  - name: RequestRateLimiter 
    args:
      redis-rate-limiter.replenishRate: 2 #令牌桶每秒填充平均速率
      redis-rate-limiter.burstCapacity: 2 #令牌桶总容量

如果想对单个路由进行限制，则在其后添加过滤器并配置即可