js Promise併發控制數量的方法
阿新 • • 發佈:2021-08-28
目錄
- 問題
- 背景
- 思路 & 實現
問題
要求寫一個方法控制 Promise 併發數量,如下:
promiseConcurrencyLimit(limit,array,iteratorFn)
limit 是同一時間執行的 promise 數量,array 是引數陣列,iteratorFn 每個 promise 中執行的非同步操作。
背景
開發中需要在多個promise處理完成後執行後置邏輯,通常使用Promise.all:
Primise.all([p1,p2,p3]).the客棧n((res) => ...)
但是有個問題是,因為 promise 建立後會立即執行,也就是說傳入到 promise.all 中的多個 promise 例項,在其建立的時候就已經開始執行了,如果這些例項中執行的非同步操作都是 http 請求,那麼就會在瞬間發出 n 個 http 請求,這樣顯然是不合理的;更合理的方式是:對 Promise.all 中非同步操作的執行數量加以限制,同一時間只允許有 limit 個非同步操作同時執行。
思路 & 實現
在背景中提到,promise 在建立後就會立即執行,所以控制併發的核心在於控制 promise 例項的生成。最開始只生成 limit 個 promise 例項,然後等待這些 promisewww.cppcns.com 狀態變更,只要其中某一個 promise 例項的狀態發生變更,就立即再建立一個 promise 例項...如此迴圈,直到所有的 promise 都被建立並執行。
npm 上有很多庫實現了此功能,個人覺得 tiny-async-pool 這個庫比較好,因為它直接使用了原生的 Promise 實現了此功能,而其他庫大多重新實現了 promise。其核心程式碼如下:
async function asyncPool(poolLimit,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 <www.cppcns.com;= 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迴圈,生成新的promi程式設計客棧se進行補充 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 例項的執行結果
到此這篇關於 Promise併發控制數量的方法的文章就介紹到這了,更多相關js Promise併發控制內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!