问题
要求写一个办法管制 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); })();
代码的外围思路为:
- 先初始化
limit
个 promise 实例,将它们放到executing
数组中 - 应用
Promise.race
期待这limit
个 promise 实例的执行后果 - 一旦某一个 promise 的状态产生变更,就将其从
executing
中删除,而后再执行循环生成新的 promise,放入executing
中 - 反复2、3两个步骤,直到所有的 promise 都被执行完
- 最初应用
Promise.all
返回所有 promise 实例的执行后果