Promise.all併發限制

背景

一般，咱們在須要保證代碼在多個異步處理以後執行，會用到：

Promise.all(promises: []).then(fun: function);

Promise.all能夠保證，promises數組中全部promise對象都達到resolve狀態，才執行then回調。

這時候考慮一個場景：若是你的promises數組中每一個對象都是http請求，或者說每一個對象包含了複雜的調用處理。而這樣的對象有幾十萬個。

那麼會出現的狀況是，你在瞬間發出幾十萬http請求（tcp鏈接數不足可能形成等待），或者堆積了無數調用棧致使內存溢出。

這時候，咱們就須要考慮對Promise.all作併發限制。

Promise.all併發限制指的是，每一個時刻併發執行的promise數量是固定的，最終的執行結果仍是保持與原來的Promise.all一致。

實現

咱們知道，promise並非由於調用Promise.all才執行，而是在實例化promise對象的時候就執行了，在理解這一點的基礎上，要實現併發限制，只能從promise實例化上下手。

換句話說，就是把生成promises數組的控制權，交給併發控制邏輯。

這裏我並不打算一步步實現這個這個功能，npm中有不少實現這個功能的第三方包，好比async-pool、es6-promise-pool、p-limit，這裏我直接拿async-pool的代碼來分析一下實現原理。

代碼很簡單，去掉沒必要要的代碼，加上本身的註釋，大概內容以下：

function asyncPool(poolLimit, array, iteratorFn) {
    let i = 0;
    const ret = [];
    const executing = [];
    const enqueue = function () {
        // 邊界處理，array爲空數組
        if (i === array.length) {
            return Promise.resolve();
        }
        // 每調一次enqueue，初始化一個promise
        const item = array[i++];
        const p = Promise.resolve().then(() => iteratorFn(item, array));
        // 放入promises數組
        ret.push(p);
        // promise執行完畢，從executing數組中刪除
        const e = p.then(() => executing.splice(executing.indexOf(e), 1));
        // 插入executing數字，表示正在執行的promise
        executing.push(e);
        // 使用Promise.rece，每當executing數組中promise數量低於poolLimit，就實例化新的promise並執行
        let r = Promise.resolve();
        if (executing.length >= poolLimit) {
            r = Promise.race(executing);
        }
        // 遞歸，直到遍歷完array
        return r.then(() => enqueue());
    };
    return enqueue().then(() => Promise.all(ret));
}

由於是promise加上遞歸，因此在代碼註釋上不太好標註執行順序，可是大概的邏輯能夠總結爲：

1. 從array第1個元素開始，初始化promise對象，同時用一個executing數組保存正在執行的promise
2. 不斷初始化promise，直到達到poolLimt
3. 使用Promise.race，得到executing中promise的執行狀況，當有一個promise執行完畢，繼續初始化promise並放入executing中
4. 全部promise都執行完了，調用Promise.all返回

使用方式就是：

const timeout = i => new Promise(resolve => setTimeout(() => resolve(i), i));
return asyncPool(2, [1000, 5000, 3000, 2000], timeout).then(results => {
    ...
});

總結

所謂promise併發限制，其實根源上就是控制promise的實例化。若是是經過第三方函數，那麼就把建立promise的控制權交給第三方便可。

然而這樣的實現效果，本質上來講已經拋棄了Promise.all而另闢蹊徑。因此期待有一天promise標準能提供這個功能