前端并发——基于 Promise.race 实现异步并发控制

题目

转载自该贴的原题是这样：

场景：把fn当作一个请求，现在creator中的2表示限制一次性最多只能发送2次请求，超过了2次就需要等上一个执行完毕才能继续。

试问：现连续执行了5次setup请求，如何在creator中限制它的请求数

给出的空壳代码为：

function creator(count) {
    // fn: () => Promise
    function setup(fn) {

    }
    return setup
}
const setup = creator(2)

setup(fn)
setup(fn)
setup(fn)
setup(fn)
setup(fn)

function fn() {
    return new Promise((resolve) => {
        setTimeout(() => {
            console.log(1)
            resolve(1)
        }, 2000)
    })
}

思路

并发思路大概是这样：

1. 利用 Promise.race 可以处理执行最快的 Promise 对象这个特性，可以初始化一个大小为并发数的执行池用 race 执行。
2. 有执行池就需要替换掉已经执行完成的任务。只要在执行后加一个 then 方法返回当前任务的数组下标就可以进行替换处理。

题目还是比较有意思的，比起常见的并发控制器的封装传参为并发控制数和异步方法数组，这里支持通过多次调用去动态增加需要执行的异步函数。

我们都知道宏任务是在下一次事件循环才会进行执行调用，那么这里我们可以将整个并发放在一个宏任务里边，就支持在同步代码里（应该也可以在微任务里）多次调用setup()去进行并发处理。

这里的思路描述比较简单，具体可以看实现代码的注释去理解~~

实现

为了更好的验证并发控制是否成功，这里我将题目里的打印改成打印 Date() ，具体完整代码如下

function creator(count) {
    const arr = [];
    let flag = true;

    function setup(fn) {
        arr.push(fn);
        // 包一层异步，让同步调用全部进来
        flag && setTimeout(() => {
            // 并发执行容器，初始化长度为 count
            const promises = arr.splice(0, count).map((item, index) => {
                // 返回下标，用于执行容器的替换
                return item().then(() => index);
            })

            // 真正执行过程开始,此时 arr 中为第一轮排不上的异步任务
            arr.reduce((pres, curFn) => {
                return pres.then(() => {
                    return Promise.race(promises); // 返回先完成的下标
                }).then(fastestIndex => {
                    // 要继续将这个下标返回，以便下一次遍历
                    promises[fastestIndex] = curFn().then(() => fastestIndex);
                })
            }, Promise.resolve());
        }, 0) && (flag = false);// flag 控制异步只调用一次
    }
    return setup
}
const setup = creator(2)

setup(fn)
setup(fn)
setup(fn)
setup(fn)
setup(fn)

function fn() {
    return new Promise((resolve) => {
        setTimeout(() => {
            console.log(Date())
            resolve(1)
        }, 2000)
    })
}

执行结果：

从结果来看，也算是把这个题目给解决了！当然，异步并发控制有非常多的场景，有很多不同的、性能更好的解决方法，本文的方法只是作为一个参考。

如果大家有更好的思路或者对并发控制性能上有什么想法的话欢迎评论或者私信交流！👏

这里是 Storm⚡️，一个前端开发学习者