One person walks fast, but a group of people can go further
为了减缓服务器压力,将服务器资源留待给有价值的请求,防止恶意访问,一般的程序都会有接口防刷设置,接下来介绍一种简单灵活的接口防刷操作
主要采用的技术还是拦截+缓存,我们可以通过自定义注解,将需要防刷的接口给标记出来管理,利用缓存统计指定时间区间里,具体的某个ip访问某个接口的频率,如果超过某个阈值,就让他进一会儿小黑屋,到期自动解放
<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>
@Override
public boolean preHandle(HttpServletRequest request ,HttpServletResponse response ,Object handler) throws Exception {
log.info("------------接口防刷拦截器---------------");
if (handler instanceof HandlerMethod) {
HandlerMethod handlerMethod = (HandlerMethod) handler;
AccessLimit accessLimit = handlerMethod.getMethodAnnotation(AccessLimit.class);
// 如果没有该注解,则不在防刷目标里,直接返回
if (accessLimit == null) {
return true;
}
// 获取最大访问次数
int maxAccessCnt = accessLimit.maxAccessCnt();
// 获取ip
String key = getRealIp(request);
// ip+请求的接口路径 => 唯一标识key
key += request.getRequestURI();
//当前访问次数
Object value = redisTemplate.opsForValue().get(key);
if (value == null) {
//第一次访问 设置key保留时长为1s
redisTemplate.opsForValue().set(key, 1, 1L, TimeUnit.SECONDS);
} else {
Integer currentCnt = (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 {
OutputStream out = response.getOutputStream();
out.write("访问次数已达上线!请稍后再访问".getBytes(StandardCharsets.UTF_8));
out.flush();
out.close();
} catch (IOException e) {
e.printStackTrace();
}
return false;
}
}
}
return true;
}
/**
* 返回不携带data的(成功例子)
* @return
*/
@AccessLimit
@GetMapping("/getPaperS")
public ApiResult getPaperInfoSuccess() {
if (log.isInfoEnabled()) {
log.info("收到获取paper信息请求...");
}
//...业务逻辑
if (log.isInfoEnabled()) {
log.info("完成获取paper信息请求,准备返回对象信息");
}
return ApiResultGenerator.success();
}
正常情况下
到底时间段内最大访问次数时
版权说明 : 本文为转载文章, 版权归原作者所有 版权申明
原文链接 : https://www.cnblogs.com/iamamg97/p/15900251.html
内容来源于网络,如有侵权,请联系作者删除!