spring boot + redis 接口访问频率限制

迷南。 2023-01-08 02:21 193阅读 0赞

@Component
    public class BlackInterceper implements HandlerInterceptor {
    	
    	@Autowired
    	private RedisTemplate<String, Object> redisTemplate;
    	
    	private Logger log = LoggerFactory.getLogger(this.getClass());
    	
    	@Override
    	public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
    			throws Exception {
    		response.setHeader("Content-type", "text/html;charset=UTF-8");
    		String token = request.getHeader(Cons.TOKEN.WECHAT);
    		String requestURI = request.getRequestURI();
    
    		if (StringUtils.isBlank(token)) {
    			response.setHeader("Content-type", "text/html;charset=UTF-8");
    			response.getWriter().write(JsonUtils.marshalToString(ReturnResult.build(401, "未授权")));
    			return false;
    		}
    		Integer userId = (Integer) redisTemplate.opsForValue().get(Cons.TOKEN.WECHAT + ":" + token);
    		log.error("userId={},访问了url={},请求ip={}",token,requestURI, IpUtil.getIpAddress(request));
    		if(redisTemplate.hasKey("black")){
    			if(redisTemplate.opsForSet().isMember("black", userId)){
    				response.getWriter().write(JsonUtils.marshalToString(ReturnResult.build(500, "由于存在恶意攻击你已被限制访问")));
    				return false;
    			}
    		}
    		Integer count = (Integer)redisTemplate.opsForValue().get("limit:"+token);
    		if(count==null){
    			redisTemplate.opsForValue().set("limit:"+token, 1, 60, TimeUnit.SECONDS);
    			return true;
    		}else{
    			if(count>100 && count<150){
    				response.getWriter().write(JsonUtils.marshalToString(ReturnResult.build(500, "请求太频繁，请稍后再试")));
    				redisTemplate.opsForValue().increment("limit:"+token, 1);
    				return false;
    			}else if(count>=150){
    				redisTemplate.opsForSet().add("black",userId,2,TimeUnit.DAYS);
    				response.getWriter().write(JsonUtils.marshalToString(ReturnResult.build(500, "请求太频繁，已经被限制访问")));
    
    				//redisTemplate.opsForSet().add("black",token);
    				return false;
    			}else{
    				redisTemplate.opsForValue().increment("limit:"+token, 1);
    				return true;
    			}
    		}
    	}
    
    	@Override
    	public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler,
    			ModelAndView modelAndView) throws Exception {
    		// TODO Auto-generated method stub
    
    	}
    
    	@Override
    	public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex)
    			throws Exception {
    		// TODO Auto-generated method stub
    
    	}
    	
    
    }