实现一个带有并发限制的Promise.map
函数,可以使用async/await
和Promise
的特性来管理并发数。
javascript
function promiseMap(array, mapper, concurrencyLimit) {
return new Promise((resolve, reject) => {
const results = [];
let currentIndex = 0;
let activeCount = 0;
async function processNext() {
if (currentIndex >= array.length) {
resolve(results);
return;
}
const index = currentIndex++;
activeCount++;
try {
const result = await mapper(array[index]);
results[index] = result;
activeCount--;
processNext();
} catch (error) {
reject(error);
}
}
for (let i = 0; i < concurrencyLimit && i < array.length; i++) {
processNext();
}
});
}
这个promiseMap
函数接受三个参数:array
是要遍历的数组,mapper
是用于对每个数组元素进行处理的异步函数,concurrencyLimit
是并发限制数,表示同时执行的异步操作的最大数量。
promiseMap
函数返回一个Promise
对象,当所有的异步操作都完成时,会解析为一个包含所有结果的数组。
在实现中,我们使用currentIndex
来跟踪当前处理的元素索引,activeCount
来跟踪当前活动的异步操作数量。processNext
函数是一个递归函数,用于处理下一个元素。它会检查currentIndex
是否已经超过数组长度,如果是则表示所有元素已经处理完成,调用resolve
解析结果。如果仍有未处理的元素,会调用mapper
对当前元素进行处理,并将结果存储在results
数组中。然后递归调用processNext
处理下一个元素。如果在处理过程中出现错误,会调用reject
拒绝Promise
。
最后,我们使用一个循环来初始化初始的并发操作,调用processNext
函数。在循环中,我们将并发限制数concurrencyLimit
和数组长度的较小值作为循环的次数。
使用示例:
javascript
const array = [1, 2, 3, 4, 5];
const concurrencyLimit = 2;
const mapper = async (item) => {
// 模拟异步操作
await new Promise((resolve) => setTimeout(resolve, 1000));
return item * 2;
};
promiseMap(array, mapper, concurrencyLimit)
.then((results) => {
console.log(results);
})
.catch((error) => {
console.error(error);
});
在上面的示例中,我们传入了一个数组array
,一个异步处理函数mapper
,并发限制数为2。mapper
函数将数组中的每个元素乘以2,并使用setTimeout
模拟异步操作。最终,我们得到一个包含处理结果的数组,并打印到控制台上。