1
// limiteNum: 并发数目
2
// tasks: 需要处理的任务列表
3
// iteraterFn: 完成的处理函数
4

5
// 并发控制任务
6
const ayncPool = async (limiteNum, tasks, iteraterFn) => {
7
  // 所有的任务队列
8
  const rect = [];
9
  // 正在进行的任务队列
10
  const pendingTasks = [];
11

12
  // 这里使用for of而不是用foreach的原因是
13
  // forEach await 时候并不会停止遍历
14
  for (const task of tasks) {
15
    // 这里对外层的结束处理函数包裹一层promise
16
    let promiseTask = Promise.resolve().then(() => iteraterFn(task, tasks));
17
    // rect去存储了promiseTask
18
    rect.push(promiseTask);
19

20
    if (limiteNum <= tasks.length) {
21
      const excutingTask = promiseTask.then(() => pendingTasks.splice(pendingTasks.indexOf(excutingTask), 1))
22
      pendingTasks.push(excutingTask)
23
      if (pendingTasks.length >= limiteNum) {
24
        // 这里去获取跑的最快的一个task
25
        await Promise.race(pendingTasks);
26
      }
27
    }
28
  }
29

30
  return Promise.all(rect)
31
}
32

33
const iterater = (i) => {
34
  return new Promise((resolve) => {
35
    setTimeout(() => {
36
      resolve(i);
37
      console.log(i);
38
    }, i);
39
  })
40
}
41

42
ayncPool(2, [1000, 5000, 3000, 2000], iterater).then((data) => {
43
  console.log(data)
44
})