Spring Boot接口限制访问次数 _Spring

定义注解
import JAVA.lang.annotation.ElementType;import java.lang.annotation.Retention;import java.lang.annotation.RetentionPolicy;import java.lang.annotation.Target;/** * 接口防刷注解类 * @Author: Cyz * @Description: * @Date: create in 2022/6/28 16:55 */@Retention(RetentionPolicy.RUNTIME)@Target(ElementType.METHOD)public @interface AccessLimit {int seconds();int maxCount();boolean needLogin() default true;}【Spring Boot接口限制访问次数】编写接口防刷拦截器
import com.cyz.blog.utils.AccessLimit;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.data.redis.core.RedisTemplate;import org.springframework.stereotype.Component;import org.springframework.web.method.HandlerMethod;import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;import javax.servlet.http.HttpServletRequest;import javax.servlet.http.HttpServletResponse;import java.io.OutputStream;/** * 接口防刷拦截器 * @Author: Cyz * @Description: * @Date: create in 2022/6/28 16:58 */@Componentpublic class AntiBrushInterceptor extends HandlerInterceptorAdapter {@Autowiredprivate RedisTemplate redisTemplate;@Overridepublic boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {//判断请求是否属于方法的请求if(handler instanceof HandlerMethod){HandlerMethod handlerMethod=(HandlerMethod) handler;//获取方法中的注解，看是否有该注解AccessLimit accessLimit = handlerMethod.getMethodAnnotation(AccessLimit.class);if(accessLimit==null){return true;}int seconds = accessLimit.seconds();int maxcount = accessLimit.maxCount();boolean login = accessLimit.needLogin();String key = request.getRequestURI();//如果需要登录if(login){//获取登录的session进行判断//.......//key+=""+"1"; //用户id userId}//从redis中获取用户访问的次数//AccessKey ak = AccessKey.withExpire(seconds);Integer count= (Integer)redisTemplate.opsForValue().get(key);if(count==null){//第一次访问redisTemplate.opsForValue().set(key,1);}else if(count<maxcount){//加1redisTemplate.opsForValue().set(key,(Integer)redisTemplate.opsForValue().get(key)+1,seconds, TimeUnit.SECONDS);}else{//超出访问次数System.out.println(key+":访问次数超多！！！");render(response,key+":请您休息片刻再试试！");return false;}}return true;}private void render(HttpServletResponse response, String message)throws Exception {response.setContentType("Application/json;charset=UTF-8");OutputStream out = response.getOutputStream();String data = https://www.isolves.com/it/cxkf/kj/2022-06-29/"{"code":501,"flag":false,"message":"+message+"}";out.write(data.getBytes("UTF-8"));out.flush();out.close();}}将拦截器注册到spring容器中
import org.springframework.beans.factory.annotation.Autowired;import org.springframework.context.annotation.Configuration;import org.springframework.web.servlet.config.annotation.InterceptorRegistry;import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;/** * @Author: Cyz * @Description: * @Date: create in 2022/6/28 17:01 */@Configurationpublic class WebConfig extends WebMvcConfigurerAdapter {@Autowiredprivate AntiBrushInterceptor interceptor;@Overridepublic void addInterceptors(InterceptorRegistry registry) {registry.addInterceptor(interceptor);}}controller接口编写
@AccessLimit(seconds = 5,maxCount = 5,needLogin = true)接口防刷测试

文章插图

参考原文：cyz

Spring Boot接口限制访问次数

推荐阅读

大盘|大盘反弹，不可忽视风险

「天蝎座」6月下旬，横财入宅，3大属相吉运相随，贵人驾临，财富延绵

电竞|LOL不是战士但出肉最恶心的4个英雄，选出来就是磨人心态的

趣读文化|盗墓贼几乎洗劫一空，专家：只留下一个文物，千年古墓被发现

浅浅说科技|这次或要霸占市场！，差强人意还是搅局市场？红米K30再跌新低价

集团|葵花药业：控股股东葵花集团解除质押3700万股

手机中国续航大提升，苹果明年新iPhone将使用增强版5nm芯片

驱动之家魅族旗舰机史上最大电池！魅族17半个小时能充入56%电量

冬天吃水果清热解毒去火注意事项冬天吃什么水果降火效果最好

中国游泳功勋教练徐国义病逝|中国游泳功勋教练徐国义病逝享年50岁

『东方财富网』报54.4元，创科(00669)跌4.73%

周朝灭亡！周朝怎么灭亡的周朝是怎么灭的

GPLP|蓝思科技回应投资人：前员工贪腐不会对生产经营造成重大不利影响

2023两个立春时间 2023立春时间几点几分

安妮的大射手|但为了爱情，最疯狂的三个星座，义无反顾，不计后果，头脑理智

如何学习集合与函数的概念

高个子穿高跟鞋是啥感觉

被告|冒充医生向老年人推销降糖假药四人被控诈骗210万元

孙权杀关羽嫁祸曹操?孙权为什么对关羽赶尽杀绝

结婚五年后，我重新思考了一下婚姻的意义