Blame view

src/utils/http/axios/axiosCancel.ts 1.49 KB
陈文彬 authored
1
2
3
4
import type { AxiosRequestConfig, Canceler } from 'axios';
import axios from 'axios';
import { isFunction } from '/@/utils/is';
Vben authored
5
// Used to store the identification and cancellation function of each request
陈文彬 authored
6
7
8
9
10
11
let pendingMap = new Map<string, Canceler>();

export const getPendingUrl = (config: AxiosRequestConfig) => [config.method, config.url].join('&');

export class AxiosCanceler {
  /**
Vben authored
12
   * Add request
陈文彬 authored
13
14
15
16
17
18
19
20
21
   * @param {Object} config
   */
  addPending(config: AxiosRequestConfig) {
    this.removePending(config);
    const url = getPendingUrl(config);
    config.cancelToken =
      config.cancelToken ||
      new axios.CancelToken((cancel) => {
        if (!pendingMap.has(url)) {
Vben authored
22
          // If there is no current request in pending, add it
陈文彬 authored
23
24
25
26
27
28
          pendingMap.set(url, cancel);
        }
      });
  }

  /**
Vben authored
29
   * @description: Clear all pending
陈文彬 authored
30
31
32
33
34
35
36
37
38
   */
  removeAllPending() {
    pendingMap.forEach((cancel) => {
      cancel && isFunction(cancel) && cancel();
    });
    pendingMap.clear();
  }

  /**
Vben authored
39
   * Removal request
陈文彬 authored
40
41
42
43
44
45
   * @param {Object} config
   */
  removePending(config: AxiosRequestConfig) {
    const url = getPendingUrl(config);

    if (pendingMap.has(url)) {
Vben authored
46
47
      // If there is a current request identifier in pending,
      // the current request needs to be cancelled and removed
陈文彬 authored
48
49
50
51
52
53
54
      const cancel = pendingMap.get(url);
      cancel && cancel(url);
      pendingMap.delete(url);
    }
  }

  /**
Vben authored
55
   * @description: reset
陈文彬 authored
56
57
58
59
60
   */
  reset(): void {
    pendingMap = new Map<string, Canceler>();
  }
}