js Promise並發控制數量的方法

Posted on by

問題

要求寫一個方法控制 Promise 並發數量，如下：

promiseConcurrencyLimit(limit, array, iteratorFn)

limit 是同一時間執行的 promise 數量，array 是參數數組，iteratorFn 每個 promise 中執行的異步操作。

背景

開發中需要在多個promise處理完成後執行後置邏輯，通常使用Promise.all：

Primise.all([p1, p2, p3]).then((res) => ...)

但是有個問題是，因為 promise 創建後會立即執行，也就是說傳入到 promise.all 中的多個 promise 實例，在其創建的時候就已經開始執行瞭，如果這些實例中執行的異步操作都是 http 請求，那麼就會在瞬間發出 n 個 http 請求，這樣顯然是不合理的；更合理的方式是：對 Promise.all 中異步操作的執行數量加以限制，同一時間隻允許有 limit 個異步操作同時執行。

思路 & 實現

在背景中提到，promise 在創建後就會立即執行，所以控制並發的核心在於控制 promise 實例的生成。最開始隻生成 limit 個 promise 實例，然後等待這些 promise 狀態變更，隻要其中某一個 promise 實例的狀態發生變更，就立即再創建一個 promise 實例…如此循環，直到所有的 promise 都被創建並執行。

npm 上有很多庫實現瞭此功能，個人覺得 tiny-async-pool 這個庫比較好，因為它直接使用瞭原生的 Promise 實現瞭此功能，而其他庫大多重新實現瞭 promise。其核心代碼如下：

async function asyncPool(poolLimit, array, iteratorFn) {
  const ret = []; // 用於存放所有的promise實例
  const executing = []; // 用於存放目前正在執行的promise
  for (const item of array) {
    const p = Promise.resolve(iteratorFn(item)); // 防止回調函數返回的不是promise，使用Promise.resolve進行包裹
    ret.push(p);
    if (poolLimit <= array.length) {
      // then回調中，當這個promise狀態變為fulfilled後，將其從正在執行的promise列表executing中刪除
      const e = p.then(() => executing.splice(executing.indexOf(e), 1));
      executing.push(e);
      if (executing.length >= poolLimit) {
        // 一旦正在執行的promise列表數量等於限制數，就使用Promise.race等待某一個promise狀態發生變更，
        // 狀態變更後，就會執行上面then的回調，將該promise從executing中刪除，
        // 然後再進入到下一次for循環，生成新的promise進行補充
        await Promise.race(executing);
      }
    }
  }
  return Promise.all(ret);
}

測試代碼如下：

const timeout = (i) => {
  console.log('開始', i);
  return new Promise((resolve) => setTimeout(() => {
    resolve(i);
    console.log('結束', i);
  }, i));
};

(async () => {
    const res = await asyncPool(2, [1000, 5000, 3000, 2000], timeout);
    console.log(res);
  })();

代碼的核心思路為：

  • 先初始化 limit 個 promise 實例，將它們放到 executing 數組中
  • 使用 Promise.race 等待這 limit 個 promise 實例的執行結果
  • 一旦某一個 promise 的狀態發生變更，就將其從 executing 中刪除，然後再執行循環生成新的 promise，放入executing 中
  • 重復2、3兩個步驟，直到所有的 promise 都被執行完
  • 最後使用 Promise.all 返回所有 promise 實例的執行結果

到此這篇關於js Promise並發控制數量的方法的文章就介紹到這瞭,更多相關js Promise並發控制內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：