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 "访问太过频繁";
        }
    }

}
相关推荐
xmh-sxh-131416 分钟前
jdk各个版本介绍
java
XINGTECODE30 分钟前
海盗王集成网关和商城服务端功能golang版
开发语言·后端·golang
天天扭码36 分钟前
五天SpringCloud计划——DAY2之单体架构和微服务架构的选择和转换原则
java·spring cloud·微服务·架构
程序猿进阶36 分钟前
堆外内存泄露排查经历
java·jvm·后端·面试·性能优化·oom·内存泄露
FIN技术铺41 分钟前
Spring Boot框架Starter组件整理
java·spring boot·后端
小曲程序1 小时前
vue3 封装request请求
java·前端·typescript·vue
凡人的AI工具箱1 小时前
15分钟学 Go 第 60 天 :综合项目展示 - 构建微服务电商平台(完整示例25000字)
开发语言·后端·微服务·架构·golang
陈王卜1 小时前
django+boostrap实现发布博客权限控制
java·前端·django
小码的头发丝、1 小时前
Spring Boot 注解
java·spring boot
先天牛马圣体1 小时前
如何提升大型AI模型的智能水平
后端