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

 7个月前     1  

文章目录

下面是使用 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 个任务,并在任务执行完毕后再执行下一个任务。

暂无评论

暂无评论...