重複提交，你是如何處理的？

Java旅途發表於2020-06-24

今天早上，新來的同事小王突然問我：“周哥，什麼是冪等性啊？”。然後我就跟他解釋了一番，冪等性就是說無論你執行幾次請求，其結果是一樣的。說到了冪等就不得不說重複提交了，你連續點選提交按鈕，理論上來說這是同一條資料，資料庫應該只能存入一條，而實際上存放了多條，這就違反了冪等性。因此我們就需要做一些處理，來保證連續點選提交按鈕後，資料庫只能存入一條資料。

防止重複提交的方式很多，這裡我就說一下我認為比較好用的一種。

自定義註解+Aop實現

我們通過獲取使用者ip及訪問的介面來判斷他是否重複提交，假如這個ip在一段時間內容多次訪問這個介面，我們則認為是重複提交，我們將重複提交的請求直接處理即可，不讓訪問目標介面。

自定義註解

@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface NoRepeatSubmit {

    /**
     * 預設1s鍾以內算重複提交
     * @return
     */
    long timeout() default 1;
}

Aop處理邏輯

我們將ip+介面地址作為key，隨機生成UUID作為value，存入redis。每次請求進來，根據key查詢redis，如果存在則說明是重複提交，丟擲異常，如果不存在，則是正常提交，將key存入redis。

@Aspect
@Component
public class NoRepeatSubmitAop {

	@Autowired
	private RedisService redisUtils;

	/**
	 * 	定義切入點
	 */
	@Pointcut("@annotation(NoRepeatSubmit)")
	public void noRepeat() {}

	/**
	 * 	前置通知：在連線點之前執行的通知
	 * @param point
	 * @throws Throwable
	 */
	@Before("noRepeat()")
	public void before(JoinPoint point) throws Exception{
		// 接收到請求，記錄請求內容
		ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
		HttpServletRequest request = attributes.getRequest();
		Assert.notNull(request, "request can not null");

		// 此處可以用token或者JSessionId
		String token = IpUtils.getIpAddr(request);
		String path = request.getServletPath();
		String key = getKey(token, path);
		String clientId = getClientId();
		List<Object> lGet = redisUtils.lGet(key, 0, -1);
		// 獲取註解
		MethodSignature signature = (MethodSignature) point.getSignature();
		Method method = signature.getMethod();
		NoRepeatSubmit annotation = method.getAnnotation(NoRepeatSubmit.class);
		long timeout = annotation.timeout();
		boolean isSuccess = false;
		if (lGet.size()==0 || lGet == null) {
			isSuccess = redisUtils.lSet(key, clientId, timeout);
		}
		if (!isSuccess) {
			// 獲取鎖失敗，認為是重複提交的請求
			redisUtils.lSet(key, clientId, timeout);
			throw new Exception("不可以重複提交");
		}

	}

	private String getKey(String token, String path) {
		return token + path;
	}

	private String getClientId() {
		return UUID.randomUUID().toString();
	}
}

提供介面用來測試

在介面上新增上我們自定義的註解@NoRepeatSubmit

@RequestMapping("/test")
@NoRepeatSubmit
public String tt(HttpServletRequest request) {

    return "1";
}

測試

我們在瀏覽器中連續請求兩次介面。發現第一次介面響應正常內容：1，第二次介面響應了不可重複提交的異常資訊。1s之後再點選介面，發現又響應了正常內容。

至此，這種防止重複提交的方式就介紹完了，這樣我們就完美防止了介面重複提交。

相關文章