使用注释AOP实现接口访问次数限制

前言

在日常开发工作中，我们常有接口会暴露出来，虽然我们增加了各种检验和拦截可以拦截大多数恶意访问，但是你不能保证对接方的猿子不会造出一个死循环来访问你的接口，尤其是我们的程序作为一个平台使用的时候，别人的一个误操作可能会造成服务器宕机，到时候成千上万的客户都会受到影响，所以在这种对接过程中一定要对对方的接口访问次数进行限制！这种方式可以理解为微服务中的服务降级！

安排栗子

新建一个注释类：

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface LimitTime {
	// 访问次数，默认为10次
	int time() default 10;
	// 过期时间，时间戳间隔
	long timeout() default 1;
}

定义一个存放调用信息的DTO：

@Data
public class LimitDTO {
    //最近一次刷新时间戳
	private Long refreshTime;
	//剩余访问次数
	private Integer time;
}

新建一个切面类：
注意在存储访问状态对象的时候一定要使用ConcurrentHashMap，此为线程安全的map，支持并发访问！

@Component
@Order
@Aspect
public class LimitTimeAspect {
	private ConcurrentHashMap<String, LimitDTO> limitMap = new ConcurrentHashMap<>();

	@Pointcut("@annotation(limitTime)")
	public void limit(LimitTime limitTime) {
	}

	@Around("limit(limitTime)")
	public Object aroundLog(ProceedingJoinPoint joinpoint, LimitTime limitTime) throws Throwable {
	    //获取传入的最大访问次数
		int time= limitKey.time();
		//获取计算时间
		long timeout = limitKey.timeout();
		//获取访问方法
		Object target = joinpoint.getTarget().getClass().getName();
		String key= target.toString();
		//如果第一次访问该方法
		if (limitMap.get(key) == null) {
		    //新建一次对象存放访问信息
			LimitDTO limitDTO=new LimitDTO();
			limitDTO.setTime(time- 1);
			limitDTO.setRefreshTime(new Date().getTime());
			limitMap.put(key, limitDTO);
		}else {
		    //如果不是第一次访问，获取上次访问的信息
			LimitDTO limitDTO=limitMap.get(key);
			//如果和上次刷新时间比已经过期
			if (new Date().getTime() - limitDTO.getRefreshTime() > timeout) {
			    //将对象中的刷新时间和访问次数刷新
				limitDTO.setRefreshTime(new Date().getTime());
				limitDTO.setTime(time);
				limitMap.put(key, limitDTO);
			}
			//获取当前访问对象中的剩余访问次数
			int t = (int) limitMap.get(key).getTime;
			//如果访问次数大于0
			if (t > 0) {
			    //允许访问，并将访问次数-1
				limitDTO.setTime(--t);
			} else {
			    //如果已经没有访问次数，返回错误信息
				ResultBO<Object> resultBO = new ResultBO<>();
				resultBO.setCode(1);
				resultBO.setMsg("已达最大访问次数");
				return resultBO;
			}
		}
		//打印信息
		System.err.println("剩余次数：" + limitMap.get(key).getTime + " 方法名称:" + key);
		return joinpoint.proceed();
	}

使用该注释：

@RequestMapping(value = "/sendCmd", method = RequestMethod.POST)
@ResponseBody
@LimitKey(time= 10, timeout = 10000)//10秒内可以访问10次
public int sendCmd(@RequestBody List<CmdDO> cmds) {
	//do something
	return new ResultBO<>();
}

测试：


撒花！完成！