Java中防止數據重復提交超簡單的6種方法
前言
有位朋友，某天突然問東哥：在 Java 中，防止重復提交最簡單的方案是什么？
這句話中包含了兩個關鍵信息，第一：防止重復提交；第二：最簡單。
于是東哥問他，是單機環(huán)境還是分布式環(huán)境？
得到的反饋是單機環(huán)境，那就簡單了，于是東哥就開始裝*了。
話不多說，我們先來復現這個問題。
模擬用戶場景
根據朋友的反饋，大致的場景是這樣的，如下圖所示：
簡化的模擬代碼如下（基于 Spring Boot）：
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
@RequestMapping("/user")
@RestController
public class UserController {
/**
* 被重復請求的方法
*/
@RequestMapping("/add")
public String addUser(String id) {
// 業(yè)務代碼...
System.out.println("添加用戶ID:" + id);
return "執(zhí)行成功！";
}
}
于是東哥就想到：通過前、后端分別攔截的方式來解決數據重復提交的問題。
前端攔截
前端攔截是指通過 HTML 頁面來攔截重復請求，比如在用戶點擊完“提交”按鈕后，我們可以把按鈕設置為不可用或者隱藏狀態(tài)。
執(zhí)行效果如下圖所示：
前端攔截的實現代碼：
<html>
<script>
function subCli(){
// 按鈕設置為不可用
document.getElementById("btn_sub").disabled="disabled";
document.getElementById("dv1").innerText = "按鈕被點擊了~";
}
</script>
<body style="margin-top: 100px;margin-left: 100px;">
<input id="btn_sub" type="button" value=" 提 交 " onclick="subCli()">
<div id="dv1" style="margin-top: 80px;"></div>
</body>
</html>
復制代碼
但前端攔截有一個致命的問題，如果是懂行的程序員或非法用戶可以直接繞過前端頁面，通過模擬請求來重復提交請求，比如充值了 100 元，重復提交了 10 次變成了 1000 元（瞬間發(fā)現了一個致富的好辦法）。
所以除了前端攔截一部分正常的誤操作之外，后端的攔截也是必不可少。
后端攔截
后端攔截的實現思路是在方法執(zhí)行之前，先判斷此業(yè)務是否已經執(zhí)行過，如果執(zhí)行過則不再執(zhí)行，否則就正常執(zhí)行。
我們將請求的業(yè)務 ID 存儲在內存中，并且通過添加互斥鎖來保證多線程下的程序執(zhí)行安全，大體實現思路如下圖所示：
然而，將數據存儲在內存中，最簡單的方法就是使用 HashMap 存儲，或者是使用 Guava Cache 也是同樣的效果，但很顯然 HashMap 可以更快的實現功能，所以我們先來實現一個 HashMap 的防重（防止重復）版本。
1.基礎版——HashMap
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.HashMap;
import java.util.Map;
/**
* 普通 Map 版本
*/
@RequestMapping("/user")
@RestController
public class UserController3 {
// 緩存 ID 集合
private Map<String, Integer> reqCache = new HashMap<>();
@RequestMapping("/add")
public String addUser(String id) {
// 非空判斷(忽略)...
synchronized (this.getClass()) {
// 重復請求判斷
if (reqCache.containsKey(id)) {
// 重復請求
System.out.println("請勿重復提交?。?！" + id);
return "執(zhí)行失敗";
}
// 存儲請求 ID
reqCache.put(id, 1);
}
// 業(yè)務代碼...
System.out.println("添加用戶ID:" + id);
return "執(zhí)行成功！";
}
}
復制代碼
實現效果如下圖所示：
存在的問題：此實現方式有一個致命的問題，因為 HashMap 是無限增長的，因此它會占用越來越多的內存，并且隨著 HashMap 數量的增加查找的速度也會降低，所以我們需要實現一個可以自動“清除”過期數據的實現方案。
2.優(yōu)化版——固定大小的數組
此版本解決了 HashMap 無限增長的問題，它使用數組加下標計數器（reqCacheCounter）的方式，實現了固定數組的循環(huán)存儲。
當數組存儲到最后一位時，將數組的存儲下標設置 0，再從頭開始存儲數據，實現代碼如下：
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.Arrays;
@RequestMapping("/user")
@RestController
public class UserController {
private static String[] reqCache = new String[100]; // 請求 ID 存儲集合
private static Integer reqCacheCounter = 0; // 請求計數器（指示 ID 存儲的位置）
@RequestMapping("/add")
public String addUser(String id) {
// 非空判斷(忽略)...
synchronized (this.getClass()) {
// 重復請求判斷
if (Arrays.asList(reqCache).contains(id)) {
// 重復請求
System.out.println("請勿重復提交?。?！" + id);
return "執(zhí)行失敗";
}
// 記錄請求 ID
if (reqCacheCounter >= reqCache.length) reqCacheCounter = 0; // 重置計數器
reqCache[reqCacheCounter] = id; // 將 ID 保存到緩存
reqCacheCounter++; // 下標往后移一位
}
// 業(yè)務代碼...
System.out.println("添加用戶ID:" + id);
return "執(zhí)行成功！";
}
}
3.擴展版——雙重檢測鎖(DCL)
上一種實現方法將判斷和添加業(yè)務，都放入 synchronized 中進行加鎖操作，這樣顯然性能不是很高，于是我們可以使用單例中著名的 DCL（Double Checked Locking，雙重檢測鎖）來優(yōu)化代碼的執(zhí)行效率，實現代碼如下：
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.Arrays;
@RequestMapping("/user")
@RestController
public class UserController {
private static String[] reqCache = new String[100]; // 請求 ID 存儲集合
private static Integer reqCacheCounter = 0; // 請求計數器（指示 ID 存儲的位置）
@RequestMapping("/add")
public String addUser(String id) {
// 非空判斷(忽略)...
// 重復請求判斷
if (Arrays.asList(reqCache).contains(id)) {
// 重復請求
System.out.println("請勿重復提交?。?！" + id);
return "執(zhí)行失敗";
}
synchronized (this.getClass()) {
// 雙重檢查鎖（DCL,double checked locking）提高程序的執(zhí)行效率
if (Arrays.asList(reqCache).contains(id)) {
// 重復請求
System.out.println("請勿重復提交?。。? + id);
return "執(zhí)行失敗";
}
// 記錄請求 ID
if (reqCacheCounter >= reqCache.length) reqCacheCounter = 0; // 重置計數器
reqCache[reqCacheCounter] = id; // 將 ID 保存到緩存
reqCacheCounter++; // 下標往后移一位
}
// 業(yè)務代碼...
System.out.println("添加用戶ID:" + id);
return "執(zhí)行成功！";
}
}
注意：DCL 適用于重復提交頻繁比較高的業(yè)務場景，對于相反的業(yè)務場景下 DCL 并不適用。
4.完善版——LRUMap
上面的代碼基本已經實現了重復數據的攔截，但顯然不夠簡潔和優(yōu)雅，比如下標計數器的聲明和業(yè)務處理等，但值得慶幸的是 Apache 為我們提供了一個 commons-collections 的框架，里面有一個非常好用的數據結構 LRUMap 可以保存指定數量的固定的數據，并且它會按照 LRU 算法，幫你清除最不常用的數據。
小貼士：LRU 是 Least Recently Used 的縮寫，即最近最少使用，是一種常用的數據淘汰算法，選擇最近最久未使用的數據予以淘汰。
首先，我們先來添加 Apache commons collections 的引用：
<!-- 集合工具類 apache commons collections --> <!-- https://mvnrepository.com/artifact/org.apache.commons/commons-collections4 --> <dependency> <groupId>org.apache.commons</groupId> <artifactId>commons-collections4</artifactId> <version>4.4</version> </dependency>
實現代碼如下：
import org.apache.commons.collections4.map.LRUMap;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
@RequestMapping("/user")
@RestController
public class UserController {
// 最大容量 100 個，根據 LRU 算法淘汰數據的 Map 集合
private LRUMap<String, Integer> reqCache = new LRUMap<>(100);
@RequestMapping("/add")
public String addUser(String id) {
// 非空判斷(忽略)...
synchronized (this.getClass()) {
// 重復請求判斷
if (reqCache.containsKey(id)) {
// 重復請求
System.out.println("請勿重復提交！?。? + id);
return "執(zhí)行失敗";
}
// 存儲請求 ID
reqCache.put(id, 1);
}
// 業(yè)務代碼...
System.out.println("添加用戶ID:" + id);
return "執(zhí)行成功！";
}
}
使用了 LRUMap 之后，代碼顯然簡潔了很多。
5.最終版——封裝
以上都是方法級別的實現方案，然而在實際的業(yè)務中，我們可能有很多的方法都需要防重，那么接下來我們就來封裝一個公共的方法，以供所有類使用：
import org.apache.commons.collections4.map.LRUMap;
/**
* 冪等性判斷
*/
public class IdempotentUtils {
// 根據 LRU(Least Recently Used，最近最少使用)算法淘汰數據的 Map 集合，最大容量 100 個
private static LRUMap<String, Integer> reqCache = new LRUMap<>(100);
/**
* 冪等性判斷
* @return
*/
public static boolean judge(String id, Object lockClass) {
synchronized (lockClass) {
// 重復請求判斷
if (reqCache.containsKey(id)) {
// 重復請求
System.out.println("請勿重復提交！??！" + id);
return false;
}
// 非重復請求，存儲請求 ID
reqCache.put(id, 1);
}
return true;
}
}
調用代碼如下：
import com.example.idempote.util.IdempotentUtils;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
@RequestMapping("/user")
@RestController
public class UserController4 {
@RequestMapping("/add")
public String addUser(String id) {
// 非空判斷(忽略)...
// -------------- 冪等性調用（開始） --------------
if (!IdempotentUtils.judge(id, this.getClass())) {
return "執(zhí)行失敗";
}
// -------------- 冪等性調用（結束） --------------
// 業(yè)務代碼...
System.out.println("添加用戶ID:" + id);
return "執(zhí)行成功！";
}
}
小貼士：一般情況下代碼寫到這里就結束了，但想要更簡潔也是可以實現的，你可以通過自定義注解，將業(yè)務代碼寫到注解中，需要調用的方法只需要寫一行注解就可以防止數據重復提交了，老鐵們可以自行嘗試一下（需要東哥擼一篇的，評論區(qū)留言 666）。
擴展知識——LRUMap 實現原理分析
既然 LRUMap 如此強大，我們就來看看它是如何實現的。
LRUMap 的本質是持有頭結點的環(huán)回雙鏈表結構，它的存儲結構如下：
AbstractLinkedMap.LinkEntry entry;
當調用查詢方法時，會將使用的元素放在雙鏈表 header 的前一個位置，源碼如下：
public V get(Object key, boolean updateToMRU) {
LinkEntry<K, V> entry = this.getEntry(key);
if (entry == null) {
return null;
} else {
if (updateToMRU) {
this.moveToMRU(entry);
}
return entry.getValue();
}
}
protected void moveToMRU(LinkEntry<K, V> entry) {
if (entry.after != this.header) {
++this.modCount;
if (entry.before == null) {
throw new IllegalStateException("Entry.before is null. This should not occur if your keys are immutable, and you have used synchronization properly.");
}
entry.before.after = entry.after;
entry.after.before = entry.before;
entry.after = this.header;
entry.before = this.header.before;
this.header.before.after = entry;
this.header.before = entry;
} else if (entry == this.header) {
throw new IllegalStateException("Can't move header to MRU This should not occur if your keys are immutable, and you have used synchronization properly.");
}
}
如果新增元素時，容量滿了就會移除 header 的后一個元素，添加源碼如下：
protected void addMapping(int hashIndex, int hashCode, K key, V value) {
// 判斷容器是否已滿
if (this.isFull()) {
LinkEntry<K, V> reuse = this.header.after;
boolean removeLRUEntry = false;
if (!this.scanUntilRemovable) {
removeLRUEntry = this.removeLRU(reuse);
} else {
while(reuse != this.header && reuse != null) {
if (this.removeLRU(reuse)) {
removeLRUEntry = true;
break;
}
reuse = reuse.after;
}
if (reuse == null) {
throw new IllegalStateException("Entry.after=null, header.after=" + this.header.after + " header.before=" + this.header.before + " key=" + key + " value=" + value + " size=" + this.size + " maxSize=" + this.maxSize + " This should not occur if your keys are immutable, and you have used synchronization properly.");
}
}
if (removeLRUEntry) {
if (reuse == null) {
throw new IllegalStateException("reuse=null, header.after=" + this.header.after + " header.before=" + this.header.before + " key=" + key + " value=" + value + " size=" + this.size + " maxSize=" + this.maxSize + " This should not occur if your keys are immutable, and you have used synchronization properly.");
}
this.reuseMapping(reuse, hashIndex, hashCode, key, value);
} else {
super.addMapping(hashIndex, hashCode, key, value);
}
} else {
super.addMapping(hashIndex, hashCode, key, value);
}
}
判斷容量的源碼：
public boolean isFull() {
return size >= maxSize;
}
** 容量未滿就直接添加數據：
super.addMapping(hashIndex, hashCode, key, value);
如果容量滿了，就調用 reuseMapping 方法使用 LRU 算法對數據進行清除。
綜合來說：LRUMap 的本質是持有頭結點的環(huán)回雙鏈表結構，當使用元素時，就將該元素放在雙鏈表 header 的前一個位置，在新增元素時，如果容量滿了就會移除 header 的后一個元素。
總結
本文講了防止數據重復提交的 6 種方法，首先是前端的攔截，通過隱藏和設置按鈕的不可用來屏蔽正常操作下的重復提交。但為了避免非正常渠道的重復提交，我們又實現了 5 個版本的后端攔截：HashMap 版、固定數組版、雙重檢測鎖的數組版、LRUMap 版和 LRUMap 的封裝版。
特殊說明：本文所有的內容僅適用于單機環(huán)境下的重復數據攔截
到此這篇關于Java中防止數據重復提交的文章就介紹到這了,更多相關Java防止數據重復提交內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
參考 & 鳴謝
相關文章
這篇文章主要介紹了 Java 本地方法Native Method詳細介紹的相關資料,需要的朋友可以參考下2017-02-02
這篇文章主要介紹了springboot錯誤處理機制實現原理解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-04-04
這篇文章主要介紹了Feign?日期格式轉換錯誤的問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-03-03
這篇文章主要介紹了SpringBoot中Redis的緩存更新策略,緩存一般是為了應對高并發(fā)場景、緩解數據庫讀寫壓力，而將數據存儲在讀寫更快的某種存儲介質中(如內存)，以加快讀取數據的速度,需要的朋友可以參考下2023-08-08
逆向工程是一個專門為 MyBatis 框架使用者設計的代碼生成器,可以根據數據庫中的表字段名,自動生成 POJO 類,mapper 接口與 SQL 映射文件,這篇文章主要給大家介紹了關于mybatis如何根據表逆向自動化生成代碼的相關資料,需要的朋友可以參考下2021-08-08
在面對公司緩存技術混亂時,張三基于Spring框架自研緩存解決方案,展現了他的專業(yè)技術能力和積極工作態(tài)度,他通過問題診斷、技術選型、編碼測試、文檔編寫和部署監(jiān)控等一系列步驟,確保了新緩存系統(tǒng)的功能正確性和性能穩(wěn)定性2024-10-10
這篇文章主要介紹了Java使用Swagger接口框架方法，Swagger是一個方便我們更好的編寫API文檔的框架，而且Swagger可以模擬http請求調用，感興趣的同學可以參考下文2023-05-05
這篇文章主要介紹了MyBatis批量插入幾千條數據為何慎用foreach問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-10-10
這篇文章主要給大家介紹了關于Spring MVC中異常處理的三種方式，文中通過示例代碼介紹的非常詳細，對大家學習或者使用Spring MVC具有一定的參考學習價值，需要的朋友們下面來一起學習學習吧2019-09-09
Arrays類中的方法都是static修飾的靜態(tài)方法,在使用的時候可以直接使用類名進行調用,而"不用"使用對象來調用(注意:是"不用"?而不是?"不能"),這篇文章主要給大家介紹了關于Java中Arrays數組工具類的基本使用,需要的朋友可以參考下2021-12-12