springboot使用guava令牌桶限流

最新推荐文章于 2024-12-02 16:00:00 发布

原创最新推荐文章于 2024-12-02 16:00:00 发布 · 189 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#spring boot #guava #java

controlle层：

    @GetMapping("/limit")
    public String limit() {
        return "hello";
    }

切面层：

@Aspect
@RestController
@Slf4j
public class LimitAspect {
    private static final double DEFAULT_COUNT = 0.5;
    private Cache<String, RateLimiter> cache = CacheBuilder.newBuilder().expireAfterAccess(10, TimeUnit.SECONDS).build();

    @Around(value = "execution(* com.boot.mpdemo.controller..*.*(..))")
    public Object around(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {
        Signature signature = proceedingJoinPoint.getSignature();
        MethodSignature methodSignature = (MethodSignature) signature;
        String name = methodSignature.getMethod().getName();
        RateLimiter rateLimiter = cache.get(name, () -> RateLimiter.create(DEFAULT_COUNT));
        if (!rateLimiter.tryAcquire()) {
            log.warn("请求过多");
            throw new RuntimeException();
        }
        return proceedingJoinPoint.proceed();

    }
}