自定义限流注解

自定义注解
/*** 速率限制注解** @author: 张定辉* @date: 2024/3/5 21:29* @description: 速率限制注解*/
@Target({ElementType.TYPE,ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface RateLimit {/*** SPEL表达式* <p>* 1.使用方法的基本类型参数作为限流Key* <p>* &#064;RateLimit(value="#id")* public void test(String id){}* <p><p>* 2.使用方法的对象类型参数中的某个属性作为限流Key* <p>* &#064;RateLimit(value="#user.username")* public void test(User user){}* <p><p>* 3.将方法参数作为bean方法的参数并获取返回值作为限流Key,暂时只支持bean的方法是String类型* <p>* &#064;Service(value="parseBean")<p>* public class ParseBean{<p>* &nbsp;&nbsp;&nbsp;public String parse(String arg){<p>* &nbsp;&nbsp;&nbsp;&nbsp;return arg+"limitKey";<p>* &nbsp;&nbsp;&nbsp;}<p>* }<p>*<p>* &#064;RateLimit(value="@parseBean.parse(username)")<p>* public void test(String username){}*/String value();/*** 限流间隔，以秒为单位*/int interval()default 3;/*** 单位之间内的速率限制*/int frequency()default 20;
}
SPEL配置类
/*** Spel表达式配置类** @author: 张定辉* @date: 2024/3/7 14:20* @description: Spel表达式配置类*/
@Configuration
public class SpelConfig {@Beanpublic StandardEvaluationContext evaluationContext(ApplicationContext applicationContext) {StandardEvaluationContext context = new StandardEvaluationContext();context.addPropertyAccessor(new BeanFactoryAccessor());context.setBeanResolver(new BeanFactoryResolver(applicationContext));context.setTypeLocator(new StandardTypeLocator(applicationContext.getClassLoader()));context.setTypeConverter(new StandardTypeConverter());return context;}
}
AOP切面
/*** 速率限制注解处理器** @author: 张定辉* @date: 2024/3/5 21:37* @description: 速率限制注解处理器*/
@Aspect
@Component
@RequiredArgsConstructor
public class RateLimitHandler {private final ApplicationContext applicationContext;private final SpelExpressionParser parser = new SpelExpressionParser();private final StandardEvaluationContext context;private final RedisTemplate<String, Object> redisTemplate;@SneakyThrows@Around("@within(com.ai.common.annotation.RateLimit) || @annotation(com.ai.common.annotation.RateLimit)")public Object handler(ProceedingJoinPoint joinPoint) {Object target = joinPoint.getTarget();String spelValue;int interval;int frequency;//如果注解是标注在类上if (target.getClass().isAnnotationPresent(RateLimit.class)) {Class<?> aClass = target.getClass();RateLimit annotation = aClass.getAnnotation(RateLimit.class);spelValue = annotation.value();interval = annotation.interval();frequency = annotation.frequency();if (spelValue.startsWith("@")) {addBeanResultToContext(context, spelValue);}}//注解标注在方法上else {Object[] args = joinPoint.getArgs();MethodSignature signature = (MethodSignature) joinPoint.getSignature();Method method = signature.getMethod();RateLimit rateLimit = method.getAnnotation(RateLimit.class);interval = rateLimit.interval();frequency = rateLimit.frequency();spelValue = rateLimit.value();String[] parameterNames = signature.getParameterNames();for (int i = 0; i < args.length; i++) {//这行代码在后续的使用bean的方法返回值作为KEY限流时有用处context.setVariable(parameterNames[i], args[i]);if (args[i] != null && !isPrimitive(args[i].getClass())) {addObjectPropertiesToContext(context, parameterNames[i], args[i]);}}if (spelValue.startsWith("@")) {spelValue = addBeanResultToContext(context, spelValue);}}Expression expression = parser.parseExpression(spelValue);Object key = expression.getValue(context);//使用Redis进行限流redisRateLimit(JSON.toJSONString(key), interval, frequency);return joinPoint.proceed();}/*** 添加对象属性值到SPEL上下文环境中*/@SneakyThrowsprivate void addObjectPropertiesToContext(StandardEvaluationContext context, String paramName, Object arg) {Class<?> clazz = arg.getClass();Method[] methods = clazz.getMethods();for (Method method : methods) {String methodName = method.getName();if (methodName.startsWith("get") && !methodName.equals("getClass")) {String propertyName = methodName.substring(3, 4).toLowerCase() + methodName.substring(4);Object propertyValue = method.invoke(arg);context.setVariable(paramName + "." + propertyName, propertyValue);}}}/*** 将Bean方法的执行结果设置到SPEL上下文环境中*/@SneakyThrowsprivate String addBeanResultToContext(StandardEvaluationContext context, String spelValue) {Object bean = applicationContext.getBean(spelValue.substring(1, spelValue.indexOf(".")));String methodName = spelValue.substring(spelValue.indexOf(".") + 1, spelValue.indexOf("("));String[] methodArgs = spelValue.substring(spelValue.indexOf("(") + 1, spelValue.indexOf(")")).split(",");Object[] methodArgsValues = new Object[methodArgs.length];for (int i = 0; i < methodArgs.length; i++) {methodArgsValues[i] = context.lookupVariable(methodArgs[i]);if (Objects.isNull(methodArgsValues[i])) {methodArgsValues[i] = methodArgs[i];}}Class<?>[] argumentsTypes = getArgumentsTypes(methodArgsValues);boolean b = Arrays.stream(argumentsTypes).allMatch(Objects::isNull);Method beanMethod = bean.getClass().getMethod(methodName, b?new Class<?>[0]:argumentsTypes);Object beanMethodResult = beanMethod.invoke(bean, b?null:methodArgsValues);context.setVariable("beanMethodResult", beanMethodResult);return "#beanMethodResult";}/*** 获取参数的类型*/private Class<?>[] getArgumentsTypes(Object[] args) {Class<?>[] types = new Class<?>[args.length];for (int i = 0; i < args.length; i++) {Class<?> aClass = args[i].getClass();if (aClass.isAssignableFrom(String.class)) {String arg = (String) args[i];types[i] = StringUtils.isBlank(arg) ? null : aClass;} else {types[i] = aClass;}}return types;}/*** 判断是否是基础数据类型*/private boolean isPrimitive(Class<?> clazz) {return clazz.isPrimitive() || clazz == String.class || clazz == Integer.class|| clazz == Long.class || clazz == Double.class || clazz == Float.class|| clazz == Boolean.class || clazz == Character.class || clazz == Short.class|| clazz == Byte.class;}/*** 结合Redis进行限流操作*/private void redisRateLimit(String key, int interval, int frequency) throws OperationsException {long l = execLua(key, interval);if (l > frequency) {throw new OperationsException("操作过于频繁，请稍后再试!");}}/*** 使用Lua脚本执行原子性的Redis操作，* 如果key不存在则设置value为1并且设置过期时间为5秒，* 如果key存在则进行累加。避免多线程并发时，由于key被修改过导致设置过期时间时失败从而导致key永不失效** @return 如果没有key则返回1，如果有key则返回累加后的value*/private long execLua(String key, int expireTime) {String luaScript = """if redis.call('exists', KEYS[1]) == 0 thenredis.call('set', KEYS[1], 1, 'ex', %s)return 1elsereturn redis.call('incr',KEYS[1])end""".formatted(expireTime);RedisScript<Long> script = new DefaultRedisScript<>(luaScript, Long.class);Long result = redisTemplate.execute(script, Collections.singletonList(key));return Objects.isNull(result) ? 0 : result;}
}
定义Bean方法解析的业务类

该业务类主要是为了满足在使用自定义注解时我们会使用某个类的方法的返回值作为限流Key，这个类自己自定义即可，这里只是做简单的演示使用

/*** @author: 张定辉* @date: 2024/3/7 11:48* @description: 使用方法返回值作为限流Key的业务方法*/
@Service(value = "parseService")
public class ParseService {public String parse(String param){return param+"yyds";}public String parse2(){return "yyds";}
}
实际应用
注解标注在接口方法上，使用方法参数作为限流Key

5秒内只能访问两次该接口

  @GetMapping("/test")@RateLimit(value = "#id",interval = 5,frequency = 2)public Res<Object> test(@RequestParam String id){return Res.success();}
标注在接口方法上，使用对象的属性值作为限流Key

5秒内只能访问两次该接口

   @PostMapping("/test")@RateLimit(value = "#user.username",interval = 5,frequency = 2)public Res<Object> test(@RequestBody User user){return Res.success();}
标注在接口方法上，使用 parseService 业务类型的 parse方法返回值作为限流Key

5秒内只能访问两次该接口

   @GetMapping("/test")@RateLimit(value = "@parseService.parse(id)",interval = 5,frequency = 2)public  Res<Object> test(@RequestParam String id){return Res.success();}
标注在接口类下，实现该接口下的所有接口方法都限流
/**
* @author: 张定辉
* @date: 2024/3/7 14:14
* @description:
*/
@RequestMapping("/test")
@RestController
@RateLimit(value = "@parseService.parse2()",interval = 5,frequency = 2)
public class Text2Controller {@GetMapping("/test1")public Res<Object> test1(){return Res.success();}@GetMapping("/test2")public Res<Object> test2(){return Res.success();}
}

写的可能不是很完善，如果有大佬能够指正的话不甚感激

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.mzph.cn/news/729717.shtml

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！

相关文章

SpringAOP面向切面编程-代理

SpringAOP面向切面编程-代理

目录 1.静态代理 2.动态代理 代理在开发中实现的方式具体有两种&#xff1a;静态代理&#xff0c;[动态代理技术] 1.静态代理 主动创建代理类 public class CalculatorStaticProxy implements Calculator {// 将被代理的目标对象声明为成员变量private Calculator target;…
阅读更多...
面试问答之MySQL数据库进阶

面试问答之MySQL数据库进阶

文章目录 &#x1f412;个人主页&#xff1a;信计2102罗铠威&#x1f3c5;JavaEE系列专栏&#x1f4d6;前言&#xff1a;&#x1f380; MySQL架构&#x1f415;数据库引擎&#x1f415; InnoDB存储存储引擎&#x1f415;MYISAM &#x1f3e8;索引&#x1f415;哪些情况需要创建…
阅读更多...
vue3+element-plus el-input 自动获取焦点

vue3+element-plus el-input 自动获取焦点

虽然element有提供input的autofocus属性&#xff0c;但是当我们第二次进入页面就会发现autofocus已经不再生效&#xff0c;需要通过onMounted去触发input的focus解决这个问题。 1.先给el-input绑定一个ref&#xff1a;<el-input ref"inputRef" v-model"inp…
阅读更多...
景联文科技：专业提供高质量大语言模型训练数据

景联文科技：专业提供高质量大语言模型训练数据

2024年&#xff0c;数字经济被再次写入政府工作报告中&#xff0c;报告指出要深化大数据、人工智能等研发应用&#xff0c;打造具有国际竞争力的数字产业集群。 大模型作为生成式人工智能的基础&#xff0c;日益成为国际科技竞争的焦点。人大代表杨剑宇指出&#xff0c;尽管我国…
阅读更多...
008-slot插槽

008-slot插槽

slot插槽 1、插槽 slot 的简单使用2、插槽分类2.1 默认插槽2.2 具名插槽2.3 作用域插槽 插槽就是子组件中的提供给父组件使用的一个占位符&#xff0c;用<slot></slot> 表示&#xff0c;父组件可以在这个占位符中填充任何模板代码&#xff0c;如 HTML、组件等&…
阅读更多...
老阳分享：视频号带货的四大技巧

老阳分享：视频号带货的四大技巧

视频号带货作为新兴的电商模式&#xff0c;在微信这个庞大的社交平台上展现出了巨大的潜力。要想在视频号带货领域取得成功&#xff0c;需要掌握一定的技巧。本文将为您详细解析视频号带货的四大技巧&#xff0c;感兴趣的朋友一起去看看吧。 一、内容创新 内容创新是视频号带货…
阅读更多...
【深度学习笔记】5_11 残差网络ResNet

【深度学习笔记】5_11 残差网络ResNet

注&#xff1a;本文为《动手学深度学习》开源内容&#xff0c;部分标注了个人理解&#xff0c;仅为个人学习记录&#xff0c;无抄袭搬运意图 5.11 残差网络&#xff08;ResNet&#xff09; 让我们先思考一个问题&#xff1a;对神经网络模型添加新的层&#xff0c;充分训练后的…
阅读更多...
Express框架的产生

Express框架的产生

Express框架的产生&#xff0c;解决的痛点是什么&#xff1f; 1.优化Node.js在Web的开发 Express框架是一个基于Node.js的Web应用程序开发框架&#xff0c;它的产生主要是为了解决Node.js在Web开发中的一些痛点。 在Node.js出现之前&#xff0c;Web开发主要是基于传统的后端…
阅读更多...
springboot项目集成，项目流程概述

springboot项目集成，项目流程概述

一、项目介绍 二、项目设计原则 2.1整体原则 2.2持久层 2.3业务逻辑层 具体分析 三、实战 3.1项目搭建 <dependency><groupId>org.springframework.security</groupId><artifactId>spring-security-crypto</artifactId></dependency>&l…
阅读更多...
双链表的实现（数据结构）

双链表的实现（数据结构）

链表总体可以分为三大类 一、无头和有头 二、单向和双向 三、循环和不循环 从上面分类得知可以组合成8种不同类型链表&#xff0c;其中单链表最为简单&#xff0c;双联表最为复杂&#xff0c;两种链表都实现后其余链表都不成问题。 我们前期博客已将完成了单向无头不循环链表…
阅读更多...
基于PHP的景点数据分析系统设计与实现

基于PHP的景点数据分析系统设计与实现

目 录 摘 要 I Abstract II 引 言 1 1 关键理论与技术 3 1.1 框架技术 3 1.1.1 QueryList 3 1.1.2 ThinkPHP 3 1.1.3 Amaze UI 3 1.2 数据可视化技术 4 1.3 数据库技术 4 1.4 本章小结 4 2 需求分析 5 2.1 业务流程分析 5 2.2 功能需求分析 5 2.3 用例分析 7 2.4 非功能性需求…
阅读更多...
it-tools工具箱

it-tools工具箱

it-tools 是一个在线工具集合&#xff0c;包含各种实用的开发工具、网络工具、图片视频工具、数学工具等 github地址&#xff1a;https://github.com/CorentinTh/it-tools 部署 docker run -d --name it-tools --restart unless-stopped -p 8080:80 corentinth/it-tools:lat…
阅读更多...
gradle 相关

gradle 相关

aar 不加 aar 以及 transitive true library可以通过多种格式上传到远程仓库&#xff0c;比如大部分情况下用到的.jar或.aar。当没有指定后缀的话&#xff0c;依赖的时候将会下载它的默认格式&#xff08;由上传方定义&#xff0c;如果没有定义则默认为.jar&#xff09;的Lib…
阅读更多...
yolov8多batch推理，nms后处理

yolov8多batch推理，nms后处理

0. 背景 在高速公路监控视频场景下&#xff0c;图像分辨率大都是1920 * 1080或者2560 * 1440&#xff0c;远处的物体&#xff08;车辆和行人等&#xff09;都比较小。考虑需要对图像进行拆分&#xff0c;然后把拆分后的数据统一送入模型中&#xff0c;推理的结果然后再做nms&am…
阅读更多...
redis centos7 单点搭建

redis centos7 单点搭建

redis centos 安装步骤 下载源文件编译Redis拷贝编译后文件修改配置文件启动redis 下载源文件 wget https://download.redis.io/redis-stable.tar.gz编译Redis tar -xzvf redis-stable.tar.gz cd redis-stable make如果编译成功&#xff0c;你会在src目录中找到几个 Redis 二…
阅读更多...
Android Selinux详解[一]---整体介绍

Android Selinux详解[一]---整体介绍

Android 使用安全增强型 Linux (SELinux) 对所有进程强制执行强制访问控制 (MAC)&#xff0c;甚至包括以 Root/超级用户权限运行的进程&#xff08;Linux 功能&#xff09;。 借助 SELinux&#xff0c;Android 可以更好地保护和限制系统服务、控制对应用数据和系统日志的访问、…
阅读更多...
【微前端乾坤】 vue2主应用、vue2+webpack子应用，vue3+webpack子应用、vue3+vite子应用的配置

【微前端乾坤】 vue2主应用、vue2+webpack子应用，vue3+webpack子应用、vue3+vite子应用的配置

因公司需求 需要将原本vue2iframe 形式的项目改成微前端乾坤的方式。 之前iframe都是直接嵌套到vue2项目的二级目录或者三级目录下的(反正就是要随处可嵌) 用乾坤的原因&#xff1a; 1、iframe嵌套的方式存在安全隐患&#xff1b; 2、项目是联合开发的&#xff0c; 乾坤的方便…
阅读更多...
Doris画像存储实践系列二

Doris画像存储实践系列二

上一篇: Doris画像存储系列一(https://editor.csdn.net/md/?articleId120416295) 六、画像宽表bitmap倒排表 重复一下bitmap倒排表的优点和缺点 标签类型标签值user_ids性别男1,2性别发3 优点: doris bitmap聚合表在对做用户画像群体计算时很友好,交集/并集/差集因为数据…
阅读更多...
SQL23 统计每个学校各难度的用户平均刷题数

SQL23 统计每个学校各难度的用户平均刷题数

题解 | #统计每个学校各难度的用户平均刷题数# 题意明确&#xff1a; 计算每个学校用户不同难度下的用户平均答题题目数 问题分解&#xff1a; 限定条件&#xff1a;无&#xff1b;每个学校&#xff1a;按学校分组group by university不同难度&#xff1a;按难度分组group b…
阅读更多...
Hack The Box-Crafty

Hack The Box-Crafty

目录 信息收集 rustscan whatweb WEB 漏洞利用 漏洞说明 漏洞验证 提权 get user.txt get Administrator 总结 信息收集 rustscan ┌──(root㉿ru)-[~/kali/hackthebox] └─# rustscan -a 10.10.11.249 --range0-65535 --ulimit5000 -- -A -sC [~] Automatically…
阅读更多...
最新文章

Copyright @ 2022~2023