雪忆
2024-07-29 ecfe66a0199606241c73a52519bbe800c9aa31f8
src/utils/http/axios/axiosCancel.ts
@@ -1,60 +1,60 @@
import type { AxiosRequestConfig, Canceler } from 'axios';
import axios from 'axios';
import type { AxiosRequestConfig } from 'axios';
import { isFunction } from '/@/utils/is';
// 用于存储每个请求的标识和取消函数
const pendingMap = new Map<string, AbortController>();
// 声明一个 Map 用于存储每个请求的标识 和 取消函数
let pendingMap = new Map<string, Canceler>();
export const getPendingUrl = (config: AxiosRequestConfig) => [config.method, config.url].join('&');
const getPendingUrl = (config: AxiosRequestConfig): string => {
  return [config.method, config.url].join('&');
};
export class AxiosCanceler {
  /**
   * 添加请求
   * @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) => {
        if (!pendingMap.has(url)) {
          // 如果 pending 中不存在当前请求,则添加进去
          pendingMap.set(url, cancel);
        }
      });
    const controller = new AbortController();
    config.signal = config.signal || controller.signal;
    if (!pendingMap.has(url)) {
      // 如果当前请求不在等待中,将其添加到等待中
      pendingMap.set(url, controller);
    }
  }
  /**
   * @description: 清空所有pending
   * 清除所有等待中的请求
   */
  removeAllPending() {
    pendingMap.forEach((cancel) => {
      cancel && isFunction(cancel) && cancel();
  public removeAllPending(): void {
    pendingMap.forEach((abortController) => {
      if (abortController) {
        abortController.abort();
      }
    });
    pendingMap.clear();
    this.reset();
  }
  /**
   * 移除请求
   * @param {Object} config
   * @param config 请求配置
   */
  removePending(config: AxiosRequestConfig) {
  public removePending(config: AxiosRequestConfig): void {
    const url = getPendingUrl(config);
    if (pendingMap.has(url)) {
      // 如果在 pending 中存在当前请求标识,需要取消当前请求,并且移除
      const cancel = pendingMap.get(url);
      cancel && cancel(url);
      // 如果当前请求在等待中,取消它并将其从等待中移除
      const abortController = pendingMap.get(url);
      if (abortController) {
        abortController.abort(url);
      }
      pendingMap.delete(url);
    }
  }
  /**
   * @description: 重置
   * 重置
   */
  reset(): void {
    pendingMap = new Map<string, Canceler>();
  public reset(): void {
    pendingMap.clear();
  }
}