Spring Boot实现接口限流

API限流是一种重要的策略，用于控制对API的访问速率，以保护后端服务免受过载和滥用。以下是API限流的必要性：

防止服务过载：

当API的请求量突然激增时，如果没有限流措施，可能会导致服务器资源耗尽，从而影响服务的稳定性和可用性。
提高系统稳定性：

通过限制每个用户的请求频率，可以确保系统在高负载下仍能正常运行，避免因单个用户或服务的过度请求而导致的系统崩溃。
防止恶意攻击：

限流可以作为一种安全措施，防止恶意用户通过发起大量请求来攻击系统，如DDoS攻击或暴力破解尝试。

在 Spring Boot 中，可以通过多种方式实现接口限流。

以下是几种常用的实现方法：

1. 使用 Bucket4j

Bucket4j 是一个 Java 的限流库，可以很容易地集成到 Spring Boot 项目中。

步骤：

添加 Maven 依赖：

XML 复制代码

<!-- https://mvnrepository.com/artifact/com.bucket4j/bucket4j-core -->
<dependency>
    <groupId>com.bucket4j</groupId>
    <artifactId>bucket4j-core</artifactId>
    <version>8.10.1</version>
</dependency>

创建限流配置：

java 复制代码

import net.jodah.bucket4j.Bucket;
import net.jodah.bucket4j.BucketBuilder;
import java.time.Duration;

@Service
public class RateLimiterService {
    private final Bucket bucket;

    public RateLimiterService() {
        this.bucket = Bucket.builder()
                .addLimit(BucketLimit.of(10, Duration.ofMinutes(1)))
                .build();
    }

    public boolean tryConsume() {
        return bucket.tryConsume(1);
    }
}

在控制器中使用限流：

java 复制代码

@RestController
public class MyController {
    private final RateLimiterService rateLimiterService;

    @Autowired
    public MyController(RateLimiterService rateLimiterService) {
        this.rateLimiterService = rateLimiterService;
    }

    @GetMapping("/api")
    public ResponseEntity<String> api() {
        if (!rateLimiterService.tryConsume()) {
            return ResponseEntity.status(HttpStatus.TOO_MANY_REQUESTS).body("请求过于频繁，请稍后再试。");
        }
        return ResponseEntity.ok("请求成功！");
    }
}

2. 使用 Spring Cloud Gateway

如果你使用 Spring Cloud Gateway，可以在配置文件中设置限流规则。

示例配置：

bash 复制代码

spring:
  cloud:
    gateway:
      routes:
        - id: my_route
          uri: lb://my-service
          predicates:
            - Path=/api/**
          filters:
            - requestRateLimiter:
                rateLimiter:
                  refillPolicy:
                    tokens: 10
                    duration: 1s
                  burstCapacity: 20

3. 使用 AOP 方式

通过 AOP（面向切面编程）也可以实现限流。

步骤：

创建注解：

java 复制代码

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface RateLimit {
    int limit() default 10; // 限制次数
    int timeout() default 60; // 超时时间
}

使用 AOP 切面：

java 复制代码

@Aspect
@Component
public class RateLimitAspect {
    // 实现限流逻辑
    @Around("@annotation(rateLimit)")
    public Object limit(ProceedingJoinPoint joinPoint, RateLimit rateLimit) throws Throwable {
        // 限流逻辑
        return joinPoint.proceed();
    }
}

在需要限流的控制器方法上使用：

java 复制代码

@RestController
public class MyController {
    @RateLimit(limit = 10, timeout = 60)
    @GetMapping("/api")
    public ResponseEntity<String> api() {
        return ResponseEntity.ok("请求成功！");
    }
}

4. 集成第三方库Resilience4j：

Resilience4j是一个轻量级的容错库，它提供了多种限流器实现，如SemaphoreBasedRateLimiter。
添加Resilience4j依赖后，可以配置限流器，并在控制器中使用注解@RateLimiter进行限流。

5.使用分布式锁实现限流：

在某些情况下，可以使用分布式锁（如Redisson）来实现限流，尤其是在需要防止用户重复操作的场景中。

结论

以上是常用的几种限流实现方式，可以根据项目需求选择适合的方法。

选择哪种限流方案取决于具体的业务需求和系统架构。对于分布式系统，通常推荐使用Redis或第三方库如Resilience4j来实现限流，以保证限流的准确性和一致性。而对于单机应用，Guava RateLimiter或Spring Boot Actuator的@RateLimiter注解可能是更简单的选择。