使用 Spring AOP 和 Guava RateLimiter 实现 API 限流

embedded/2024/11/15 0:50:04/

在高并发的应用场景下,合理的限流策略是保证系统稳定性的重要手段之一。限流可以防止系统资源被耗尽,避免雪崩效应的发生。本文将介绍如何使用 Spring AOP 和 Guava RateLimiter 实现API限流,并支持自定义限流超时时间。

引入依赖

首先,需要在 pom.xml 中引入 Guava 依赖:

<dependency><groupId>com.google.guava</groupId><artifactId>guava</artifactId><version>31.1-jre</version>
</dependency>

定义注解

其次,定义一个 @ApiRateLimit 注解,在需要限流的方法上标注该注解:

java">import java.lang.annotation.*;
import java.util.concurrent.TimeUnit;@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface ApiRateLimit {double qps() default 1; // 每秒钟生成令牌的速率long timeout() default 0; // 尝试获取令牌的超时时间TimeUnit timeUnit() default TimeUnit.SECONDS; // 超时时间单位
}
  • qps 参数控制每秒生成令牌数,即控制限流速率
  • timeouttimeUnit 参数控制获取令牌的超时时间,0表示无超时

实现切面

接下来实现 ApiRateLimitAspect 切面类,在方法执行前通过 RateLimiter 判断是否被限流:

java">import com.google.common.util.concurrent.RateLimiter;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.springframework.stereotype.Component;import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.TimeUnit;@Aspect
@Component
public class ApiRateLimitAspect {private final Map<String, RateLimiter> rateLimiters = new ConcurrentHashMap<>();@Before("@annotation(apiRateLimit)")public void limit(JoinPoint joinPoint, ApiRateLimit apiRateLimit) {String methodName = joinPoint.getSignature().toLongString();double qps = apiRateLimit.qps();RateLimiter limiter = rateLimiters.computeIfAbsent(methodName, k -> RateLimiter.create(qps));long timeout = apiRateLimit.timeout();TimeUnit timeUnit = apiRateLimit.timeUnit();if (timeout > 0) {if (!limiter.tryAcquire(timeout, timeUnit)) {throw new RuntimeException("API rate limit exceeded");}} else {if (!limiter.tryAcquire()) {throw new RuntimeException("API rate limit exceeded");}}}
}
  • 使用 ConcurrentHashMap 缓存每个方法对应的 RateLimiter 实例
  • 根据 @ApiRateLimit 注解的参数尝试获取令牌
    • 若超时时间大于0,使用 tryAcquire(timeout, timeUnit) 获取令牌
    • 若超时时间为0,使用 tryAcquire() 获取令牌
  • 如果无法获取令牌,抛出 RuntimeException 限流异常

使用示例

在 Controller 方法上标注 @ApiRateLimit 注解即可实现限流:

java">@RestController
public class DemoController {@GetMapping("/test")@ApiRateLimit(qps = 2, timeout = 200, timeUnit = TimeUnit.MILLISECONDS)public String test() {return "hello world";}
}

上述代码对 /test 接口限流,限流速率为每秒 2 个请求,获取令牌超时时间为 200 毫秒。

总结

通过简单的注解和 AOP 切面,就可以实现 API 限流功能,并支持自定义限流速率和限流超时时间。这种实现方式无侵入性,添加或移除限流只需要在方法上增加或移除注解即可,降低了维护成本。

值得注意的是,在分布式环境下,单机限流的方式可能无法满足需求,我们需要结合分布式限流组件如 Redis 等来实现全局限流。另外,限流策略也可以根据实际场景进行优化,如配合熔断、降级等策略使用,以提高系统的稳定性。


http://www.ppmy.cn/embedded/7064.html

相关文章

【C++风云录】解锁智慧之门:物联网安全工具和库助力打造安全可靠的智能家居

物联网安全&#xff1a;保护智能家居和设备数据的关键 前言 随着物联网的快速发展&#xff0c;智能家居和物联网设备正成为我们日常生活中不可或缺的一部分。然而&#xff0c;随之而来的是对设备安全性的关注&#xff0c;因为这些设备存储了大量的个人和敏感数据。为了确保智…

C#Lock锁

问题&#xff1a;避免使用lock(this)或lock&#xff08;string)&#xff0c;因为无法保证你提供的方法&#xff0c;在外部类中使用的时候&#xff0c;开发人员会不会锁定当前对象。解决方法&#xff1a;最佳做法是定义 private 对象来锁定, 或 private static 对象变量来保护所…

贪心算法中常见的使用方法逻辑整理

贪心算法 常见的使用方法逻辑整理 1. 贪心算法 特点 贪心算法&#xff0c;又名贪婪法&#xff0c;是寻找最优解问题的常用方法&#xff0c;这种方法模式一般将求解过程分成若干个步骤&#xff0c;但每个步骤都应用 贪心原则 &#xff0c;选取当前状态下最好/最优的选择&#x…

Android网络安全配置:允许明文HTTP通信的正确姿势20240418

引言&#xff1a; 随着Android 9 (Pie) 的发布&#xff0c;Google加强了平台的安全性要求&#xff0c;特别是对网络通信的安全性进行了重大更新。默认情况下&#xff0c;Android系统禁止应用使用未加密的明文HTTP进行网络通信。这篇博客将介绍如何在遵守新安全政策的同时&…

React中redux、react-redux、@reduxjs/toolkit状态管理库的使用方式

效果 下载依赖 npm install redux react-redux reduxjs/toolkit --save在src目录下创建文件 创建index.ts文件 import { configureStore } from reduxjs/toolkit import userSlice from ./userReducerconst store configureStore({reducer: {user: userSlice.reducer} }) //…

Python 数据结构和算法实用指南(二)

原文&#xff1a;zh.annas-archive.org/md5/66ae3d5970b9b38c5ad770b42fec806d 译者&#xff1a;飞龙 协议&#xff1a;CC BY-NC-SA 4.0 第四章&#xff1a;列表和指针结构 我们已经在 Python 中讨论了列表&#xff0c;它们方便而强大。通常情况下&#xff0c;我们使用 Python…

【linux运维】系统常见管理命令

系列综述&#xff1a; &#x1f49e;目的&#xff1a;本系列是个人整理为了学习基本的shell编程和linux命令&#xff0c;整理期间苛求每个知识点&#xff0c;平衡理解简易度与深入程度。 &#x1f970;来源&#xff1a;材料主要源于b站大学——linux运维课程进行的&#xff0c;…

目标检测——食品饮料数据集

一、重要性及意义 对食品和饮料进行目标检测的重要性和意义体现在多个方面&#xff1a; 商业应用与市场分析&#xff1a;目标检测技术在食品和饮料行业有着广泛的应用前景。通过对超市货架、餐馆菜单或广告海报中的食品和饮料进行自动识别和计数&#xff0c;商家可以获取关于产…