使用限制器异步/等待发送请求

Question

我正在尝试限制发送到 API 的请求数量。

我正在使用 Limiter，它的工作方式与我的需要一样，唯一的问题是我找不到将它与 await 一起使用的方法（在呈现我的页面之前我需要所有响应）

有人可以帮帮我吗？

顺便说一句，日志返回一个布尔值。

const RateLimiter = require('limiter').RateLimiter;

const limiter = new RateLimiter(50, 5000)

for (let i = 0; i < arrayOfOrders.length; i++) {
    const response = limiter.removeTokens(1, async (err, remainingRequests) => {
        console.log('request')
        return await CoreServices.load('updateOrder', {
            "OrderNumber": arrayOfOrders[i],
            "WorkFlowID": status
        })
    })
    console.log('response', response)
}

console.log('needs to log after all the request');

这是登录：

response true
response true
response false
needs to log after all the request
request
request
request
...

Answer 1

承诺.removeTokens 会有所帮助，看看这段代码是否有效

const RateLimiter = require('limiter').RateLimiter;

const limiter = new RateLimiter(50, 5000);

const tokenPromise = n => new Promise((resolve, reject) => {
    limiter.removeTokens(n, (err, remainingRequests) => {
        if (err) {
            reject(err);
        } else {
            resolve(remainingRequests);
        }
    });
});
(async() => { // this line required only if this code is top level, otherwise use in an `async function`
    const results = await Promise.all(arrayOfOrders.map(async (order) => {
        await tokenPromise(1);
        console.log('request');
        return CoreServices.load('updateOrder', {
            "OrderNumber": order,
            "WorkFlowID": status
        });
    }));
    console.log('needs to log after all the request');
})(); // this line required only if this code is top level, otherwise use in an `async function`

解释

首先：

const tokenPromise = n => new Promise((resolve, reject) => {
    limiter.removeTokens(n, (err, remainingRequests) => {
        if (err) {
            reject(err);
        } else {
            resolve(remainingRequests);
        }
    });
});

承诺 limiter.removeTokens 在 async/await 中使用 - 在 nodejs 中，您可以使用内置的 promisifier，但是最近我遇到了太多失败的实例 - 所以手动承诺（我编了很多这里的单词！）同样有效

现在代码很简单 - 您可以使用 arrayOfOrders.map 而不是 for 循环来创建一个承诺数组，这些承诺在速率限制允许的范围内并行运行，（速率限制在回调内部完成）

await Promise.all(... 将等到所有 CoreServices.load 都完成（或者一个失败 - 如果需要，您可以使用 await Promise.allSettled(... 代替）

地图回调中的代码标记为async 所以：

await tokenPromise(1);

将等到removeTokens 回调被调用 - 然后是请求

return CoreServices.load

制作

注意，这最初是 return await CoreServices.load，但 await 是多余的，因为异步函数中的 return await somepromise 与 return somepromise 相同 - 所以，也要调整你的代码