首页 > 网络编程 > JavaScript > JS Promise异步并发控制

JavaScript Promise实现异步并发任务控制器

2023-06-06 11:18:18 作者：熊的猫

这篇文章主要为大家介绍了JavaScript 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);
    })
})

最后

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

以上就是JavaScript Promise实现异步并发任务控制器的详细内容，更多关于JS Promise异步并发控制的资料请关注脚本之家其它相关文章！

JavaScript Promise实现异步并发任务控制器

前言

异步 & 并发

异步

单线程的 JavaScript

JavaScript 异步的产生

并发

实现异步并发任务控制器

方式一：并发控制函数 concurrencyControl

核心思路

方式二：并发控制器 ConcurrencyControl

核心思路

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

核心思路

最后

您可能感兴趣的文章:

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