在日常开发过程中,你可能会遇到并发控制的场景,比如控制请求并发数。那么在 JavaScript
中如何实现并发控制呢?在回答这个问题之前,我们来简单介绍一下并发控制。
1、并发控制简介
假设有 6 个待办任务要执行,而我们希望限制同时执行的任务个数,即最多只有 2 个任务能同时执行。当 正在执行任务列表 中的任何 1 个任务完成后,程序会自动从 待办任务列表 中获取新的待办任务并把该该任务添加到 正在执行任务列表 中。为了让大家能够更直观地理解上述的过程,博主特意画了以下 3 张图:
1.1 阶段一
1.1 阶段二
1.1 阶段三
2、并发控制的实现
async-pool
这个库提供了 ES7
和 ES6
两种不同版本的实现,在分析其具体实现之前,我们来看一下它如何使用。
2.1 asyncPool 的使用
const timeout = i => new Promise(resolve => setTimeout(() => resolve(i), i));
await asyncPool(2, [1000, 5000, 3000, 2000], timeout);
在以上代码中,我们使用 async-pool
这个库提供的 asyncPool
函数来实现异步任务的并发控制。asyncPool
函数的签名如下所示:
function asyncPool(poolLimit, array, iteratorFn){ ... }
该函数接收 3 个参数:
poolLimit
(数字类型):表示限制的并发数;
array
(数组类型):表示任务数组;
iteratorFn
(函数类型):表示迭代函数,用于实现对每个任务项进行处理,该函会返回一个 Promise
对象或异步函数。
对于以上示例来说,在使用了 asyncPool 函数之后,对应的执行过程如下所示:
const timeout = i => new Promise(resolve => setTimeout(() => resolve(i), i));
await asyncPool(2, [1000, 5000, 3000, 2000], timeout);
// Call iterator (i = 1000)
// Call iterator (i = 5000)
// Pool limit of 2 reached, wait for the quicker one to complete...
// 1000 finishes
// Call iterator (i = 3000)
// Pool limit of 2 reached, wait for the quicker one to complete...
// 3000 finishes
// Call iterator (i = 2000)
// Itaration is complete, wait until running ones complete...
// 5000 finishes
// 2000 finishes
// Resolves, results are passed in given array order `[1000, 5000, 3000, 2000]`.
2.2 asyncPool ES7 实现
在以上代码中,充分利用了 Promise.all
和 Promise.race
函数特点,再结合 ES7
中提供的 async await
特性,最终实现了并发控制的功能。利用 await Promise.race(executing)
; 这行语句,我们会等待 正在执行任务列表 中较快的任务执行完成之后,才会继续执行下一次循环。
asyncPool ES7
实现相对比较简单,接下来我们来看一下不使用 async await
特性要如何实现同样的功能。
2.3 asyncPool ES6 实现
在 ES6
的实现版本中,通过内部封装的 enqueue
函数来实现核心的控制逻辑。当 Promise.race(executing)
返回的 Promise
对象变成已完成状态时,才会调用enqueue
函数,从 array
数组中获取新的待办任务。
在 asyncPool
这个库的 ES7
和 ES6
的具体实现中,我们都使用到了 Promise.all
和 Promise.race
函数。其中手写 Promise.all
是一道常见的面试题。刚好趁着这个机会,阿宝哥跟大家一起来手写简易版的 Promise.all
和 Promise.race
函数。
简单介绍完毕,需要码起来的可以关注我一下,或者给我留言。