拦截器
@Aspect @Component public class NoRepeatAspect { @Resource private RedisTemplate<String, Object> redisTemplate; @Pointcut("@annotation(com.sunline.project.aop.NoRepeat) || @within(com.sunline.project.aop.NoRepeat)") public void pointCut(){} @Around("pointCut()") private Object around(ProceedingJoinPoint point) { try { // 获取当前用户的token Subject currStaff = SecurityUtils.getSubject(); UserVo user = (UserVo) currStaff.getPrincipal(); String token = user.getToken(); StaticLog.info("检测是否重复提交"); StaticLog.info("point:{}", point); // 获取当前请求的方法名 MethodSignature signature = (MethodSignature) point.getSignature(); Method method = signature.getMethod(); String name = method.getName(); StaticLog.info("token:{},======methodName:{}", token, name); if (redisTemplate.hasKey(token+name)) { StaticLog.error("监测到重复提交>>>>>>>>>>"); return ResponseData.fail(ResponseCode.FAIL_CODE, "请勿重复提交"); } // 获取注解 NoRepeat annotation = method.getAnnotation(NoRepeat.class); Long timeout = annotation.timeOut(); // 此处我用token和请求方法名为key存入redis中,有效期为timeout 时间, 也可以使用ip地址做为key redisTemplate.opsForValue().set(token+name, token+name, timeout, TimeUnit.SECONDS); return point.proceed(); } catch (Throwable throwable) { StaticLog.error("=====>>>>>>操作失败:{}", throwable); return ResponseData.fail(ResponseCode.FAIL_CODE, "操作失败"); } } }
注解
@Target(ElementType.METHOD) @Retention(RetentionPolicy.RUNTIME) @Documented public @interface NoRepeat { // 默认失效时间 long timeOut() default 10; }
测试:@NoRepeat(timeOut = 5)
timeOut默认时间为10s
@NoRepeat(timeOut = 5) @GetMapping(value = "/test") @ApiOperation(value = "测试幂等注解") @SysLogs("测试幂等注解") @ApiImplicitParam(paramType = "header", name = "Authorization", value = "身份认证Token") public ResponseData<T> testIdempotent() { try { System.err.println(new Date()); return ResponseData.ok(ResponseCode.SUCCESS_CODE, "操作成功"); } catch (Exception e) { StaticLog.error("操作失败:{}", e); return ResponseData.ok(ResponseCode.FAIL_CODE, "操作失败"); } }