温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

怎么在spring boot中限制redis 接口的访问频率

发布时间:2021-01-29 15:48:50 来源:亿速云 阅读:239 作者:Leah 栏目:开发技术

本篇文章为大家展示了怎么在spring boot中限制redis 接口的访问频率,内容简明扼要并且容易理解,绝对能使你眼前一亮,通过这篇文章的详细介绍希望你能有所收获。

生产环境下可以解决的问题:

  1.短信验证码请求评率限制(防止抓包短信轰炸)

  2.热点数据请求评率限制(防止数据库爆炸)

@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    } }

上述内容就是怎么在spring boot中限制redis 接口的访问频率,你们学到知识或技能了吗?如果还想学到更多技能或者丰富自己的知识储备,欢迎关注亿速云行业资讯频道。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI