在Spring Boot中实现一个注解用于防止数据重复提交,可以通过自定义注解、切面编程以及使用缓存来实现。下面是一个完整的实现示例。
首先引入依赖
<dependency>
<groupId>org.aspectj</groupId>
<artifactId>aspectjrt</artifactId>
<version>1.9.7</version> <!-- 请根据需要选择合适的版本 -->
</dependency>
<dependency>
<groupId>org.aspectj</groupId>
<artifactId>aspectjweaver</artifactId>
<version>1.9.7</version> <!-- 请根据需要选择合适的版本 -->
</dependency>
1. 创建自定义注解
首先,定义一个注解 @NoRepeatSubmit
,用于标识需要防止重复提交的方法。
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface NoRepeatSubmit {
String value() default "";
}
2. 实现切面逻辑
接下来,创建一个切面类来处理这个注解。使用一个简单的缓存机制(如ConcurrentHashMap
)来存储已提交的请求标识。
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.stereotype.Component;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
@Aspect
@Component
public class NoRepeatSubmitAspect {
private final Map<String, Long> requestCache = new ConcurrentHashMap<>();
private static final long TIMEOUT = 5000; // 5秒
@Around("@annotation(noRepeatSubmit)")
public Object around(ProceedingJoinPoint joinPoint, NoRepeatSubmit noRepeatSubmit) throws Throwable {
String key = generateKey(joinPoint);
Long lastSubmitTime = requestCache.get(key);
if (lastSubmitTime != null && System.currentTimeMillis() - lastSubmitTime < TIMEOUT) {
throw new RuntimeException("请勿重复提交");
}
requestCache.put(key, System.currentTimeMillis());
return joinPoint.proceed();
}
private String generateKey(ProceedingJoinPoint joinPoint) {
// 可以根据需求生成唯一的key,比如结合方法名和参数
return joinPoint.getSignature().toShortString();
}
}
3. 使用注解
在需要防止重复提交的方法上使用这个注解。
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;
@RestController
public class MyController {
@PostMapping("/submit")
@NoRepeatSubmit
public String submitData() {
// 处理提交逻辑
return "提交成功";
}
}
4. 配置
确保你的Spring Boot应用程序启用了AOP支持。在application.properties
文件中添加以下内容(如果使用的是Spring Boot的默认配置,则不需要额外配置):
spring.aop.proxy-target-class=true
5. 测试
使用Postman或其他工具测试请求。如果在5秒内重复提交请求,将返回“请勿重复提交”的错误信息。
总结
以上代码提供了一个简单的实现,使用自定义注解和AOP切面来防止数据重复提交。可以根据实际需求调整超时时间和请求的唯一标识生成逻辑。