轮询简单封装 - 2024-02-06

/**

• 轮询封装
• @param fn 轮询function
• @param validate 轮询状态校验规则
• @param interval 轮询间隔
• @param retry 网络错误重试次数
• @returns {poller, stop}
  */
  export function promisePoller(
  fn: (data: { controller?: AbortController }) => Promise<any>,
  validate: (value: any) => boolean,
  interval = 2500,
  retry = 3,
  ): {
  poller: Promise<any>
  stop: (cb?: () => void) => void
  } {
  let timer: any = 0
  // 记录本次请求失败计数
  let failures = 0
  let controller: any
  const clean = () => {
  if (timer) {
  clearTimeout(timer)
  timer = 0
  }
  }

const resolver = async (resolve: (value: any) => void, reject: (e: any) => void) => {
try {
clean()
// fix: 异步接口请求过程中终止操作失效问题
if (controller?.signal.aborted) return
controller = new AbortController()

  const result = await fn({ controller })
  const valid = validate(result)
  // 本次请求成功后, 重置失败计数
  failures = 0
  if (valid === true) {
    clean()
    resolve(result)
  } else if (valid === false) {
    timer = setTimeout(resolver, result.interval || interval, resolve, reject)
  }
} catch (e) {
  // 网络错误尝试retry
  // TODO 失败类型分类判断
  failures += 1
  if (failures <= retry) {
    timer = setTimeout(resolver, interval, resolve, reject)
    return
  }
  reject(e)
}

}

return {
poller: new Promise(resolver),
stop: cb => {
clean()
controller?.abort()
if (cb) cb()
},
}
}