欢迎来到尧图网

客户服务 关于我们

您的位置:首页 > 财经 > 创投人物 > SpringBoot实现一个Redis限流注解

SpringBoot实现一个Redis限流注解

2025/3/16 4:28:07 来源:https://blog.csdn.net/weixin_60523038/article/details/146259662  浏览:    关键词:SpringBoot实现一个Redis限流注解

springBoot实现的一个限流注解

文章目录

目录

文章目录

一、使用步骤

1.引入库

2.代码实现

1.添加注解

2.新增限流AOP实现

 3.实现代码的拦截

 4.最终结果

总结

一、使用步骤

1.引入库

代码如下(示例):

        <!-- 引入SpringBoot Aop依赖 --><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-aop</artifactId></dependency>

2.代码实现

1.添加注解

package com.hhh.springai_test.annotation;import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface RedisLimiting {int number() default 3;int time() default 60;String message() default "请求过于频繁,请稍后再试";}

2.新增限流AOP实现

package com.hhh.springai_test.aop;import cn.hutool.crypto.digest.MD5;
import com.hhh.springai_test.annotation.RedisLimiting;
import com.hhh.springai_test.common.ErrorCode;
import com.hhh.springai_test.exception.BusinessException;
import com.hhh.springai_test.utils.RedisUtils;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;import java.lang.reflect.Method;
import java.lang.reflect.Parameter;@Aspect
@Component("redisLimitingAspect")
@Slf4j
public class RedisLimitingAspect {@Autowiredprivate RedisUtils redisUtils;@Around("@annotation(com.hhh.springai_test.annotation.RedisLimiting)")  // 只拦截带 @redisLimiting 的方法public Object redisLimiting(ProceedingJoinPoint joinPoint) throws Throwable {MethodSignature signature = (MethodSignature) joinPoint.getSignature();Method method = signature.getMethod(); // 直接获取被代理的方法// 获取 @redisLimiting 注解RedisLimiting annotation = method.getAnnotation(RedisLimiting.class);if (annotation == null) {return joinPoint.proceed(); // 没有注解,直接执行方法}int limit = annotation.number(); // 限制次数int expire = annotation.time();  // 过期时间String message = annotation.message();log.info("拦截方法: {}, 限流 key: {}, 限流次数: {}, 过期时间: {} 秒",method.getName(), limit, expire);// 执行限流逻辑boolean isAllowed = checkRedisLimiting(method, joinPoint.getArgs(), limit, expire);if (!isAllowed) {throw new BusinessException(ErrorCode.BUSY_ERROR,message);}return joinPoint.proceed(); // 执行原方法}private boolean checkRedisLimiting(Method method, Object[] args, int limit, int expire) {// 生成 Redis KeyString redisKey = generateRedisKey(method, args);// 查询 Redis 是否存在Object o = redisUtils.get(redisKey);if (o == null) {redisUtils.setex(redisKey, 1, expire); // 初始值设为1,并设置过期时间return true;} else {int count = Integer.parseInt(o.toString());if (count >= limit) {return false; // 超过限制} else {redisUtils.increment(redisKey, 1); // 递增计数return true;}}}private String generateRedisKey(Method method, Object[] args) {StringBuilder builder = new StringBuilder();builder.append(method.getDeclaringClass().getName()).append(":").append(method.getName()).append(":");Parameter[] parameters = method.getParameters();for (int i = 0; i < parameters.length; i++) {builder.append(parameters[i].getName()).append("=").append(args[i]).append("&");}return MD5.create().digestHex16(builder.toString()); // 生成唯一 Redis Key}
}

 3.实现代码的拦截

@GetMapping("/getAllModel")@RedisLimiting(number = 3, time = 60,message = "不要再请求我的获取aiModel方法了")public BaseResponse<List<AiModelVO>> getAllModel() {return ResultUtils.success(aiModelService.getAllModel());}

 4.最终结果


总结

提示:这里对文章进行总结:
例如:以上就是今天要讲的内容,本文仅仅简单介绍了pandas的使用,而pandas提供了大量能使我们快速便捷地处理数据的函数和方法。

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

热搜词