解决思路:
拦截所有接口进行统一处理
技术上主要以下两个
- ConcurrentHashMap 线程安全
- ExpiringMap 可以设置过期时间的map并且线程安全
ConcurrentHashMap 的Key部分存储接口名,Value部分存储ExpiringMap
ExpiringMap 的Key可以存储session、uid、ip等唯一值(这里可以和前端沟通设置值),value进行设置在规定时间内可访问次数
每次访问会将访问次数+1,且过期时间重新刷新,超出阈值直接拦截拒绝。
<!-- Map依赖 -->
<dependency>
<groupId>net.jodah</groupId>
<artifactId>expiringmap</artifactId>
<version>0.5.8</version>
</dependency>
// 全局配置(JDK8以上能这么用,低版本就把这些常量替换即可)
public interface URLInterceptor {
Integer EXPIRING_TIME = 2;
TimeUnit TIMEUNIT = TimeUnit.SECONDS;
Integer TIMES = 2;
}
代码如下:
public class ExpiringTest {
@org.testng.annotations.Test
void d() {
mockService("localhost/login","192.168.1.1");
}
static ConcurrentHashMap<String, ExpiringMap<String, Integer>> cm = new ConcurrentHashMap<>();
public static void mockService(String url,String ip){
ExpiringMap<String, Integer> em = cm.get(url);
if(Objects.isNull(em)){
System.out.println("【FIND】A new URL has been found and has been replaced.\r\n --"+url);
cm.put(url,em = ExpiringMap.builder().expiration(URLInterceptor.EXPIRING_TIME,URLInterceptor.TIMEUNIT).build());
}
Integer get = em.getOrDefault(ip, -1);
if(get>=URLInterceptor.TIMES){
System.out.println("【ERROR】 \r\n --请勿多次点击!");
}else{
em.put(ip,++get);
System.out.println("读取成功"+get);
}
}
}