Spring -每秒只允许一个发送到端点的post请求

基础概念

Spring框架中的“每秒只允许一个发送到端点的POST请求”通常指的是限流（Rate Limiting）的概念。限流是一种保护系统资源不被过度使用的策略，通过限制单位时间内某个接口的请求次数来防止系统过载。

类型

令牌桶算法：系统以恒定速率向桶中添加令牌，请求时必须从桶中获取令牌才能被处理。
漏桶算法：请求像水一样倒入漏桶，系统以恒定速率处理请求。
滑动窗口计数器：结合固定窗口计数器和滑动时间窗口，更精确地控制流量。

应用场景

API网关：对外提供的API接口需要进行流量控制。
高并发场景：如电商平台的秒杀活动、社交媒体的热门话题推送等。
安全防护：对抗DDoS攻击、暴力破解等恶意行为。

实现方法（Spring Boot示例）

以下是一个使用Spring Cloud Gateway和Redis实现限流的简单示例：

依赖引入

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

配置文件

spring:
  redis:
    host: localhost
    port: 6379
  cloud:
    gateway:
      routes:
        - id: rate_limit_route
          uri: http://example.org
          predicates:
            - Path=/api/**
          filters:
            - name: RequestRateLimiter
              args:
                key-resolver: "#{@userKeyResolver}"
                redis-rate-limiter.replenishRate: 1
                redis-rate-limiter.burstCapacity: 1

自定义Key Resolver

import org.springframework.cloud.gateway.filter.ratelimit.KeyResolver;
import org.springframework.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

@Component
public class UserKeyResolver implements KeyResolver {
    @Override
    public Mono<String> resolve(ServerWebExchange exchange) {
        return Mono.just(exchange.getRequest().getQueryParams().getFirst("userId"));
    }
}