SpringBoot接口限流的实现方法小结

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

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);
    }
}

@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("请求成功！");
    }
}

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

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

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

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