基于Guava和AOP实现SpringBoot注解接口限流方案

本文介绍了如何在SpringBoot项目中使用Guava的RateLimiter和AOP实现接口限流,包括创建自定义注解、配置限流规则,并通过RateLimitAspect进行拦截和控制。还提到可考虑使用更复杂的限流策略如分布式限流。
摘要由CSDN通过智能技术生成

在Spring Boot中实现接口限流,可以通过创建一个自定义注解来指定限流规则,然后用AOP(面向切面编程)来拦截注解标识的方法。以下是详细的例子,演示如何使用Guava的RateLimiter来实现一个简单的限流方案一步一步实现的过程:

  1. 引入相关依赖
<dependency>
    <groupId>com.google.guava</groupId>
    <artifactId>guava</artifactId>
    <version>30.1-jre</version>
</dependency>
<dependency>
    <groupId>org.aspectj</groupId>
    <artifactId>aspectjweaver</artifactId>
    <version>1.9.7</version>
</dependency>
  1. 创建自定义限流注解(RateLimit
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Target(ElementType.METHOD) // 方法注解
@Retention(RetentionPolicy.RUNTIME) // 运行时保留
public @interface RateLimit {
    double permitsPerSecond() default 1.0;  // 每秒生成的令牌数
    long timeout() default 0;               // 获取令牌等待时间,默认0,不等待
}
  1. 使用AOP创建切面包含限流逻辑
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.*;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;
import com.google.common.util.concurrent.RateLimiter;
import java.util.concurrent.ConcurrentHashMap;

@Aspect
@Component
public class RateLimitAspect {

    private ConcurrentHashMap<String, RateLimiter> limiters = new ConcurrentHashMap<>();

    @Pointcut("@annotation(com.yourpackage.RateLimit)")
    public void rateLimit() {
    }

    @Around("rateLimit()")
    public Object around(ProceedingJoinPoint joinPoint) throws Throwable {
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        RateLimit rateLimit = signature.getMethod().getAnnotation(RateLimit.class);
        
        if (rateLimit != null) {
            double permitsPerSecond = rateLimit.permitsPerSecond();
            long timeout = rateLimit.timeout();
            // 构建或获取RateLimiter
            RateLimiter rateLimiter = limiters.computeIfAbsent(signature.toLongString(), (key) -> RateLimiter.create(permitsPerSecond));
            
            // 获取令牌
            boolean acquired = timeout == 0L ? rateLimiter.tryAcquire() : rateLimiter.tryAcquire(timeout, TimeUnit.MILLISECONDS);
            if (!acquired) {
                throw new RuntimeException("System is too busy, please try again later.");
            }
        }
        
        return joinPoint.proceed();
    }
}

如需更复杂的限流策略,可考虑算法如令牌桶(Guava的RateLimiter已采用)、漏桶、固定窗口、滑动窗口或者是使用 Redis + Lua脚本来做分布式限流。

  1. 在对应的方法上使用你的自定义注解
@RestController
public class RateLimitController {

    @RequestMapping("/ratelimitTest")
    @RateLimit(permitsPerSecond = 1)
    public String rateLimitTest() {
        return "Request was allowed.";
    }
}
  1. 启动应用

配置好以上内容后,启动Spring Boot应用,通过实际请求来测试限流效果。当请求的速率超过注解中设定的permitsPerSecond值时,就会抛出异常或返回限流提示。
在这里插入图片描述
6. 总结
请注意这仅是一个简单的示例。负载不均和配置更新可能会导致此限流策略在更复杂场景下效果有限。在生产级别应用中,您可能需要实现更复杂的策略,如集成 Redis 做分布式限流。

  • 12
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
Spring Boot 中,我们可以使用 AOP(面向切面编程)和拦截器的方式来实现接口限流。以下是一个简单的实现方式: 1. 引入 Guava 库,它包含了令牌桶算法和漏斗算法的实现。 2. 定义一个注解 `@RateLimit`,用来标记需要进行限流接口。 ```java @Target(ElementType.METHOD) @Retention(RetentionPolicy.RUNTIME) public @interface RateLimit { int value() default 100; // 默认每秒限制100个请求 } ``` 3. 定义一个切面 `RateLimitAspect`,在接口被调用时进行限流检查。 ```java @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` 注解,指定每秒允许的请求个数。 ```java @RestController public class DemoController { @GetMapping("/demo") @RateLimit(10) // 每秒限制10个请求 public String demo() { return "Hello World!"; } } ``` 以上代码实现了一个简单的接口限流功能,可以根据实际需求进行调整。需要注意的是,这种方式只适用于单机部署的场景,如果是分布式部署的系统,需要考虑使用分布式限流方案

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值