相应的切面
@Slf4j
@Aspect
@Component
public class RepeatSubmitAspect {
@Pointcut("@annotation(noRepeatSubmit)")
public void pointCut(NoRepeatSubmit noRepeatSubmit) {
}
@Around("pointCut(noRepeatSubmit)")
public ResultNoEncryptVo around(ProceedingJoinPoint pjp, NoRepeatSubmit noRepeatSubmit) throws Throwable {
int lockSeconds = noRepeatSubmit.lockTime();
ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
HttpServletRequest request = requestAttributes.getRequest();
Assert.notNull(request, "request can not null");
String token = request.getHeader("CQ-TOKEN");
String path = request.getServletPath();
String key = getKey(token, path);
String s = RedisUtils.get(key);
log.info("Set Redis = [{}]", key);
if (s == null) {
log.info("getRedisKey == null , key = [{}]", key);
// 加入redis缓存
RedisUtils.set(key,"datas");
RedisUtils.expire(key,lockSeconds, TimeUnit.SECONDS);
// 加入redis缓存
pjp.proceed();
return ResultNoEncryptVo.ok("你已提交成功");
} else {
// 缓存不为空，重复提交的请求
log.info("重复提交, key = [{}]", key);
return ResultNoEncryptVo.ok("亲，请不要重复提交哦");
}
}
private String getKey(String token, String path) {
return token + path;
}
}
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface NoRepeatSubmit {
/**
* 设置请求锁定时间
*/
int lockTime() default 3;
}
