JavaScript 限制 Promise 并发数量的实现原理与代码

作者:函数茶馆

时间:2026-10-24

浏览:505

本文详解如何在 JavaScript 中限制 Promise 的并发执行数量。针对 Promise.all 无法控制并发的痛点,通过维护执行队列、利用 Promise.race 监听状态变更及动态生成新任务的方式,实现了类似 tiny-async-pool 的核心逻辑,确保同一时刻仅运行指定数量的异步操作。

文章目录

问题

需要编写一个函数来限制 Promise 的并发执行数量,具体需求如下:


promiseConcurrencyLimit(limit, array, iteratorFn)

其中 limit 参数用于设定同一时刻允许执行的 Promise 最大数量,array 为待处理的任务数组,iteratorFn 则是针对每个任务执行的异步操作函数。

背景

在开发场景中,往往需要在多个 Promise 全部完成后执行后续逻辑,此时通常会使用 Promise.all:


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

然而,Promise.all 存在一个潜在问题:Promise 实例在创建后会立即开始执行。这意味着传入 Promise.all 的多个 Promise 在生成瞬间就会同时运行,如果这些异步操作涉及 HTTP 请求,将导致瞬间发出大量请求,造成资源浪费或不合理负载。更合理的做法是对异步操作的并发数量进行限制,确保同一时刻仅有 limit 个异步操作在运行。

思路与实现

如前所述,Promise 创建即执行,因此控制并发的关键在于控制 Promise 实例的生成时机。核心策略是:初始阶段仅生成 limit 个 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,直到 array 中的所有 Promise 均被执行完毕。
  • 最后,使用 Promise.all 收集并返回所有 Promise 实例的执行结果。

至此,关于 JavaScript 中限制 Promise 并发数量的方法介绍完毕。如需了解更多相关 JS Promise 内容,建议查阅本站往期文章或继续浏览后续相关教程,感谢大家的支持!

实用工具推荐

电子包浆

绿化、压缩、噪点和虚拟水印,自由调节做旧程度。

使用 ›
图片压缩

选择画质和输出格式,预览压缩前后的大小。

使用 ›
相关文章