Java中 自定义注解怎么实现防刷限流 RateLimiter 拦截器

冬丽姑娘_9368

冬丽姑娘_9368

2026-07-21

951人浏览

原创

java中通过自定义@ratelimit注解+ratelimiter+拦截器实现声明式防刷限流:注解定义qps、超时等规则,拦截器解析注解并调用guava ratelimiter校验令牌,限流器按方法级缓存复用,业务代码零侵入。

java中 自定义注解怎么实现防刷限流 ratelimiter 拦截器

Java中通过自定义注解 + RateLimiter + 拦截器实现防刷限流,核心是把限流规则声明式地写在方法上,由拦截器统一解析、校验和拦截。关键不在“造轮子”,而在清晰分离:注解定义规则、拦截器执行判断、限流器(如 Guava 的 RateLimiter)提供底层令牌桶能力。

定义自定义限流注解

用 @interface 声明一个运行时保留、作用于方法的注解,支持配置 QPS(每秒请求数)、超时时间、拒绝提示等:

  • 必须包含 value():表示每秒允许通过的请求数(如 10)
  • 可选 timeout():获取令牌失败时等待的毫秒数(默认 0,即不等待直接拒绝)
  • 可选 msg():触发限流时返回的提示语(如“请求过于频繁,请稍后再试”)

示例:

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface RateLimit {
    double value();           // QPS,如 5.0 表示每秒最多 5 次
    long timeout() default 0; // 获取令牌超时毫秒数,0 表示非阻塞
    String msg() default "操作太频繁,请稍后重试";
}

基于 Guava RateLimiter 实现限流管理器

Guava 的 RateLimiter 是线程安全、轻量级的令牌桶实现。注意:不能为每个接口创建独立的 RateLimiter 实例并长期持有(内存泄漏风险),应按接口或路径做缓存管理:

  • 用 ConcurrentMap<string ratelimiter></string> 缓存不同限流规则对应的限流器
  • Key 可以是“类名.方法名”或自定义业务标识(如 user:login)
  • 首次访问时根据注解的 value() 创建并缓存;后续复用

示例工具类:

Java Maven Secondary Analysis
Java Maven Secondary Analysis

分析ZIP压缩包或GitLab仓库中的Java Maven项目,确定二次开发范围、类数量、模块分布及生产相关指标。

下载
public class RateLimitManager {
    private static final ConcurrentMap<string ratelimiter> LIMITERS = new ConcurrentHashMap();

    public static RateLimiter getOrCreate(String key, double qps) {
        return LIMITERS.computeIfAbsent(key, k -> RateLimiter.create(qps));
    }
}</string>

编写 Spring MVC 拦截器统一处理

继承 HandlerInterceptor,在 preHandle 中解析目标方法上的 @RateLimit 注解,获取限流参数,尝试获取令牌:

  • 从 HandlerMethod 提取 @RateLimit 注解
  • 构造唯一 key(如 handlerMethod.getMethod().getDeclaringClass().getName() + "." + handlerMethod.getMethod().getName())
  • 调用 RateLimiter.tryAcquire(timeout, TimeUnit.MILLISECONDS) 判断是否放行
  • 若失败,设置响应状态码(如 429 Too Many Requests)和 JSON 错误体,返回 false 中断流程

示例拦截器片段:

@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response,
                         Object handler) throws Exception {
    if (!(handler instanceof HandlerMethod)) return true;

    HandlerMethod hm = (HandlerMethod) handler;
    RateLimit annotation = hm.getMethodAnnotation(RateLimit.class);
    if (annotation == null) return true;

    String key = buildKey(hm);
    RateLimiter limiter = RateLimitManager.getOrCreate(key, annotation.value());

    boolean acquired = annotation.timeout() == 0
            ? limiter.tryAcquire()
            : limiter.tryAcquire(annotation.timeout(), TimeUnit.MILLISECONDS);

    if (!acquired) {
        response.setStatus(HttpStatus.TOO_MANY_REQUESTS.value());
        response.setContentType("application/json;charset=UTF-8");
        response.getWriter().write("{\"code\":429,\"msg\":\"" + annotation.msg() + "\"}");
        return false;
    }
    return true;
}

注册拦截器并使用注解

在 Spring 配置类中注册该拦截器(如继承 WebMvcConfigurer),指定拦截路径(建议精确匹配需限流的 API):

  • 避免对静态资源、健康检查等路径误拦截
  • 可在 addInterceptors().addInterceptor(...).excludePathPatterns(...) 中排除

在 Controller 方法上直接使用注解:

@PostMapping("/api/login")
@RateLimit(value = 3.0, timeout = 100, msg = "登录操作受限,请1分钟内最多尝试3次")
public Result login(@RequestBody LoginDTO dto) {
    return userService.login(dto);
}

这样就完成了声明式限流——无需改业务逻辑,只需加个注解,拦截器自动生效。

Java免费学习笔记:立即使用
解锁 Java 大师之旅:从入门到精通的终极指南

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

java

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
java
java

Java是一个通用术语,用于表示Java软件及其组件,包括“Java运行时环境 (JRE)”、“Java虚拟机 (JVM)”以及“插件”。php中文网还为大家带了Java相关下载资源、相关课程以及相关文章等内容,供大家免费下载使用。

2023.06.15

9977

6

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

2023.07.05

7122

9

java自学难吗
java自学难吗

Java自学并不难。Java语言相对于其他一些编程语言而言,有着较为简洁和易读的语法,本专题为大家提供java自学难吗相关的文章,大家可以免费体验。

2023.07.31

6272

8

java配置jdk环境变量
java配置jdk环境变量

Java是一种广泛使用的高级编程语言,用于开发各种类型的应用程序。为了能够在计算机上正确运行和编译Java代码,需要正确配置Java Development Kit(JDK)环境变量。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.01

1084

3

java保留两位小数
java保留两位小数

Java是一种广泛应用于编程领域的高级编程语言。在Java中,保留两位小数是指在进行数值计算或输出时,限制小数部分只有两位有效数字,并将多余的位数进行四舍五入或截取。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.02

908

3

java基本数据类型
java基本数据类型

java基本数据类型有:1、byte;2、short;3、int;4、long;5、float;6、double;7、char;8、boolean。本专题为大家提供java基本数据类型的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

1316

5

java有什么用
java有什么用

java可以开发应用程序、移动应用、Web应用、企业级应用、嵌入式系统等方面。本专题为大家提供java有什么用的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

2629

5

java在线网站
java在线网站

Java在线网站是指提供Java编程学习、实践和交流平台的网络服务。近年来,随着Java语言在软件开发领域的广泛应用,越来越多的人对Java编程感兴趣,并希望能够通过在线网站来学习和提高自己的Java编程技能。php中文网给大家带来了相关的视频、教程以及文章,欢迎大家前来学习阅读和下载。

2023.08.03

19951

3

配置java环境变量
配置java环境变量

配置Java环境变量是为了让操作系统能够识别和使用Java的相关命令和功能。本专题为大家提供配置java环境变量相关文章,帮助大家解决问题。

2023.08.03

1175

8

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
dev.java 官方:Learn Java
dev.java 官方:Learn Java

共0课时 | 0人学习

Java JDBC数据库连接官方教程
Java JDBC数据库连接官方教程

共0课时 | 0人学习