您现在的位置是：亿华云 > 系统运维

聊聊 Ahooks 是怎么解决用户多次提交问题？

亿华云2025-10-03 15:47:29【系统运维】6人已围观

简介本文来探索一下 ahooks 的 useLockFn。场景试想一下，有这么一个场景，有一个表单，你可能多次提交，就很可能导致结果不正确。解决这类问题的方法有很多，比如添加 loading，在第一次点击

本文来探索一下 ahooks 的聊聊 useLockFn。

场景

试想一下，解决有这么一个场景，用户有一个表单，多次你可能多次提交，提交就很可能导致结果不正确。问题

解决这类问题的聊聊方法有很多，比如添加 loading，解决在第一次点击之后就无法再次点击。用户另外一种方法就是多次给请求异步函数添加上一个静态锁，防止并发产生。提交这就是问题 ahooks 的 useLockFn 做的事情。

useLockFn

useLockFn 用于给一个异步函数增加竞态锁，聊聊防止并发执行。解决

它的用户源码比较简单，如下所示：

import { useRef, useCallback } from react;

// 用于给一个异步函数增加竞态锁，防止并发执行。

function useLockFn

(fn: (...args: P) => Promise) {

// 是否现在处于一个锁中

const lockRef = useRef(false);

// 返回的是增加了竞态锁的函数

return useCallback(

async (...args: P) => {

// 判断请求是否正在进行

if (lockRef.current) return;

// 请求中

lockRef.current = true;

try {

// 执行原有请求

const ret = await fn(...args);

// 请求完成，状态锁设置为 false

lockRef.current = false;

return ret;

} catch (e) {

// 请求失败，状态锁设置为 false

lockRef.current = false;

throw e;

}

[fn],

);

}

export default useLockFn;

可以看到，它的入参是源码库异步函数，返回的是一个增加了竞态锁的函数。通过 lockRef 做一个标识位，初始化的时候它的值为 false。当正在请求，则设置为 true，从而下次再调用这个函数的时候，就直接 return，不执行原函数，从而达到加锁的目的。

缺点

虽然实用，但缺点很明显，我需要给每一个需要添加竞态锁的请求异步函数都手动加一遍。那有没有比较通用和方便的方法呢？

答案是可以通过 axios 自动取消重复请求。

axios 自动取消重复请求

axios 取消请求

对于原生的 XMLHttpRequest 对象发起的 HTTP 请求，可以调用 XMLHttpRequest 对象的 abort 方法。

那么我们项目中常用的 axios 呢？它其实底层也是用的网站模板 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.); // 取消请求，参数是可选的

另外一种使用的方法是调用 CancelToken 的构造函数来创建 CancelToken，具体使用如下：

const CancelToken = axios.CancelToken;

let cancel;

axios.get(/user/12345, {

cancelToken: new CancelToken(function executor(c) {

cancel = c;

})

});

cancel(); // 取消请求如何自动取消重复的请求

知道了如何取消请求，那怎么做到自动取消呢？答案是通过 axios 的拦截器。

请求拦截器：该类拦截器的作用是在请求发送前统一执行某些操作，比如在请求头中添加 token 相关的字段。响应拦截器：该类拦截器的作用是在接收到服务器响应后统一执行某些操作，比如发现响应状态码为 401 时，自动跳转到登录页。

具体的做法如下：

第一步，定义几个重要的辅助函数。

generateReqKey：用于根据当前请求的信息，生成请求 Key。高防服务器只有 key 相同才会判定为是重复请求。这一点很重要，而且可能跟具体的业务场景有关，比如有一种请求，输入框模糊搜索，用户高频输入关键字，一次性发出多个请求，可能先发出的请求，最后才响应，导致实际搜索结果与预期不符。这种其实就只需要根据 URL 和请求方法判定其为重复请求，然后取消之前的请求就可以了。

这里我认为，如果有需要的话，可以暴露一个 API 给开发者进行自定义重复的规则。这里我们先根据请求方法、url、以及参数生成唯一的 key 去做。

function generateReqKey(config) {

const { method, url, params, data } = config;

return [method, url, Qs.stringify(params), Qs.stringify(data)].join("&");

}addPendingRequest。用于把当前请求信息添加到 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);

}

});