使用 Promise 实现一个异步流量控制的函数（限制并发数）

Question

关键词：异步流量控制的函数

下面是使用 Promise 实现异步流量控制的函数的示例：

function asyncFlowControl(tasks, limit) {
  let runningCount = 0; // 当前正在运行的任务数
  let index = 0; // 当前执行的任务索引
  const results = []; // 存储任务的结果

  return new Promise((resolve, reject) => {
    function runTask() {
      if (runningCount >= limit || index >= tasks.length) {
        // 达到并发限制或所有任务已执行完毕，返回结果
        if (results.length === tasks.length) {
          resolve(results);
        }
        return;
      }

      const task = tasks[index];
      const currentIndex = index; // 保存当前任务索引

      index++;
      runningCount++;

      task().then((result) => {
        results[currentIndex] = result; // 存储任务结果
        runningCount--;
        runTask(); // 递归执行下一个任务
      }).catch((error) => {
        reject(error);
      });

      runTask(); // 递归执行下一个任务
    }

    runTask(); // 开始执行任务
  });
}

// 示例用法
function asyncTask(value) {
  return new Promise((resolve) => {
    setTimeout(() => {
      console.log(value);
      resolve(value);
    }, Math.random() * 1000);
  });
}

const tasks = [
  () => asyncTask(1),
  () => asyncTask(2),
  () => asyncTask(3),
  () => asyncTask(4),
  () => asyncTask(5),
];

asyncFlowControl(tasks, 2).then((results) => {
  console.log('All tasks completed:', results);
}).catch((error) => {
  console.error('Error occurred:', error);
});

以上示例中的 asyncFlowControl 函数接受一个任务数组 tasks 和一个并发限制 limit ，它会按照并发限制逐个执行任务，并返回一个 Promise 对象。在示例中，任务数组中的每个任务都是一个返回 Promise 的函数，通过 setTimeout 模拟异步操作。

在执行过程中， asyncFlowControl 函数会维护一个 runningCount 变量来跟踪当前正在运行的任务数，并使用递归的方式执行任务。当达到并发限制或所有任务都已执行完毕时，函数会返回结果。

通过控制并发任务的数量，我们可以限制同时执行的异步操作，实现异步流量控制。在上述示例中，设置并发限制为 2，可以确保最多同时执行 2 个任务，并在任务执行完毕后再执行下一个任务。