forEach循环中的Javascript API队列

Posted 2021-05-05

目标摘要

我需要在Node.js后端中实现API队列以进行API调用。我需要了解的API速率限制是每2秒1个请求，这是一个硬限制。我正在forEach循环中进行API调用，因为我需要为每个用户执行一个API调用。

我在线上找到了很多有关如何创建队列的文章，但是它们大多涉及向数组添加API调用，因此我不确定在这种情况下如何实现队列。

非常感谢任何帮助，如果有帮助，我可以分享更多代码。

代码

async function refreshStats() {
  try {
    // get list of all fortnite users
    const fnUserList = await Users.find({}, "_id fnUserPlatform"); // my fnUser _id 5cca01ea8f52f40117b2ff51
    fnUserList.forEach(async fnUser => {
      //make  API call.  apiCall is a function I created to make the API call and format the response
      const { lifeStats, statsEqual } = await apiCall(
        fnUser.fnUserPlatform
      );
      //execute other functions with apiCall response
    });
  } catch (err) {
    console.error("error in refreshStats", err);
  }
}

答案

如果我正确理解。您可以利用generator functions并将其与setInterval组合。您可以使一个队列功能以指定的时间间隔将其列队。

创建一个生成器函数基本上使一个apiCall并暂停

    async function* queueGenerator(userList) {
        for (let fnUser of userList) {
            const result = {lifeStats, statsEqual} = await apiCall(fnUser.fnUserPlatform);      
            yield result;
        }
    }

然后在您的方法中创建一个队列，并使用setInterval使项目入队

    async function refreshStats() {
        try {
            // get list of all fortnite users
            let handle;
            const fnUserList = await Users.find({}, "_id fnUserPlatform"); // my fnUser _id 5cca01ea8f52f40117b2ff51
            const queue = queueGenerator(fnUserList);
            const results = [];
            handle = setInterval(async () => {
                const result = await queue.next();
                results.push(result.value);
                if (results.length === users.length) clearInterval(handle);
            }, 2000);

        } catch (err) {
            console.error("error in refreshStats", err);
        }
    }

[还有另一种方法是结合使用setTimeout和Promise。这涉及创建在setTimeOut中解析并有足够延迟的promise。

async function refreshStatsV2() {
  const fnUserList = await Users.find({}, "_id fnUserPlatform");
  const promises = fnUserList.map((fnUser, ix) => (
    new Promise(resolve =>
      setTimeout(async() => {
        const result = {
          lifeStats,
          statsEqual
        } = await apiCall(ix.fnUserPlatform);
        resolve(result);
      }, ix * 2000) // delay every next item 2sec
    )));

  const result = await Promise.all(promises); // wait all
  console.log(result);
}