基于Redisson实现分布式系统下的接口限流

<!-- 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 {

    /**
     * 设置请求锁定时间(秒)
     */
    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("无法获取请求信息");
        }

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

        if (token == null || token.isEmpty()) {
            throw new IllegalArgumentException("缺少token请求头");
        }

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

        // 尝试获取锁，等待0秒，自动释放时间由注解指定
        boolean isSuccess = false;
            isSuccess = lock.tryLock(0, annotation.lockTime(), TimeUnit.SECONDS);
            if (isSuccess) {
                log.info("获取锁成功: {}", key);
                // 执行目标方法
                return pjp.proceed();
            } else {
                log.info("重复请求，获取锁失败: {}", key);
                return Result.fail("请勿重复提交请求");
            }
    }
}

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

    @PostMapping("/create")
    @NoRepeatSubmit(lockTime = 10) // 设置5秒内不允许重复提交
    public Result createOrder() {
        // 模拟订单创建过程
        try {
            Thread.sleep(2000); // 模拟业务处理耗时2秒
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        return Result.success("订单创建成功");
    }
}