<!-- Redisson 分布式工具 -->
<dependency>
    <groupId>org.redisson</groupId>
    <artifactId>redisson-spring-boot-starter</artifactId>
    <version>3.23.3</version>
</dependency>

<!-- Spring AOP -->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-aop</artifactId>
</dependency>

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface NoRepeatSubmit {

    /**
     * 設(shè)置請(qǐng)求鎖定時(shí)間(秒)
     */
    int lockTime() default 5;
}

Aspect
@Component
public class RepeatSubmitAspect {

    private static final Logger log = LoggerFactory.getLogger(RepeatSubmitAspect.class);

    @Resource
    private RedissonClient redissonClient;

    @Pointcut("@annotation(com.example.demo.config.NoRepeatSubmit)")
    public void pointCut() {
    }

    @Around("pointCut()")
    public Object around(ProceedingJoinPoint pjp) throws Throwable {
        MethodSignature signature = (MethodSignature) pjp.getSignature();
        Method method = signature.getMethod();
        NoRepeatSubmit annotation = method.getAnnotation(NoRepeatSubmit.class);

        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        if (attributes == null) {
            throw new IllegalArgumentException("無法獲取請(qǐng)求信息");
        }

        HttpServletRequest request = attributes.getRequest();
        String token = request.getHeader("token");
        String path = request.getServletPath();

        if (token == null || token.isEmpty()) {
            throw new IllegalArgumentException("缺少token請(qǐng)求頭");
        }

        // 使用token+path作為鎖的key
        String key = "repeat_submit:" + token + ":" + path;
        RLock lock = redissonClient.getLock(key);

        // 嘗試獲取鎖，等待0秒，自動(dòng)釋放時(shí)間由注解指定
        boolean isSuccess = false;
            isSuccess = lock.tryLock(0, annotation.lockTime(), TimeUnit.SECONDS);
            if (isSuccess) {
                log.info("獲取鎖成功: {}", key);
                // 執(zhí)行目標(biāo)方法
                return pjp.proceed();
            } else {
                log.info("重復(fù)請(qǐng)求，獲取鎖失敗: {}", key);
                return Result.fail("請(qǐng)勿重復(fù)提交請(qǐng)求");
            }
    }
}

@RestController
@RequestMapping("/api/order")
public class OrderController {

    @PostMapping("/create")
    @NoRepeatSubmit(lockTime = 10) // 設(shè)置5秒內(nèi)不允許重復(fù)提交
    public Result createOrder() {
        // 模擬訂單創(chuàng)建過程
        try {
            Thread.sleep(2000); // 模擬業(yè)務(wù)處理耗時(shí)2秒
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        return Result.success("訂單創(chuàng)建成功");
    }
}