乐趣区

关于javascript:Promise并发控制

jiezi

问题

要求写一个办法管制 Promise 并发数量，如下：

promiseConcurrencyLimit(limit, array, iteratorFn)

limit 是同一时间执行的 promise 数量，array 是参数数组，iteratorFn 每个 promise 中执行的异步操作。

背景

开发中须要在多个 promise 解决实现后执行后置逻辑，通常应用 Promise.all：

Primise.all([p1, p2, p3]).then((res) => ...)

然而有个问题是，因为 promise 创立后会立刻执行，也就是说传入到 promise.all 中的多个 promise 实例，在其创立的时候就曾经开始执行了，如果这些实例中执行的异步操作都是 http 申请，那么就会在霎时收回 n 个 http 申请，这样显然是不合理的；更正当的形式是：对 Promise.all 中异步操作的执行数量加以限度，同一时间只容许有 limit 个异步操作同时执行。

思路 & 实现

在背景中提到，promise 在创立后就会立刻执行，所以管制并发的外围在于管制 promise 实例的生成。最开始只生成 limit 个 promise 实例，而后期待这些 promise 状态变更，只有其中某一个 promise 实例的状态产生变更，就立刻再创立一个 promise 实例 … 如此循环，直到所有的 promise 都被创立并执行。

npm 上有很多库实现了此性能，集体感觉 tiny-async-pool 这个库比拟好，因为它间接应用了原生的 Promise 实现了此性能，而其余库大多从新实现了 promise。其外围代码如下：

async function asyncPool(poolLimit, array, iteratorFn) {const ret = []; // 用于寄存所有的 promise 实例
  const executing = []; // 用于寄存目前正在执行的 promise
  for (const item of array) {const p = Promise.resolve(iteratorFn(item)); // 避免回调函数返回的不是 promise，应用 Promise.resolve 进行包裹
    ret.push(p);
    if (poolLimit <= array.length) {
      // then 回调中，当这个 promise 状态变为 fulfilled 后，将其从正在执行的 promise 列表 executing 中删除
      const e = p.then(() => executing.splice(executing.indexOf(e), 1));
      executing.push(e);
      if (executing.length >= poolLimit) {
        // 一旦正在执行的 promise 列表数量等于限度数，就应用 Promise.race 期待某一个 promise 状态产生变更，// 状态变更后，就会执行下面 then 的回调，将该 promise 从 executing 中删除，// 而后再进入到下一次 for 循环，生成新的 promise 进行补充
        await Promise.race(executing);
      }
    }
  }
  return Promise.all(ret);
}

测试代码如下：

const timeout = (i) => {console.log('开始', i);
  return new Promise((resolve) => setTimeout(() => {resolve(i);
    console.log('完结', i);
  }, i));
};

(async () => {const res = await asyncPool(2, [1000, 5000, 3000, 2000], timeout);
    console.log(res);
  })();

代码的外围思路为：

  1. 先初始化 limit 个 promise 实例，将它们放到 executing 数组中
  2. 应用 Promise.race 期待这 limit 个 promise 实例的执行后果
  3. 一旦某一个 promise 的状态产生变更，就将其从 executing 中删除，而后再执行循环生成新的 promise，放入 executing 中
  4. 反复 2、3 两个步骤，直到所有的 promise 都被执行完
  5. 最初应用 Promise.all 返回所有 promise 实例的执行后果
退出移动版