refactor: axios Canceler use AbortController (#2676)

This commit is contained in:
Kirk Lin 2023-04-06 22:03:12 +08:00 committed by GitHub
parent ae5f5cb13c
commit b14a15e66b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 33 additions and 33 deletions

View File

@ -1,60 +1,60 @@
import type { AxiosRequestConfig, Canceler } from 'axios';
import axios from 'axios';
import { isFunction } from '/@/utils/is';
import type { AxiosRequestConfig } from 'axios';
// Used to store the identification and cancellation function of each request
let pendingMap = new Map<string, Canceler>();
// 用于存储每个请求的标识和取消函数
const pendingMap = new Map<string, AbortController>();
export const getPendingUrl = (config: AxiosRequestConfig) => [config.method, config.url].join('&');
const getPendingUrl = (config: AxiosRequestConfig): string => {
return [config.method, config.url].join('&');
};
export class AxiosCanceler {
/**
* Add request
* @param {Object} config
*
* @param config
*/
addPending(config: AxiosRequestConfig) {
public addPending(config: AxiosRequestConfig): void {
this.removePending(config);
const url = getPendingUrl(config);
config.cancelToken =
config.cancelToken ||
new axios.CancelToken((cancel) => {
const controller = new AbortController();
config.signal = config.signal || controller.signal;
if (!pendingMap.has(url)) {
// If there is no current request in pending, add it
pendingMap.set(url, cancel);
// 如果当前请求不在等待中,将其添加到等待中
pendingMap.set(url, controller);
}
});
}
/**
* @description: Clear all pending
*
*/
removeAllPending() {
pendingMap.forEach((cancel) => {
cancel && isFunction(cancel) && cancel();
public removeAllPending(): void {
pendingMap.forEach((abortController) => {
if (abortController) {
abortController.abort();
}
});
pendingMap.clear();
this.reset();
}
/**
* Removal request
* @param {Object} config
*
* @param config
*/
removePending(config: AxiosRequestConfig) {
public removePending(config: AxiosRequestConfig): void {
const url = getPendingUrl(config);
if (pendingMap.has(url)) {
// If there is a current request identifier in pending,
// the current request needs to be cancelled and removed
const cancel = pendingMap.get(url);
cancel && cancel(url);
// 如果当前请求在等待中,取消它并将其从等待中移除
const abortController = pendingMap.get(url);
if (abortController) {
abortController.abort(url);
}
pendingMap.delete(url);
}
}
/**
* @description: reset
*
*/
reset(): void {
pendingMap = new Map<string, Canceler>();
public reset(): void {
pendingMap.clear();
}
}