Node.js:防止多个同时调用异步函数

The*_*lus 4 javascript asynchronous node.js

在单线程、同步、非递归代码中,我们可以确定对于任何给定的函数,一次不会有超过一次的调用。

然而,在async/await现实世界中,上述情况不再适用:当我们在执行异步函数 f 期间等待某些内容时,它可能会被再次调用。

我突然想到,使用事件发射器和队列,我们​​可以为异步函数编写一个包装器,以保证它一次不会有多个调用。像这样的东西:

const events = require('events')

function locked(async_fn) {
    const queue = [] // either actively running or waiting to run
    const omega = new events()

    omega.on('foo', () => {
        if (queue.length > 0) {
            queue[0].emit('bar')
        }
    })

    return function(...args) {
        return new Promise((resolve) => {
            const alpha = new events()
            queue.push(alpha)
            alpha.on('bar', async () => {
                resolve(await async_fn(...args))
                queue.shift()
                omega.emit('foo')
            })
            if (queue.length === 1) omega.emit('foo')
        })
    }
}
Run Code Online (Sandbox Code Playgroud)

这个想法是, iff是一个异步函数 thenlocked(f)是一个执行相同操作的函数,除了 iff在 执行期间被调用f,新的调用直到第一个调用返回才开始。

我怀疑我的解决方案还有很大的改进空间,所以我想知道:是否有更好的方法来做到这一点?事实上,Node 中是否已经内置了一个,或者可以通过 npm 获得?


编辑以显示如何使用它:

async function f() {
    console.log('f starts')
    await new Promise(resolve => setTimeout(resolve, 1000))
    console.log('f ends')
}

const g = locked(f)

for (let i = 0; i < 3; i++) {
    g()
}
Run Code Online (Sandbox Code Playgroud)

运行此命令需要 3 秒,我们得到以下输出:

f starts
f ends
f starts
f ends
f starts
f ends
Run Code Online (Sandbox Code Playgroud)

而如果我们在循环中替换g()为,则执行需要 1 秒并得到以下结果:f()for

f starts
f starts
f starts
f ends
f ends
f ends
Run Code Online (Sandbox Code Playgroud)

(我意识到这是一个相当小的问题,如果它不适合 stackoverflow,我表示歉意,但我不知道有更好的地方。)

Hen*_*ker 8

如果您偶然发现这个问题,这里的代码完全符合OP的要求:

const disallowConcurrency = (fn) => {
  let inprogressPromise = Promise.resolve()

  return (...args) => {
    inprogressPromise = inprogressPromise.then(() => fn(...args))
    
    return inprogressPromise
  }
}
Run Code Online (Sandbox Code Playgroud)

像这样使用它:

const someAsyncFunction = async (arg) => {
    await new Promise( res => setTimeout(res, 1000))
    console.log(arg)
}

const syncAsyncFunction = disallowConcurrency(someAsyncFunction)

syncAsyncFunction('I am called 1 second later')
syncAsyncFunction('I am called 2 seconds later')

Run Code Online (Sandbox Code Playgroud)

您可能还想将函数名称更改为更清晰的名称,因为 Promise 实际上与并发性无关。