并发控制中如何获取所有任务的请求结果?

并发控制中如何获取所有任务的请求结果?

并发控制中逐个获取请求结果

在控制并发数的场景中,需要依次得到每个任务执行后的请求结果。本文将解决如何解决这个问题。

创建的并发控制池控制着并发数,但是需要修改代码逻辑以在循环中获取每个请求的结果。

问题根源

100 次循环同步执行,初始产生了 100 个并发任务。由于并发数控制为 5,只有前 5 个任务会被处理。其他 95 个任务的 promise 被丢弃,导致无法得到这些任务的结果。

解决方案

可以使用一个 map 来缓存每个 promise 的 resolve 和 reject 函数,确保所有任务的请求结果都可以被获取。

修改后的代码

const createPool = (task, { concurrency } = {}) => {
  let runningCount = 0;
  const pool = [];
  const promiseMap = new Map();

  return function (i) {
    return new Promise((resolve, reject) => {
      promiseMap.set(i, { resolve, reject });

      pool.push(() => task(i));

      function run() {
        while (pool.length && runningCount < concurrency) {
          const runTask = pool.shift();
          runningCount++;
          runTask()
            .then((val) => {
              const { resolve } = promiseMap.get(val);
              resolve(val);
            })
            .catch((e) => reject(e))
            .finally(() => {
              runningCount--;
              run();
            });
        }
      }
      run();
    });
  };
};

在修改后的代码中,promise 的 resolve 和 reject 函数在 map 中进行缓存。当任务执行完成后,可以从 map 中获取相应的函数并将其调用,从而得到请求结果。

使用该并发控制池,可以通过监听 promise 来逐个获取每个请求的结果,从而解决最初的问题。

以上就是并发控制中如何获取所有任务的请求结果?的详细内容,更多请关注硕下网其它相关文章!