JavaScript 控制 Promise 多并发的实现原理与代码解析

作者:木棉时间:2026-10-31 06:22:00浏览:465

本文详解如何在 JavaScript 中控制 Promise 的多并发执行。通过维护任务队列与活跃计数,实现限制最大并行数的功能。文章从基础思路出发,逐步推导 limitFn 的实现,并解析知名库 p-limit 的核心源码,帮助开发者深入理解异步并发控制的机制。

在讨论并发控制之前,我们先回顾一下什么是多并发场景。

多并发通常指多个异步任务同时运行。由于运行环境的资源有限,短时间内过多的并发请求会给环境带来巨大压力,例如前端浏览器或后端服务器。常见的多并发场景包括:

  • 前端同时发起多个接口请求
  • 前端对多条数据进行异步处理
  • Node.js 中同时进行多个数据操作
  • Node.js 中同时修改多个文件

代码实现

正因为 多并发 会造成压力,所以我们才需要去控制它,降低这个压力~,比如我可以控制最大并发数是 3,这样的话即使有100个并发,我也能保证最多同时并发的最大数量是 3

代码实现

代码实现

实现思路

大致思路就是,假设现在有 9 个并发,我设置最大并发为 3,那么我将会走下面这些步骤:

  • 1、先定好三个坑位
  • 2、让前三个并发进去坑位执行
  • 3、看哪个坑位并发先执行完,就从剩余的并发中拿一个进去补坑
  • 4、一直重复第 3 步,一直到所有并发执行完

代码实现

Promise.all

在进行多并发的时候,我们通常会使用Promise.all,但是Promise.all并不能控制并发,或者说它本来就没这个能力,我们可以看下面的例子

const fetchFn = (delay, index) => {
  return new Promise(resolve => {
    console.log(index)
    setTimeout(() => {
      resolve(index)
    }, delay);
  })
}
const promises = [
  fetchFn(1000, 1),
  fetchFn(1000, 2),
  fetchFn(1000, 3),
  fetchFn(1000, 4),
  fetchFn(1000, 5),
  fetchFn(1000, 6)
]
Promise.all(promises)

最后是同时输出,这说明这几个并发是同时发生的

代码实现

所以我们需要做一些改造,让Promise.all执行 promises 时支持控制并发,但是我们改造的不应该是Promise.all,而是这一个个的fetchFn

期望效果

const limitFn = (limit) => {
  // ...coding
}
// 最大并发数 2
const generator = limitFn(2)
const promises = [
  generator(() => fetchFn(1000, 1)),
  generator(() => fetchFn(1000, 2)),
  generator(() => fetchFn(1000, 3)),
  generator(() => fetchFn(1000, 4)),
  generator(() => fetchFn(1000, 5)),
  generator(() => fetchFn(1000, 6))
]
Promise.all(promises)

代码实现

实现 limitFn

我们需要在函数内部维护两个变量:

  • queue:队列,用来存每一个改造过的并发
  • activeCount:用来记录正在执行的并发数

并声明函数 generator ,这个函数返回一个 Promise,因为 Promise.all 最好是接收一个 Promise 数组

const limitFn = (concurrency) => {
  const queue = [];
  let activeCount = 0;
  const generator = (fn, ...args) =>
    new Promise((resolve) => {
      enqueue(fn, resolve, ...args);
    });
  return generator;
};

接下来我们来实现 enqueue 这个函数做两件事:

  • 将每一个 fetchFn 放进队列里
  • 将坑位里的 fetchFn 先执行
const enqueue = (fn, resolve, ...args) => {
  queue.push(run.bind(null, fn, resolve, ...args));
  if (activeCount < limit && queue.length > 0) {
    queue.shift()();
  }
};

假如我设置最大并发数为 2,那么这一段代码在一开始的时候只会执行 2 次,因为一开始只会有 2 次符合 if 判断,大家可以思考一下为什么~

if (activeCount < limit && queue.length > 0) {
    queue.shift()(); // 这段代码
  }

一开始执行 2 次,说明这时候两个坑位已经各自有一个 fetchFn 在执行了

接下来我们实现 run 函数,这个函数是用来包装 fetch 的,他完成几件事情:

  • 1、将 activeCount++ ,这时候执行中的并发数 +1
  • 2、将 fetchFn 执行,并把结果 resolve 出去,说明这个并发执行完了
  • 3、将 activeCount--,这时候执行中的并发数 -1
  • 4、从 queue 中取一个并发,拿来补坑执行
const run = async (fn, resolve, ...args) => {
  activeCount++;
  const result = (async () => fn(...args))();
  try {
    const res = await result;
    resolve(res);
  } catch { }
  next();
};

其实第 3、4 步,是在 next 函数里面执行的

const next = () => {
  activeCount--;
  if (queue.length > 0) {
    queue.shift()();
  }
};

完整代码

const limitFn = (limit) => {
  const queue = [];
  let activeCount = 0;
  const next = () => {
    activeCount--;
    if (queue.length > 0) {
      queue.shift()();
    }
  };
  const run = async (fn, resolve, ...args) => {
    activeCount++;
    const result = (async () => fn(...args))();
    try {
      const res = await result;
      resolve(res);
    } catch { }
    next();
  };
  const enqueue = (fn, resolve, ...args) => {
    queue.push(run.bind(null, fn, resolve, ...args));
    if (activeCount < limit && queue.length > 0) {
      queue.shift()();
    }
  };
  const generator = (fn, ...args) =>
    new Promise((resolve) => {
      enqueue(fn, resolve, ...args);
    });
  return generator;
};

这不是我写的

其实这是一个很出名的库的源码,就是p-limit,哈哈,但是重要吗?知识嘛,读懂了,它就是你的,到时跟面试官唠嗑的时候,他哪知道是不是真的是你写的~

到此这篇关于JavaScript Promise多并发问题的解决方法详解的文章就介绍到这了,更多相关JavaScript Promise内容请搜索本站以前的文章或继续浏览下面的相关文章希望大家以后多多支持本站!

作者最新文章