实现一个注解用于防止数据重复提交

在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切面来防止数据重复提交。可以根据实际需求调整超时时间和请求的唯一标识生成逻辑。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Java.程序猿

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值