JavaScript Promise并发控制策略

答案:通过并发控制限制同时执行的异步任务数,可避免资源耗尽。使用Promise队列或async/await实现任务调度,如手动创建任务池或利用p-limit等库控制最大并发,结合Promise.allSettled统一处理结果与错误,平衡性能与稳定性。

在处理大量异步任务时,直接发起全部请求可能导致资源耗尽或服务端限流。这时候需要对 Promise 的并发数量进行控制,避免系统过载。常见的做法是通过“并发控制”机制,让任务按指定数量同时执行,完成一个再启动下一个,从而平滑地处理大量异步操作。

使用 Promise + 队列实现并发控制

核心思路是维护一个任务队列,限制同时运行的 Promise 数量。当某个任务完成后,自动从队列中取出下一个任务执行。

以下是一个简单的并发控制器实现:

function createPool(tasks, concurrency) {
  let index = 0;
  let completed = 0;
  let resolving = false;
  const results = new Array(tasks.length);

  return new Promise((resolve) => {
    function runNext() {
      if (index < tasks.length && !resolving) {
        const currentIndex = index++;
        const task = tasks[currentIndex]();
        task.then(
          (res) => {
            results[currentIndex] = { status: 'fulfilled', value: res };
          },
          (err) => {
            results[currentIndex] = { status: 'rejected', reason: err };
          }
        ).finally(() => {
          completed++;
          if (completed === tasks.length) {
            resolve(results);
          } else {
            runNext();
          }
        });
        // 控制并发数
        for (let i = 1; i < concurrency && index < tasks.length; i++) {
          runNext();
        }
        resolving = true; // 标记已启动初始批次
      }
    }
    runNext();
  });
}

使用方式:

const tasks = Array(10).fill().map((_, i) => () =>
  fetch(`/api/data/${i}`).then(res => res.json())
);

createPool(tasks, 3).then(results => {
  console.log('所有任务完成:', results);
});

基于 async/await 的简化实现

利用 async 函数和数组的 reduce 或递归调用,可以更清晰地管理流程。

示例:逐个执行但保持最大并发数:

async function asyncPool(poolLimit, array, iteratorFn) {
  const ret = [];
  const executing = [];

  for (const item of array) {
    const p = Promise.resolve().then(() => iteratorFn(item));
    ret.push(p);

    if (poolLimit <= array.length) {
      const e = p.finally(() => {
        executing.splice(executing.indexOf(e), 1);
      });
      executing.push(e);

      if (executing.length >= poolLimit) {
        await Promise.race(executing);
      }
    }
  }

  return Promise.allSettled(ret);
}

调用示例:

asyncPool(3, [1, 2, 3, 4, 5], (num) =>
  fetch(`/api/${num}`).then(res => res.text())
).then(results => {
  console.log('结果:', results);
});

第三方库方案(如 p-limit)

实际项目中,推荐使用成熟工具库简化开发。例如 p-limit 是一个轻量级的并发控制工具。

安装:

npm install p-limit

使用:

import pLimit from 'p-limit';

const limit = pLimit(3); // 最大并发3

const promises = [
  limit(() => fetch('/api/1').then(res => res.json())),
  limit(() => fetch('/api/2').then(res => res.json())),
  limit(() => fetch('/api/3').then(res => res.json()))
];

Promise.allSettled(promises).then(results => {
  console.log('完成:', results);
});

它内部自动管理队列,语法简洁,适合生产环境。

基本上就这些。手动实现有助于理解原理,但在工程化项目中优先考虑稳定性和可维护性,选择像 p-limit 这样的小而美的工具会更高效。关键是根据业务需求平衡速度与稳定性。不复杂但容易忽略的是错误处理和最终状态收集,建议统一使用 Promise.allSettled 获取完整结果。