Promise？异步并发任务控制器？

前言

欢迎关注同名公众号《熊的猫》，文章会同步更新，也可快速加入前端交流群！

“ 实现一个控制并发数的任务队列 、实现一个异步并发任务控制器” 等，已经是非常经典的手写题目了，因为其中涉及 异步 和 并发 的内容，在正式开始实现之前我们先来简单了解一下它们的概念，毕竟只有知道为什么才能更好的实现，而不是单纯的去记忆。

异步 & 并发

异步

单线程的 JavaScript

我们都知道 默认情况 下 JavaScript 是 单线程 的，又或者说 JavaScript 只在一个线程上运行。

【注意】JavaScript 虽然只在一个线程上运行，但不表示 JavaScript 引擎只有一个线程，实际上，JavaScript 引擎有多个线程，单个脚本只能在一个线程上运行（即 主线程），其他线程都是在后台配合

而 单线程 就意味着，所有任务需要 排队，前一个任务结束，才会执行后一个任务，如果前一个任务耗时很长，后一个任务就不得不一直等着。

JavaScript 异步的产生

如果排队是因为计算量大，CPU 处理不过来，这时候也算合理，但很多时候 CPU 是空闲的，是因为 IO 设备（输入/输出设备）很慢（比如 Ajax 操作从网络读取数据），CPU 不得不等着结果返回，才能继续往下执行。

JavaScript 语言的设计者意识到，这时主线程完全可以不管 IO 设备，挂起处于等待中的任务，先运行排在后面的任务，等到 IO 设备返回了结果，再回过头，把挂起的任务继续执行下去。

在 JavaScript 为了更好的处理异步问题，我们通常都会选择使用 Promise 或 async/await。

并发

早期计算机的 CPU 是 单核的，一个 CPU 在 同一时间 只能执行 一个进程/线程，当系统中有 多个进程/线程 等待执行时，CPU 只能执行完一个再执行下一个。

而所谓的 并发，指在同一时刻只能有一条 进程指令 执行，但多个 进程指令 被快速的 交替执行，那么在宏观上看就是多个进程同时执行的效果，但在微观上并不是同时执行的，只是把时间分成若干段，使多个进程快速交替的执行。

实现异步并发任务控制器

通过上述内容我们已经知道了 异步 和 并发 的基本概念，现在开始具体实现吧！

题目如下：

假设现在要发送多个请求，但要实现并发控制，即可以通过一个 limit 控制并发数，当任务数量超过对应的 limit 限制的并发数时，后续的任务需要延迟到 正在执行中 的任务执行完后 再执行，并且需要支持动态添加 额外的异步任务，同时当 最后一个任务 执行完成，需要执行对应的 callback 函数。

生成任务集合

// 生成用于测试的任务集合

const tasks = new Array(10).fill(0).map((v, i) => {
    return function task() {
        return new Promise((resolve) => {
            setTimeout(() => {
                resolve(i + 1)
            }, i * 1000);
        })
    }
})

方式一：并发控制函数 concurrencyControl

核心思路

• 通过循环执行当前队列头部的任务

• 当前队列头部任务执行完毕

  • 若是最后一个任务，则执行 callback
  • 否则，继续执行 下一个队头任务

// 并发控制函数
function concurrencyControl(tasks, limit, callback) {
    const queue = tasks.slice()  // 当前执行的任务队列
    let count = 0 // 已完成的任务数量

    const runTask = () => {
        while (limit) {
            limit--

            if (queue.length) {
                const task = queue.shift() // 取出当前队头任务

                task().then(() => {
                    limit++
                    count++

                    if (count === tasks.length) { // 最后一个任务
                        callback() // 执行回调函数
                    }else{
                        runTask() // 继续执行下一个任务
                    }
                })
            }
        }
    }

    return runTask
}

// 测试代码
const sendRequest = concurrencyControl(tasks, 3, (taskId) => {
    console.log(`task ${taskId} finish！`)
})

sendRequest()

不同时间的任务：

相同时间的任务：

方式二：并发控制器 ConcurrencyControl

方式一 虽然能够简单的完成自动化的并发控制，但是不支持 动态添加任务 的要求，这就意味着要 保持状态 了，并且如果当前执行的 promise 任务状态为 rejected 时就无法执行完全部的任务，因为 task().then 对应的 onreject 的回调没有被提供，下面我们就可以通过一个 ConcurrencyControl 类来实现。

核心思路

• 将原本使用到的变量，转换成对应的实例属性
• 新增 addTask() 方法用于动态添加任务，并且在其内部自动启动任务执行
• task().then 替换为 task().finally，目的是当对应的 promise 任务为 reject 状态时仍能够执行

class ConcurrencyControl {
    constructor(tasks, limit, callback) {
        this.queue = tasks.slice() // 当前执行的任务队列
        this.tasks = tasks // 原始任务集合
        this.count = 0 // 已完成的任务数量
        this.limit = limit
        this.callback = callback
    }

    runTask() {
        while (this.limit) {
            this.limit--

            if (this.queue.length) {
                const task = this.queue.shift() // 取出队头任务

                task().finally(() => {
                    this.limit++
                    this.count++

                    if (this.count === this.tasks.length) { // 最后一个任务
                        this.callback() // 执行回调函数
                    } else {
                        this.runTask() // 继续执行下一个任务
                    }
                })
            }
        }
    }

    addTask(task) {
        // 同步添加任务
        this.queue.push(task)
        this.tasks.push(task)
        // 当直接调用 addTask 也可直接执行
        this.runTask()
    }
}

// 测试代码
const Control = new ConcurrencyControl(tasks, 3, () => {
    console.log(`task all finish！`)
})

// 执行队列任务
Control.runTask()

// 添加新任务
Control.addTask(function task() {
    return new Promise((resolve) => {
        setTimeout(() => {
            console.log(`task ${Control.tasks.length} finish！`)
            resolve(Control.tasks.length)
        }, Control.tasks.length * 200);
    })
})

方式三：优化 并发控制器 ConcurrencyControl

核心思路

• 优化掉 this.count 计数，通过 this.queue.size 来代替
• 优化掉 this.addTask() 方法中的 this.queue.push(task)，通过 this.tasks 的变化来自动影响 this.queue 队列
• 优化掉 this.limit ++/--，通过 this.queue.size < this.limit 来替换

class ConcurrencyControl {
    constructor(tasks, limit, callback) {
        this.tasks = tasks.slice() // 浅拷贝，避免修改原数据
        this.queue = new Set() // 任务队列
        this.limit = limit // 最大并发数
        this.callback = callback // 回调
    }

    runTask() {
        // 边界判断
        if(this.tasks.length == 0) return
        
        // 当任务队列有剩余，继续添加任务 
        while (this.queue.size < this.limit) {

            const task = this.tasks.shift() // 取出队头任务
            
            this.queue.add(task) // 往队列中添加当前执行的任务

            task()
                .finally(() => {
                    this.queue.delete(task) // 当前任务执行完毕，从队列中删除改任务

                    if (this.queue.size == 0) {
                        this.callback() // 执行回调函数
                    } else {
                        this.runTask() // 继续执行下一个任务
                    }
                })
        }

    }

    addTask(task) {
        // 同步添加任务
        this.tasks.push(task)
        // 当直接调用 addTask 也可直接执行
        this.runTask()
    }
}

// 测试代码
const Control = new ConcurrencyControl(tasks, 3, () => {
    console.log(`task all finish！`)
})

Control.runTask() // 执行队列任务

Control.addTask(function task() { // 添加新任务
    return new Promise((resolve) => {
        setTimeout(() => {
            console.log(`task 9999 finish！`)
            resolve(999)
        }, 100);
    })
})

最后

欢迎关注同名公众号《熊的猫》，文章会同步更新，也可快速加入前端交流群！

以上就是本文的全部内容，通过以上三种实现方式的 逐步优化 最终得到了一个比较合适的结果，当然实现方式并不是只有文中提到的这种，只需要选择自己 最容易理解 的方式即可。