封装各种轮询

polling.ts

export default class Polling {
  private timer: number

  private interval: number

  private times: number

  constructor(interval = 5e3, times = -1) {
    this.interval = interval
    this.times = times
    this.timer = 0
  }

  action(callback: () => Promise<any>, count = 0) {
    const isPolling = this.times === -1 || (this.times >= 0 && count < this.times)

    if (isPolling) {
      const start = new Date().getTime()

      callback().finally(() => {
        this.clear()

        if (isPolling) count += 1

        const timeout = this.interval - new Date().getTime() + start

        if (timeout <= 0) {
          this.action(callback, count)
        } else {
          this.timer = window.setTimeout(() => this.action(callback, count), timeout)
        }
      })
    }
  }

  clear() {
    window.clearTimeout(this.timer)
  }
}