限制用户单位时间请求接口次数（aop和filter实现）

在项目中实现单位时间内限制用户访问指定接口次数问题，如有错误烦请指正交流

如限制用户10秒中内只能请求某一接口10次 这里我使用的是map方式存储计数，如果并发量大考虑放入redis中

核心限制类

package com.xuxu.limit; import java.util.concurrent.ConcurrentHashMap; import org.springframework.stereotype.Component; @Component public class LimitByMap { //全局MAP key为userId+URI value为CountParam 统计对象 记录时间和次数 public static ConcurrentHashMap<String, CountParam> cacheMap = new ConcurrentHashMap<>(); /** * * @param key 自定义key userId+URI * @param num 自定义限制请求次数 超出次数后限制访问 * @param limitTime 自定义限制时间，限制时间内限制访问次数 */ public void limitRequest(String key,int num,Long limitTime){ //1通过key获取到countPatam对象 CountParam countParam = cacheMap.get(key); //2判断对象是否为空，如果为空则代表第一次访问 如果不为空则代表之前请求过 if(countParam!=null){ //3获取时间，用当前时间减去最后一次访问时间，如果规定时间则清零访问次数，小于则累加计数 Long time = System.currentTimeMillis()-countParam.getTime(); int currentCount = countParam.getCount(); if(time<limitTime){ //4如果累计次数大于10，证明用户在限制时间内访问过多 if(currentCount>=num){ System.out.println("访问超出限制次数"); return; }else{ countParam.setCount(++currentCount); } }else{ countParam.setCount(1); } System.out.println(countParam.getCount()); countParam.setTime(System.currentTimeMillis()); }else{ countParam = new CountParam(); countParam.setCount(1); countParam.setTime(System.currentTimeMillis()); } cacheMap.put(key, countParam); } }

统计对象

package com.xuxu.limit; public class CountParam { private Long time; private int count; public Long getTime() { return time; } public void setTime(Long time) { this.time = time; } public int getCount() { return count; } public void setCount(int count) { this.count = count; } }

要被限制的接口

package com.xuxu.controller; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController; import com.xuxu.annotation.MyLimit; @RestController public class UserController { @RequestMapping("/index") public String index(){ return "index"; } }

1过滤器方式

package com.xuxu.filter; import java.io.IOException; import javax.servlet.Filter; import javax.servlet.FilterChain; import javax.servlet.ServletException; import javax.servlet.ServletRequest; import javax.servlet.ServletResponse; import javax.servlet.annotation.WebFilter; import javax.servlet.http.HttpServletRequest; import org.springframework.beans.factory.annotation.Autowired; import com.xuxu.limit.LimitByMap; @WebFilter(filterName="MyFilter",urlPatterns="/index") public class MyFilter implements Filter{ @Autowired private LimitByMap limitMap; @Override public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException { HttpServletRequest httpRequest = (HttpServletRequest) request; String requestURI = httpRequest.getRequestURI(); String userID = "123456"; String key = userID+requestURI; limitMap.limitRequest(key,10, 10000L); chain.doFilter(request, response); } }

启动类上加入@ServletComponentScan 过滤器生效

@SpringBootApplication @ServletComponentScan public class LimitUser01Application { public static void main(String[] args) { SpringApplication.run(LimitUser01Application.class, args); } }

2自定义注解加AOP方式

自定义注解

@Target({ElementType.METHOD}) @Retention(RetentionPolicy.RUNTIME) @Documented public @interface MyLimit { long limitTime() default 60000; int num() default 50; }

切面类

@Aspect @Component public class MyLimitAop { @Autowired private LimitByMap limitMap; @Pointcut("execution(* com.xuxu.controller.*.*(..))") public void cutPoint(){ } @Around("cutPoint()") public Object limit(ProceedingJoinPoint joinPoint) throws Throwable{ MethodSignature signature = (MethodSignature) joinPoint.getSignature(); Method method = signature.getMethod(); if(method.isAnnotationPresent(MyLimit.class)){ MyLimit myLimit = method.getAnnotation(MyLimit.class); long limitTime = myLimit.limitTime(); int num=myLimit.num(); String userID = "123456"; ServletRequestAttributes attributes = (ServletRequestAttributes)RequestContextHolder.getRequestAttributes(); HttpServletRequest request = attributes.getRequest(); String requestURI = request.getRequestURI(); String key = userID+requestURI; limitMap.limitRequest(key,num,limitTime); } return joinPoint.proceed(); } }

controller中方法上加入注解

@RequestMapping("/index") @MyLimit(limitTime=10000L,num=10) public String index(){ return "index"; }

启动类上加入aop支持

@SpringBootApplication @EnableAspectJAutoProxy public class LimitUser01Application { public static void main(String[] args) { SpringApplication.run(LimitUser01Application.class, args); } }