SpringBoot单机限流

文章目录

如何实现单机限流

谷歌的 RateLimiter 介绍

使用谷歌的 guava 包中 RateLimiter 类来实现,先来介绍一个这个类:

  1. RateLimiter 是一个频率限制器,通过配置频率来发放许可,如果 1 秒内可以访问十次,那么这十次许可发送的间隔是完全相同的
  2. 并发使用是安全的
  3. RateLimiter 还可以去配置先处于一个预热器,每秒增加发放的许可知道达到稳定的频率
  4. RateLimiter 不影响请求本身的节流,而是影响下一次请求的节流,比如当前任务如果占用许可较多,到达 RateLimiter 之后,会立即占用,当下一个请求到达 RateLimiter 时就会经历节流,因为上一个请求已经占用大量的许可。

一个小示例用法,如果想要发送一组数据,我们限制他在 5kb 每秒:

java 复制代码
// 给每一个字节发放 1 个许可,限制在 5kb 每秒的话,只需要每秒发放 5000 个许可即可
final RateLimiter rateLimiter = RateLimiter.create(5000.0);
void submitPacket(byte[] packet) {
   rateLimiter.acquire(packet.length);
   networkService.send(packet);
}

使用 AOP 实现单机限流

实现步骤:

  1. 定义切面,拦截 Controller 层方法
  2. 创建一个 RateLimiter,定义访问频率
  3. 当执行方法时,发放一个许可,如果拿不到许可,直接拦截 Controller 层方法的执行,返回一个访问频繁的提示

Controller

java 复制代码
@RestController
@RequestMapping("/rate")
public class RateController {
  
  @GetMapping
  public String testRate() {
    return "测试 Rate ";
  }
}

切面

java 复制代码
@Component
@Aspect
public class ServiceLogAspect {
    /**
     * 对 controller 限流
     */
    @Pointcut("execution(* com.javagpt.back.controller.*.*(..))")
    public void rateLimitPointCut() {}

    private static final RateLimiter rateLimiter = RateLimiter.create(10);

    @SneakyThrows // 使用之后不需要抛出异常,lombok会自动在编译时加上try/catch
    @Around("rateLimitPointCut()")
    public Object rateLimit(ProceedingJoinPoint joinPoint) {
        double rate = rateLimiter.getRate();
        System.out.println(rate);
        if (rateLimiter.tryAcquire()) {
            return joinPoint.proceed();
        } else {
            // 如果超出限流次数,拦截方法的执行,注意这里返回的对象要和 Controller 方法的返回对象类型相同,否则会报错
            return "访问太过频繁";
        }
    }

}
相关推荐
飞的肖4 分钟前
前端使用 Element Plus架构vue3.0实现图片拖拉拽,后等比压缩,上传到Spring Boot后端
前端·spring boot·架构
Q_19284999066 分钟前
基于Spring Boot的摄影器材租赁回收系统
java·spring boot·后端
Code_流苏9 分钟前
VSCode搭建Java开发环境 2024保姆级安装教程(Java环境搭建+VSCode安装+运行测试+背景图设置)
java·ide·vscode·搭建·java开发环境
良许Linux11 分钟前
0.96寸OLED显示屏详解
linux·服务器·后端·互联网
求知若饥23 分钟前
NestJS 项目实战-权限管理系统开发(六)
后端·node.js·nestjs
禁默1 小时前
深入浅出:AWT的基本组件及其应用
java·开发语言·界面编程
Cachel wood1 小时前
python round四舍五入和decimal库精确四舍五入
java·linux·前端·数据库·vue.js·python·前端框架
Code哈哈笑1 小时前
【Java 学习】深度剖析Java多态:从向上转型到向下转型,解锁动态绑定的奥秘,让代码更优雅灵活
java·开发语言·学习
gb42152871 小时前
springboot中Jackson库和jsonpath库的区别和联系。
java·spring boot·后端
程序猿进阶1 小时前
深入解析 Spring WebFlux:原理与应用
java·开发语言·后端·spring·面试·架构·springboot