使用Redis实现的限流注解机制-CSDN博客

本文链接：https://blog.csdn.net/qq_42977003/article/details/129433676

该文章介绍了一个基于注解的限流实现，通过@RateLimiter定义限流规则，如限流key、时间和次数。在AOP切面中，利用RedisTemplate进行限流操作，检查请求频率并抛出异常以控制访问。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

限流注解

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface RateLimiter {
    //限流的key
    String key() default Constants.RATE_LIMIT_KEY;

    //限流的时间
    int time() default 10;

    //限流的次数
    int count() default 1;

    //类型
    String type() default "";

    //限流后返回的文字
    String msg() default "访问太过频繁，请求稍后重试!";
}

@Aspect
@Component
@RequiredArgsConstructor
public class RateLimiterAspect {
    public static final Logger logger = LoggerFactory.getLogger(RateLimiterAspect.class);

    @Resource
    RedisTemplate redisTemplate;

    @Pointcut(value = "@annotation(rateLimiter)")
    public void pointCut(RateLimiter rateLimiter) {
    }

    @Before("pointCut(rateLimiter)")
    public void doBefore(JoinPoint joinPoint, RateLimiter rateLimiter) throws Exception {
        this.handle(joinPoint, rateLimiter);
    }

    private void handle(JoinPoint joinPoint, RateLimiter rateLimiter) throws Exception {
        //获取当前用户的id
        //拿到HttpServletRequest对象
        RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
        HttpServletRequest request = (HttpServletRequest) requestAttributes.resolveReference(requestAttributes.REFERENCE_REQUEST);

        try {
            MethodSignature signature = (MethodSignature) joinPoint.getSignature();
            Method method = signature.getMethod();
            Class<?> targetClass = method.getDeclaringClass();

            //key = ratelimiter:ip:类名
            String ip = IpUtils.getIp(request) + ":";

            StringBuffer stringBuffer = new StringBuffer(RATELIMITER);
            stringBuffer.append(ip).append(":").append(method.getName());

            String key = stringBuffer.toString();

            if (redisTemplate.hasKey(key)) {
                redisTemplate.opsForValue().increment(key);
                int num = Integer.parseInt((String) redisTemplate.opsForValue().get(key.toString()));
                if (num > rateLimiter.count()) {
                    throw new RateLimiterException("网络繁忙，请稍候再试!");
                }
            } else {
                redisTemplate.opsForValue().set(key, "1", Long.valueOf(rateLimiter.time()), TimeUnit.SECONDS);
            }
        } catch (RateLimiterException e) {
            throw new RateLimiterException("网络繁忙，请稍候再试");
        } catch (Exception e) {
            logger.error(e.getMessage());
            throw new Exception();
        }
    }
}