用30行代码封装一个工具,解决Promise的多并发问题

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

背景

提起控制并发,大家应该不陌生,我们可以先来看看多并发,再去聊聊为什么要去控制它。

多并发一般是指多个异步操作同时进行,而运行的环境中资源是有限的,短时间内过多的并发,会对所运行的环境造成很大的压力,比如前端的浏览器,后端的服务器,常见的多并发操作有:

  • 前端的多个接口同时请求
  • 前端多条数据异步处理
  • Nodejs的多个数据操作同时进行
  • Nodejs对多个文件同时进行修改

图片图片

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

图片图片

代码实现

实现思路

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

1、先定好三个坑位

2、让前三个并发进去坑位执行

3、看哪个坑位并发先执行完,就从剩余的并发中拿一个进去补坑

4、一直重复第 3 步,一直到所有并发执行完

图片图片

Promise.all

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

图片图片

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

图片图片

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

期望效果

图片图片

图片图片

实现 limitFn

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

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

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

图片图片

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

  • 将每一个 fetchFn 放进队列里
  • 将坑位里的 fetchFn 先执行

图片图片

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

图片图片

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

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

1、将 activeCount++ ,这时候执行中的并发数 +1

2、将 fetchFn 执行,并把结果 resolve 出去,说明这个并发执行完了

3、将 activeCount--,这时候执行中的并发数 -1

4、从 queue 中取一个并发,拿来补坑执行

图片图片

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

图片图片

完整代码

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,哈哈,但是重要吗?知识嘛,读懂了,它就是你的,到时跟面试官唠嗑的时候,他哪知道是不是真的是你写的!

责任编辑:武晓燕 来源: 前端之神
相关推荐

2024-03-20 09:31:00

图片懒加载性能优化React

2023-12-25 15:28:57

Python工具pywebio

2024-05-20 01:10:00

Promise变量

2019-10-28 08:00:00

Keras神经网络人工智能

2018-07-03 15:20:36

Promise函数借钱

2023-10-23 08:12:34

并发问题有锁和无锁

2023-10-20 08:01:08

2009-11-25 13:33:39

并发

2012-02-02 15:57:09

HibernateJava

2022-09-13 13:49:05

数据库隔离

2023-11-29 08:10:36

javascriptH5游戏

2018-09-30 08:00:15

区块链碳排放气候

2021-08-09 10:24:21

技术分类数学

2020-09-24 11:46:03

Promise

2021-08-10 07:00:01

Redis单线程并发

2009-06-29 09:38:50

JSF标签JSF

2021-09-22 09:43:47

Python 开发编程语言

2011-05-19 14:40:33

SQL Server

2020-07-07 07:47:07

Java无锁技术

2021-10-25 09:50:57

Redis分布式技术
点赞
收藏

51CTO技术栈公众号