聊聊 Ahooks 是怎么解決用戶多次提交問題？
本文來探索一下 ahooks 的 useLockFn。
場景
試想一下，有這么一個(gè)場景，有一個(gè)表單，你可能多次提交，就很可能導(dǎo)致結(jié)果不正確。
解決這類問題的方法有很多，比如添加 loading，在第一次點(diǎn)擊之后就無法再次點(diǎn)擊。另外一種方法就是給請求異步函數(shù)添加上一個(gè)靜態(tài)鎖，防止并發(fā)產(chǎn)生。這就是 ahooks 的 useLockFn 做的事情。
useLockFn
useLockFn 用于給一個(gè)異步函數(shù)增加競態(tài)鎖，防止并發(fā)執(zhí)行。
它的源碼比較簡單，如下所示：
import { useRef, useCallback } from 'react';
// 用于給一個(gè)異步函數(shù)增加競態(tài)鎖，防止并發(fā)執(zhí)行。
function useLockFn<P extends any[] = any[], V extends any = any>(fn: (...args: P) => Promise<V>) {
// 是否現(xiàn)在處于一個(gè)鎖中
const lockRef = useRef(false);
// 返回的是增加了競態(tài)鎖的函數(shù)
return useCallback(
async (...args: P) => {
// 判斷請求是否正在進(jìn)行
if (lockRef.current) return;
// 請求中
lockRef.current = true;
try {
// 執(zhí)行原有請求
const ret = await fn(...args);
// 請求完成，狀態(tài)鎖設(shè)置為 false
lockRef.current = false;
return ret;
} catch (e) {
// 請求失敗，狀態(tài)鎖設(shè)置為 false
lockRef.current = false;
throw e;
}
},
[fn],
);
}
export default useLockFn;
可以看到，它的入?yún)⑹钱惒胶瘮?shù)，返回的是一個(gè)增加了競態(tài)鎖的函數(shù)。通過 lockRef 做一個(gè)標(biāo)識(shí)位，初始化的時(shí)候它的值為 false。當(dāng)正在請求，則設(shè)置為 true，從而下次再調(diào)用這個(gè)函數(shù)的時(shí)候，就直接 return，不執(zhí)行原函數(shù)，從而達(dá)到加鎖的目的。
缺點(diǎn)
雖然實(shí)用，但缺點(diǎn)很明顯，我需要給每一個(gè)需要添加競態(tài)鎖的請求異步函數(shù)都手動(dòng)加一遍。那有沒有比較通用和方便的方法呢？
答案是可以通過 axios 自動(dòng)取消重復(fù)請求。
axios 自動(dòng)取消重復(fù)請求
axios 取消請求
對于原生的 XMLHttpRequest 對象發(fā)起的 HTTP 請求，可以調(diào)用 XMLHttpRequest 對象的 abort 方法。
那么我們項(xiàng)目中常用的 axios 呢？它其實(shí)底層也是用的 XMLHttpRequest 對象，它對外暴露取消請求的 API 是 CancelToken。可以使用如下：
const CancelToken = axios.CancelToken;
const source = CancelToken.source();
axios.post('/user/12345', {
name: 'gopal'
}, {
cancelToken: source.token
})
source.cancel('Operation canceled by the user.'); // 取消請求，參數(shù)是可選的
另外一種使用的方法是調(diào)用 CancelToken 的構(gòu)造函數(shù)來創(chuàng)建 CancelToken，具體使用如下：
const CancelToken = axios.CancelToken;
let cancel;
axios.get('/user/12345', {
cancelToken: new CancelToken(function executor(c) {
cancel = c;
})
});
cancel(); // 取消請求
如何自動(dòng)取消重復(fù)的請求
知道了如何取消請求，那怎么做到自動(dòng)取消呢？答案是通過 axios 的攔截器。
- 請求攔截器：該類攔截器的作用是在請求發(fā)送前統(tǒng)一執(zhí)行某些操作，比如在請求頭中添加 token 相關(guān)的字段。
- 響應(yīng)攔截器：該類攔截器的作用是在接收到服務(wù)器響應(yīng)后統(tǒng)一執(zhí)行某些操作，比如發(fā)現(xiàn)響應(yīng)狀態(tài)碼為 401 時(shí)，自動(dòng)跳轉(zhuǎn)到登錄頁。
具體的做法如下：
第一步，定義幾個(gè)重要的輔助函數(shù)。
- generateReqKey：用于根據(jù)當(dāng)前請求的信息，生成請求 Key。只有 key 相同才會(huì)判定為是重復(fù)請求。這一點(diǎn)很重要，而且可能跟具體的業(yè)務(wù)場景有關(guān)，比如有一種請求，輸入框模糊搜索，用戶高頻輸入關(guān)鍵字，一次性發(fā)出多個(gè)請求，可能先發(fā)出的請求，最后才響應(yīng)，導(dǎo)致實(shí)際搜索結(jié)果與預(yù)期不符。這種其實(shí)就只需要根據(jù) URL 和請求方法判定其為重復(fù)請求，然后取消之前的請求就可以了。
這里我認(rèn)為，如果有需要的話，可以暴露一個(gè) API 給開發(fā)者進(jìn)行自定義重復(fù)的規(guī)則。這里我們先根據(jù)請求方法、url、以及參數(shù)生成唯一的 key 去做。
function generateReqKey(config) {
const { method, url, params, data } = config;
return [method, url, Qs.stringify(params), Qs.stringify(data)].join("&");
}
- addPendingRequest。用于把當(dāng)前請求信息添加到 pendingRequest 對象中。
const pendingRequest = new Map();
function addPendingRequest(config) {
const requestKey = generateReqKey(config);
config.cancelToken = config.cancelToken || new axios.CancelToken((cancel) => {
if (!pendingRequest.has(requestKey)) {
pendingRequest.set(requestKey, cancel);
}
});
}
- removePendingRequest。檢查是否存在重復(fù)請求，若存在則取消已發(fā)的請求。
function removePendingRequest(config) {
const requestKey = generateReqKey(config);
if (pendingRequest.has(requestKey)) {
const cancelToken = pendingRequest.get(requestKey);
cancelToken(requestKey);
pendingRequest.delete(requestKey);
}
}
第二步，添加請求攔截器。
axios.interceptors.request.use(
function (config) {
removePendingRequest(config); // 檢查是否存在重復(fù)請求，若存在則取消已發(fā)的請求
addPendingRequest(config); // 把當(dāng)前請求信息添加到pendingRequest對象中
return config;
},
(error) => {
return Promise.reject(error);
}
);
第二步，添加響應(yīng)攔截器。
axios.interceptors.response.use(
(response) => {
removePendingRequest(response.config); // 從pendingRequest對象中移除請求
return response;
},
(error) => {
removePendingRequest(error.config || {}); // 從pendingRequest對象中移除請求
if (axios.isCancel(error)) {
console.log("已取消的重復(fù)請求：" + error.message);
} else {
// 添加異常處理
}
return Promise.reject(error);
}
);
到這一步，我們就通過 axios 完成了自動(dòng)取消重復(fù)請求的功能。
思考與總結(jié)
雖然可以通過類似 useLockFn 這樣的 hook或方法給請求函數(shù)添加競態(tài)鎖的方式解決重復(fù)請求的問題。但這種還是需要依賴于開發(fā)者的習(xí)慣，如果沒有一些規(guī)則的約束，很難避免問題。
通過 axios 攔截器以及其 CancelToken 功能，我們能夠在攔截器中自動(dòng)將已發(fā)的請求取消，當(dāng)然假如有一些接口就是需要重復(fù)發(fā)送請求，可以考慮加一下白名單功能，讓請求不進(jìn)行取消。
參考
- Axios 如何取消重復(fù)請求？
參考資料
[1]詳情: https://github.com/GpingFeng/hooks。
