Aspect实现接口调用频率限制

小灰灰 2023-07-18 15:30 38阅读 0赞

aop 的思想，就是讲方法外到内执行，中间给切成三段，**未进入，进入，已退出**

# 背景： #

最近在做一个毕业设计，需要发送邮件（发送密码重置验证码），为了避免用户频繁调用邮件发送接口。  
前端：将按钮置灰，开启倒计时，倒计时结束恢复按钮可用  
由于没有登录，恶意的可能会使用代码去刷接口（在**未进入时将其拦截处理**）  
后端：  
使用自定义注解+aop 通过用户ip 限制短时间内的访问次数

# 准备： #

自定注解，可以配置多长时间内最多访问多少次  
aspect:处理标注了自定义注解的参数，执行相关逻辑。

# 代码实现 #

注解：

@Documented
    @Target({ ElementType.METHOD})
    @Retention(RetentionPolicy.RUNTIME)
    public @interface RequestFrequencyLimit { 
        //访问时间间隔 默认是60 秒
        int interval() default 60000;
    
        //访问次数限制
        int count() default 3;//默认interval 时间内最多访问3次
    }

aspect:

/** * 接口访问频率 限制aop */
    
    @Aspect
    @Component
    @Slf4j
    public class RequestFrequencyLimitAop { 
        Map<String, RequestRecord> records = new ConcurrentHashMap<>();//由于是单例，需要注意线程安全，当然也可以使用redis 替代
        @Autowired
        HttpServletRequest request;
    
        @Pointcut("@annotation(top.forethought.orderfoodyoulike.aops.RequestFrequencyLimit)")
        public void requestLimit() { 
    
        }
    
        @Around("requestLimit() && @annotation(requestFrequencyLimit)")
        public Object doAround(ProceedingJoinPoint joinPoint, RequestFrequencyLimit requestFrequencyLimit) throws Throwable { 
            Object res = null;
            long now = System.currentTimeMillis();
            if (isNull(requestFrequencyLimit)) { 
                return joinPoint.proceed();
            }
            //根据ip查找访问记录
            String clientIp = IpUtil.getIpAddress(request);
            RequestRecord requestRecord = records.get(clientIp);
            if (isNull(requestRecord) || now - requestRecord.firstRequestTime > requestFrequencyLimit.interval()) { 
                requestRecord = new RequestRecord();
                requestRecord.firstRequestTime = now;
                requestRecord.requestCount = 1;
            } else { 
                //更新访问次数
                requestRecord.requestCount++;
            }
            if (requestRecord.requestCount > requestFrequencyLimit.count()) { 
                log.error("访问太过频繁，请稍后再试，ip={}", clientIp);
                throw GlobalException.builder().msg("访问太过频繁，请稍后再试").build();
            }
            //保存访问记录
            records.put(clientIp, requestRecord);
            return joinPoint.proceed();
    
        }
    
        class RequestRecord { 
            int requestCount;//访问次数
            Long firstRequestTime;//第一次访问时间
    
        }
    }

## 使用效果： ##

![在这里插入图片描述][20200329150149469.png]  
60秒内同一个ip 最多访问该接口3次  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTIyOTI3_size_16_color_FFFFFF_t_70]

[20200329150149469.png]: /images/20230528/bd315a21d8044c4abcd41c28d4e32645.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTIyOTI3_size_16_color_FFFFFF_t_70]: /images/20230528/b61e452236f94a50ad882471537e34b1.png