如何实现 promise.map,限制 promise 并发数

实现一个带有并发限制的Promise.map函数,可以使用async/awaitPromise的特性来管理并发数。

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模拟异步操作。最终,我们得到一个包含处理结果的数组,并打印到控制台上。

相关推荐
@areok@1 分钟前
C++mat传入C#OpencvCSharp的mat
开发语言·c++·opencv·c#
Yvonne爱编码3 分钟前
AJAX入门-AJAX 概念和 axios 使用
前端·javascript·ajax·html·js
小王C语言15 分钟前
【C++进阶】---- map和set的使用
开发语言·c++
在路上`19 分钟前
前端学习之后端java小白(三)-sql外键约束一对多
java·前端·学习
Elnaij29 分钟前
从C++开始的编程生活(8)——内部类、匿名对象、对象拷贝时的编译器优化和内存管理
开发语言·c++
yb0os11 小时前
RPC实战和核心原理学习(一)----基础
java·开发语言·网络·数据结构·学习·计算机·rpc
Pu_Nine_91 小时前
10 分钟上手 ECharts:从“能跑”到“生产级”的完整踩坑之旅
前端·javascript·echarts·css3·html5
liuyao_xianhui1 小时前
内存管理(C/C++)
java·开发语言·c++
饭碗的彼岸one1 小时前
C++设计模式之单例模式
c语言·开发语言·c++·单例模式·设计模式·饿汉模式·懒汉模式
東雪蓮☆2 小时前
从零开始掌握 Web 与 Nginx:入门详解
运维·服务器·前端·nginx