SpringBoot+AOP+Redis 防止重复请求提交

本文项目基于以下教程的代码版本： https://javaxbfs.blog.csdn.net/article/details/135224261

代码仓库: springboot一些案例的整合_1: springboot一些案例的整合

1、实现步骤

2.引入依赖

我们需要redis、aop的依赖。

<dependency>
????<groupId>org.springframework.boot</groupId>
????<artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>

<dependency>
????<groupId>org.aspectj</groupId>
????<artifactId>aspectjrt</artifactId>
????<version>1.9.6</version>
</dependency>

<dependency>
????<groupId>org.aspectj</groupId>
????<artifactId>aspectjweaver</artifactId>
????<version>1.9.6</version>
</dependency>

redis的访问参数，默认的就行，不需要特别配置。

3、定义拦截注解

我们最终希望的效果是，你想要哪个方法有防止重复提交的功能，直接加上@RepeatSubmit注解即可。 代码如下：

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public?@interface?RepeatSubmit?{
????/**
?????*?加锁过期时间，默认是5秒
?????*?@return
?????*/
????long?lockTime()?default?5L;
}

这段代码定义了一个Java注解（Annotation）叫做RepeatSubmit。注解是Java提供的一种元数据机制，它可以被用于为代码提供附加的信息，这些信息可以被编译器用于生成代码、生成文档、代码检查等。

下面是对这段代码的详细解释：

• @Target(ElementType.METHOD): 这个注解指定RepeatSubmit只能被用于方法上。ElementType.METHOD表示这个注解只能用于方法。

• @Retention(RetentionPolicy.RUNTIME): 这个注解指定了RepeatSubmit的保留策略是RUNTIME。这意味着在运行时，这个注解仍然可以被读取。

• @Documented: 这个注解表明RepeatSubmit应当被Java文档生成器包含在生成文档中。

• public @interface RepeatSubmit: 定义了一个名为RepeatSubmit的公共注解。

• long lockTime() default 5L: 这是RepeatSubmit注解的一个元素，名为lockTime。这个元素返回一个长整型值，并且有一个默认值5秒（5L表示5秒）。

这个注解是为了防止方法的重复提交，并提供了一个默认的加锁过期时间为5秒。如果一个方法被这个注解标记，那么在特定的加锁过期时间内，这个方法只会被执行一次。

4、实现防止重复提交切面

关于Spring Aop切面，可以看我之前的文章。

NoRepeatSubmitAspect

package?com.it.demo.aspect;

import?cn.hutool.core.util.StrUtil;
import?cn.hutool.crypto.digest.DigestUtil;
import?com.it.demo.annotation.RepeatSubmit;
import?lombok.extern.slf4j.Slf4j;
import?org.aspectj.lang.ProceedingJoinPoint;
import?org.aspectj.lang.annotation.Around;
import?org.aspectj.lang.annotation.Aspect;
import?org.aspectj.lang.annotation.Pointcut;
import?org.springframework.data.redis.core.RedisTemplate;
import?org.springframework.stereotype.Component;
import?org.springframework.web.context.request.RequestContextHolder;
import?org.springframework.web.context.request.ServletRequestAttributes;

import?javax.annotation.Resource;
import?javax.servlet.http.HttpServletRequest;
import?java.util.Objects;
import?java.util.concurrent.TimeUnit;

@Component
@Slf4j
@Aspect
public?class?NoRepeatSubmitAspect?{

????@Resource
????private?RedisTemplate?redisTemplate;

????@Pointcut("@annotation(repeatSubmit)")
????public?void?pointCutNoRepeatSubmitAspect(RepeatSubmit?repeatSubmit)?{

????}

????@Around("pointCutNoRepeatSubmitAspect(repeatSubmit)")
????public?Object?around(ProceedingJoinPoint?joinPoint,RepeatSubmit?repeatSubmit)?throws?Throwable?{
????????String?reqSignature?=?buildReqSignature(joinPoint);
????????//加锁时间
????????long?lockTime?=?repeatSubmit.lockTime();
????????Boolean?res?=?redisTemplate.opsForValue().setIfAbsent(reqSignature,?1,?lockTime,?TimeUnit.SECONDS);
????????if(!res){
????????????throw?new?RuntimeException("重复请求！");
????????}
????????return?joinPoint.proceed();
????}

????/**
?????*?构建请求签名
?????*?@param?joinPoint
?????*?@return
?????*/
????private?String?buildReqSignature(ProceedingJoinPoint?joinPoint)?{
????????ServletRequestAttributes?requestAttributes?=?(ServletRequestAttributes)?RequestContextHolder.getRequestAttributes();
????????if(Objects.isNull(requestAttributes)){
????????????throw?new?RuntimeException("请求参数异常！");
????????}
????????HttpServletRequest?request?=?requestAttributes.getRequest();
????????String?remoteAddr?=request.getRemoteAddr()??+?":"?+??request.getRemoteHost();
????????String?method?=?request.getMethod();
????????String?requestURI?=?request.getRequestURI();
????????StringBuffer?sb?=?new?StringBuffer();
????????for?(Object?arg?:?joinPoint.getArgs())?{
????????????sb.append(arg);
????????}
????????//请求签名(MD5加密)
????????return?DigestUtil.md5Hex(StrUtil.format("{}-{}-{}-{}",remoteAddr,method,requestURI,sb));
????}
}

这段代码是一个切面（Aspect），用于处理重复提交的情况。以下是对代码中的主要部分的解释：

1. 导入包及注解：代码首先导入了一些类，并使用了一些注解，例如 @Component, @Aspect, @Resource, @Slf4j 等。

2. NoRepeatSubmitAspect 类：这是一个切面类，用于实现对重复提交的处理逻辑。

3. 成员变量：代码中使用了 @Resource 注解注入了一个 RedisTemplate 对象，用于操作 Redis。

4. 切入点：使用 @Pointcut 注解定义了一个切入点，这里使用了 @annotation(repeatSubmit) 表达式，表示会拦截所有标注了 @RepeatSubmit 注解的方法。

5. Around 通知：使用 @Around 注解定义了一个环绕通知，在拦截的方法执行前后会执行这里定义的逻辑。

6. around() 方法：在该方法中，首先调用了 buildReqSignature(joinPoint) 方法构建了请求的签名信息，然后根据 RepeatSubmit 注解中的配置，在 Redis 中设置了一个锁，以防止重复提交。如果设置锁失败，则抛出了一个运行时异常，表示重复请求，否则执行被拦截的方法。

7. buildReqSignature() 方法：该方法用于构建请求的签名信息，首先获取了请求相关的信息，然后将这些信息进行拼接，并使用 MD5 加密生成请求签名。

这段代码是一个使用 AOP 实现的防止重复提交的切面，在方法执行前通过生成请求签名并利用 Redis 进行锁定的方式，来避免重复提交。

环绕通知实现逻辑如下图所示

构建 Redis 加锁需要使用的 key，加锁key由指定前缀 + MD5(请求签名)组成，对请求签名进行MD5，一方面减少Key的长度，另一方面保证签名信息中的敏感信息不可见，其实现逻辑如下图所示：

/**
?*?构建请求签名
?*?@param?joinPoint
?*?@return
?*/
private?String?buildReqSignature(ProceedingJoinPoint?joinPoint)?{
????ServletRequestAttributes?requestAttributes?=?(ServletRequestAttributes)?RequestContextHolder.getRequestAttributes();
????if(Objects.isNull(requestAttributes)){
????????throw?new?RuntimeException("请求参数异常！");
????}
????HttpServletRequest?request?=?requestAttributes.getRequest();
????String?remoteAddr?=request.getRemoteAddr()??+?":"?+??request.getRemoteHost();
????String?method?=?request.getMethod();
????String?requestURI?=?request.getRequestURI();
????StringBuffer?sb?=?new?StringBuffer();
????for?(Object?arg?:?joinPoint.getArgs())?{
????????sb.append(arg);
????}
????//请求签名(MD5加密)
????return?DigestUtil.md5Hex(StrUtil.format("{}-{}-{}-{}",remoteAddr,method,requestURI,sb));
}

5、 测试

查询方法加一个@RepeatSubmit

@GetMapping("/list")
@RepeatSubmit
public?List<Device>?list(){
????List<Device>?list?=?deviceService.list();
????System.out.println(list);
????return?deviceService.list();
}

访问localhost:8080/v1/device/list

五秒内再访问，报错：

{
????"code":?400,
????"msg":?"重复请求！",
????"data":?null
}

验证通过。