自定义注解 防止重复提交

定义注解

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 *  防重复提交注解
 * @author 向振华
 * @date 2018/11/20 15:53
 */
@Target(value = {ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface Submit {

    /**
     * 防重复时间,默认2s
     *
     * @return
     */
    long time() default 2L;

    /**
     * 重复提交提示语
     *
     * @return
     */
    String msg() default "请勿重复提交!";
}

注解处理切面

/**
 * @author 向振华
 * @date 2018/11/20 16:13
 */
@Order(10)//使用order属性，设置该类在spring容器中的加载顺序
@Aspect
@Configuration
public class SubmitAspect {

    @Resource
    private RedisTemplate redisTemplate;

    @Around("execution(* com.mmtvip.collection.controller..*.*(..))")
    public Object getReqAndResInfo(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {
        RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
        ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) requestAttributes;
        HttpServletRequest request = servletRequestAttributes.getRequest();
        // 获取出方法上的@Submit注解
        Method method = ((MethodSignature) proceedingJoinPoint.getSignature()).getMethod();
        Submit submit = method.getAnnotation(Submit.class);
        if (submit != null) {
            //1.生成防重复提交token,准备放入redis
            String token = request.getRequestURI() + request.getSession().getId();
            //2.若该token已经在redis存在,则认为重复提交了
            if (redisTemplate.hasKey(token)) {
                return new ResponseMessage(-1, submit.msg());
            }
            //3.若该token不存在与redis,则认为可以提交,将token存入redis
            redisTemplate.opsForValue().set(token, token, submit.time(), TimeUnit.SECONDS);
        }
        return proceedingJoinPoint.proceed();
    }
}