springboot 接口限流

在 Spring Boot 中，我们可以使用 AOP（面向切面编程）和拦截器的方式来实现接口限流。以下是一个简单的实现方式：

1. 引入 Guava 库，它包含了令牌桶算法和漏斗算法的实现。

2. 定义一个注解 `@RateLimit`，用来标记需要进行限流的接口。

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface RateLimit {
    int value() default 100; // 默认每秒限制100个请求
}

3. 定义一个切面 `RateLimitAspect`，在接口被调用时进行限流检查。

@Aspect
@Component
public class RateLimitAspect {
    private final RateLimiter rateLimiter = RateLimiter.create(100.0); // 每秒100个请求

    @Around("@annotation(rateLimit)")
    public Object limit(ProceedingJoinPoint joinPoint, RateLimit rateLimit) throws Throwable {
        if (rateLimiter.tryAcquire(rateLimit.value(), TimeUnit.MILLISECONDS)) {
            return joinPoint.proceed();
        } else {
            throw new RuntimeException("接口限流，请稍后再试！");
        }
    }
}

4. 在接口方法上添加 `@RateLimit` 注解，指定每秒允许的请求个数。

@RestController
public class DemoController {
    @GetMapping("/demo")
    @RateLimit(10) // 每秒限制10个请求
    public String demo() {
        return "Hello World!";
    }
}

以上代码实现了一个简单的接口限流功能，可以根据实际需求进行调整。需要注意的是，这种方式只适用于单机部署的场景，如果是分布式部署的系统，需要考虑使用分布式限流方案。