Spring Boot怎么利用拦截器加缓存完成接口防刷
小编给大家分享一下SpringBoot怎么利用拦截器加缓存完成接口防刷,相信大部分人都还不怎么了解,因此分享这篇文章给大家参考一下,希望大家阅读完这篇文章后大有收获,下面让我们一起去了解一下吧!
为什么需要接口防刷为了减缓服务器压力,将服务器资源留待给有价值的请求,防止恶意访问,一般的程序都会有接口防刷设置,接下来介绍一种简单灵活的接口防刷操作
技术解析主要采用的技术还是拦截+缓存,我们可以通过自定义注解,将需要防刷的接口给标记出来管理,利用缓存统计指定时间区间里,具体的某个ip访问某个接口的频率,如果超过某个阈值,就让他进一会儿小黑屋,到期自动解放
主要代码前置环境搭建,Spring Boot项目,引入Web和Redis依赖
<parent><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-parent</artifactId><version>2.3.3.RELEASE</version></parent><dependencies><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-web</artifactId></dependency><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-data-redis</artifactId></dependency><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter</artifactId></dependency><dependency></dependencies>
自定义注解
定义拦截器,重写preHandler方法
@OverridepublicbooleanpreHandle(HttpServletRequestrequest,HttpServletResponseresponse,Objecthandler)throwsException{log.info("------------接口防刷拦截器---------------");if(handlerinstanceofHandlerMethod){HandlerMethodhandlerMethod=(HandlerMethod)handler;AccessLimitaccessLimit=handlerMethod.getMethodAnnotation(AccessLimit.class);//如果没有该注解,则不在防刷目标里,直接返回if(accessLimit==null){returntrue;}//获取最大访问次数intmaxAccessCnt=accessLimit.maxAccessCnt();//获取ipStringkey=getRealIp(request);//ip+请求的接口路径=>唯一标识keykey+=request.getRequestURI();//当前访问次数Objectvalue=redisTemplate.opsForValue().get(key);if(value==null){//第一次访问设置key保留时长为1sredisTemplate.opsForValue().set(key,1,1L,TimeUnit.SECONDS);}else{IntegercurrentCnt=(Integer)value;if(currentCnt<maxAccessCnt){//对应key值自增redisTemplate.opsForValue().increment(key);}else{//超出接口时间段内允许访问的次数,直接返回错误信息,同时设置过期时间20s自动剔除redisTemplate.expire(key,20L,TimeUnit.SECONDS);response.setContentType("application/json;charset=utf-8");try{OutputStreamout=response.getOutputStream();out.write("访问次数已达上线!请稍后再访问".getBytes(StandardCharsets.UTF_8));out.flush();out.close();}catch(IOExceptione){e.printStackTrace();}returnfalse;}}}returntrue;}
添加到需要防刷的接口上
/***返回不携带data的(成功例子)*@return*/@AccessLimit@GetMapping("/getPaperS")publicApiResultgetPaperInfoSuccess(){if(log.isInfoEnabled()){log.info("收到获取paper信息请求...");}//...业务逻辑if(log.isInfoEnabled()){log.info("完成获取paper信息请求,准备返回对象信息");}returnApiResultGenerator.success();}测试结果
正常情况下
到底时间段内最大访问次数时
以上是“SpringBoot怎么利用拦截器加缓存完成接口防刷”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注亿速云行业资讯频道!
声明:本站所有文章资源内容,如无特殊说明或标注,均为采集网络资源。如若本站内容侵犯了原著者的合法权益,可联系本站删除。