无标题

...

视频测试

录制视频

音频测试

代码测试

js
复制
export class Scheduler {
  private asyncLimit: number; // 并行个数
  private retryLimit: number; // 重试次数

  private asyncCount = 0; // 当前执行中的异步操作
  private queue: Func[] = [];

  constructor(asyncLimit = 5, retryLimit = 1) {
    this.asyncLimit = asyncLimit;
    this.retryLimit = retryLimit;
  }

  public add = async (task: AsyncFunc) => {
    // 注意这里是>=，不是>,比如max = 2，那么当count = 2时，说明已经有2个进行的任务了
    if (this.asyncCount >= this.asyncLimit) {
      // 这一块可以说是是很巧妙的设计，利用await把新的 promise 卡在这里
      // 并把这个promise的 resolve 推到队列的末尾
      // 当调用这个 resolve 的时候就相当于“放行！”
      await new Promise((resolve) => {
        this.queue.push(resolve);
      });
    }

    // 执行到这里时，说明任务马上要执行了，先把当前异步执行的个数 +1
    this.asyncCount++;

    let res: any;

    const errorList: any[] = [];

    let retryCount = this.retryLimit;

    while (retryCount > 0) {
      retryCount--;
      try {
        res = await task();
        break;
      } catch (err) {
        errorList.push(err);
      }
    }

    // 执行到这里时，说明任务完成了，异步执行的个数 -1
    this.asyncCount--;

    if (this.queue.length) {
      const resolve = this.queue.shift();
      resolve && resolve();
    }

    // 使用 res ? res : Promise.reject(errorList) 不行
    // 因为 task 的返回值是可能为空的
    return errorList.length >= this.retryLimit
      ? Promise.reject(errorList)
      : res;
  };
  public clear = () => {
    this.queue = [];
  };
}