xinyin025/soybean-admin
1
0
Fork 0
mirror of https://github.com/soybeanjs/soybean-admin.git synced 2025-09-27 21:56:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

optimize(packages): Refactor axios request cancellation: replaced deprecated CancelToken with AbortController, added AbortSignal proxy.

Browse Source
This commit is contained in:
auooru 2024-07-06 11:35:45 +08:00
parent 531432d5ff
commit f50e883ec8
1 changed files with 24 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
packages/axios/src/index.ts
View File

@ -1,5 +1,5 @@
import axios, { AxiosError } from 'axios';
import type { AxiosResponse, CancelTokenSource, CreateAxiosDefaults, InternalAxiosRequestConfig } from 'axios';
import type { AxiosResponse, CreateAxiosDefaults, InternalAxiosRequestConfig } from 'axios';
import axiosRetry from 'axios-retry';
import { nanoid } from '@sa/utils';
import { createAxiosConfig, createDefaultOptions, createRetryOptions } from './options';
@ -22,12 +22,28 @@ function createCommonRequest<ResponseData = any>(
const axiosConf = createAxiosConfig(axiosConfig);
const instance = axios.create(axiosConf);
const cancelTokenSourceMap = new Map<string, CancelTokenSource>();
const abortControllerMap = new Map<string, AbortController>();
// config axios retry
const retryOptions = createRetryOptions(axiosConf);
axiosRetry(instance, retryOptions);
function configAbortSignal(requestId: string, config: InternalAxiosRequestConfig) {
const controller = new AbortController();
if (config.signal instanceof AbortSignal) {
// proxy abort signal
const originalSignal = config.signal;
originalSignal.addEventListener!('abort', e => {
controller.abort((e.target as AbortSignal).reason);
});
}
config.signal = controller.signal;
abortControllerMap.set(requestId, controller);
controller.signal.addEventListener('abort', () => {
abortControllerMap.has(requestId) && abortControllerMap.delete(requestId);
});
}
instance.interceptors.request.use(conf => {
const config: InternalAxiosRequestConfig = { ...conf };
@ -35,10 +51,8 @@ function createCommonRequest<ResponseData = any>(
const requestId = nanoid();
config.headers.set(REQUEST_ID_KEY, requestId);
// config cancel token
const cancelTokenSource = axios.CancelToken.source();
config.cancelToken = cancelTokenSource.token;
cancelTokenSourceMap.set(requestId, cancelTokenSource);
// config abort signal
configAbortSignal(requestId, config);
// handle config by hook
const handledConfig = opts.onRequest?.(config) || config;
@ -79,18 +93,16 @@ function createCommonRequest<ResponseData = any>(
);
function cancelRequest(requestId: string) {
const cancelTokenSource = cancelTokenSourceMap.get(requestId);
const cancelTokenSource = abortControllerMap.get(requestId);
if (cancelTokenSource) {
cancelTokenSource.cancel();
cancelTokenSourceMap.delete(requestId);
cancelTokenSource.abort('cancel request');
}
}
function cancelAllRequest() {
cancelTokenSourceMap.forEach(cancelTokenSource => {
cancelTokenSource.cancel();
abortControllerMap.forEach(cancelTokenSource => {
cancelTokenSource.abort('cancel all request');
});
cancelTokenSourceMap.clear();
}
return {