升級攔截器

上一篇文章留下2個問題

1.某些情況下request獲取不到，或造成異常情況，為了處理這種情況我將獲取Request的方法進行了升級

2.能不能在異常發生的時候就將重複提交標記就移除呢？當然可以！通過@AfterThrowing即可實現

下面是改造後的攔截器程式碼：

/** * @description 防止表單重複提交攔截器 */
@Aspect
@Component
@Slf4j
public class DuplicateSubmitAspect {
    public static final String  DUPLICATE_TOKEN_KEY="duplicate_token_key"
 
;
@Pointcut("execution(public * cn.test.core.controller..*(..))")

    public void webLog() {
    }

    @Before("webLog() && @annotation(token)")
    public void before(final JoinPoint joinPoint, DuplicateSubmitToken token){
        if (token!=null){
            ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes
 
();
HttpServletRequest request = attributes.getRequest();
            boolean isSaveSession=token.save();
            if (isSaveSession){
                String key = getDuplicateTokenKey(joinPoint);
Object t = request.getSession().getAttribute(key);
                if (null==t){
                    String uuid= UUID.randomUUID
 
().toString();
request.getSession().setAttribute(key.toString(),uuid);
log.info("token-key="+key);
log.info("token-value="+uuid.toString());
}else {
                    throw new DuplicateSubmitException(TextConstants.REQUEST_REPEAT);
}
            }

        }
    }

    /**
     * 獲取重複提交key
     * @param joinPoint
* @return
*/
public String getDuplicateTokenKey(JoinPoint joinPoint) {
        String methodName = joinPoint.getSignature().getName();
StringBuilder key=new StringBuilder(DUPLICATE_TOKEN_KEY);
key.append(",").append(methodName);
        return key.toString();
}

    @AfterReturning("webLog() && @annotation(token)")
    public void doAfterReturning(JoinPoint joinPoint,DuplicateSubmitToken token) {
        // 處理完請求，返回內容
log.info("出方法：");
        if (token!=null){
            ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
HttpServletRequest request = attributes.getRequest();
            boolean isSaveSession=token.save();
            if (isSaveSession){
                String key = getDuplicateTokenKey(joinPoint);
Object t = request.getSession().getAttribute(key);
                if (null!=t){
                    request.getSession(false).removeAttribute(key);
}
            }
        }
    }

    /**
     * 異常
* @param joinPoint
* @param e
*/
@AfterThrowing(pointcut = "webLog()&& @annotation(token)", throwing = "e")
    public void doAfterThrowing(JoinPoint joinPoint, Throwable e, DuplicateSubmitToken token) {
        if (null!=token
                && e instanceof DuplicateSubmitException==false){
            //處理處理重複提交本身之外的異常
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
HttpServletRequest request = attributes.getRequest();
            boolean isSaveSession=token.save();
//獲得方法名稱
if (isSaveSession){
                String key=getDuplicateTokenKey(joinPoint);
Object t = request.getSession().getAttribute(key);
                if (null!=t){
                    //方法執行完畢移除請求重複標記
request.getSession(false).removeAttribute(key);
log.info("異常情況--移除標記！");
}
            }
        }
    }
}

3.使用：在你要使用的控制器要防止重複提交的方法添加註解@DuplicateSubmitToken即可

4.總結：有時候我們在思考問題的時候可以換一種思路，多考慮下是否還有其他方式。

如果你覺得幫助到了你，用你的愛心點個贊吧！謝謝！

也可以加入我們的qq群220282789，只為相互幫助，只為一起進步！

作者：天空藍藍的     
版權所有，歡迎保留原文連結進行轉載：)