在 Spring 中,可以使用 AOP(面向切面编程)来实现接口防刷的功能。下面是一个示例实现:
- 在项目的 pom.xml 文件中加入 AOP 的依赖:
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-aop</artifactId>
</dependency>
- 创建一个切面类,用于处理接口的防刷逻辑,例如:
``` @Aspect @Component public class AntiBrushAspect {
@Autowiredprivate RedisTemplate redisTemplate;
@Pointcut("@annotation(com.example.annotation.AntiBrush)")
public void antiBrushPointcut() {
}
@Around("antiBrushPointcut()")
public Object doAntiBrush(ProceedingJoinPoint joinPoint) throws Throwable {
MethodSignature signature = (MethodSignature) joinPoint.getSignature();
Method method = signature.getMethod();
AntiBrush antiBrush = method.getAnnotation(AntiBrush.class);
int frequency = antiBrush.frequency();
long time = antiBrush.time();
String key = getKey(joinPoint, method);
// 检查请求频率是否超过限制
if (checkFrequency(key, frequency, time)) {
return Result.fail("请求过于频繁");
}
try {
return joinPoint.proceed();
} finally {
// 更新请求频率
updateFrequency(key, frequency, time);
}
}
private String getKey(ProceedingJoinPoint joinPoint, Method method) {
// 获取请求的 IP 地址
String ip = RequestUtil.getIpAddr();
// 获取请求的方法名
String methodName = method.getName();
// 生成 Redis 的键
return String.format("anti_brush:%s:%s", ip, methodName);
}
private boolean checkFrequency(String key, int frequency, long time) {
// 获取当前时间
long now = System.currentTimeMillis();
// 获取 Redis 中的记录
List<Long> records = (List<Long>) redisTemplate.opsForValue().