通过Redis实现防止接口重复提交功能

本功能是在切面执行链基础上实现的功能，如果不知道切面执行链的同学，请看一下我之前专门介绍切面执行链的文章。

在SpringBoot项目中实现切面执行链功能-CSDN博客

1.定义防重复提交handler

/*** 重复提交handler**/
@AspectHandlerOrder
public class ResubmitAspectHandler implements AspectHandler {private StringRedisTemplate stringRedisTemplate;public void setStringRedisTemplate(StringRedisTemplate stringRedisTemplate) {this.stringRedisTemplate = stringRedisTemplate;}@Overridepublic boolean execute(ProceedingJoinPoint pjp) throws Exception {Method method = getMethod(pjp);if (!method.isAnnotationPresent(Resubmit.class)) {return true;}Resubmit annotation = method.getAnnotation(Resubmit.class);long ttl = annotation.ttl();String key = getKey();String value = "1";if (lock(key, value, ttl)) {return true;} throw new BaseRuntimeException(ExceptionEnums.ERROR_10012.getCode(), "操作频率过高，请稍后再试！");}@Overridepublic void afterCompletion(ProceedingJoinPoint pjp, Object response, Exception exception) {Method method = getMethod(pjp);if (method.isAnnotationPresent(Resubmit.class)) {unlock(getKey());}}/*** redis原子操作：如果key不存在就设置key:value** @param key* @param value* @return true：设置成功拿到锁，false：设置失败未拿到锁*/private boolean lock(final String key, final String value, final long ttl) {Boolean result = stringRedisTemplate.boundValueOps(key).setIfAbsent(value, Duration.ofSeconds(ttl));return result != null ? result : false;}/*** 解锁：删除key** @param key*/private void unlock(String key) {if (StringUtils.isNotBlank(key)) {stringRedisTemplate.delete(key);}}/*** 获取方法** @param pjp* @return*/private Method getMethod(ProceedingJoinPoint pjp) {MethodSignature signature = (MethodSignature) pjp.getSignature();Method method = signature.getMethod();return method;}/*** 获取key** @return*/private String getKey() {ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();HttpServletRequest request = requestAttributes.getRequest();String url = request.getRequestURI();String httpMethod = request.getMethod();HttpHeader httpHeader = WebContext.getHttpHeader();String deviceId = httpHeader.getDevice_id();String key = RedisConstants.REDIS_RESUBMIT_KEY + httpMethod + url + ":" + deviceId;return key;}
}

2.定义防重复提交注解

/*** 防止重复提交*/
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface Resubmit {/*** 存活时间（秒），当意外情况（例如锁定之后重启服务）* 未能执行解锁功能，redis将在${ttl}秒之后自动删除锁标志* 默认 10秒* @return*/long ttl() default 10;}

3.在配置类中注入防重复提交切面类

@Bean
public List<AspectHandler> apiAspectHandlers() {ResubmitAspectHandler resubmitAspectHandler = new ResubmitAspectHandler();resubmitAspectHandler.setStringRedisTemplate(stringRedisTemplate);return Arrays.asList(resubmitAspectHandler);
}

4.controller中应用防重复提交注解

@PostMapping("/release")
@Resubmit
public ApiResponse<?> insert(@RequestBody @Valid InsertAppRequestDTO req) {// 处理业务逻辑
}