JavaScript?Promise多并發(fā)問題的解決方法詳解

更新時間：2023年09月19日 11:25:07 作者：喆星高照

提起控制并發(fā),大家應(yīng)該不陌生,這篇文章主要來和大家介紹一下JavaScript如何解決Promise多并發(fā)問題,感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下

代碼實現(xiàn)

實現(xiàn)思路

大致思路就是，假設(shè)現(xiàn)在有 9 個并發(fā)，我設(shè)置最大并發(fā)為 3，那么我將會走下面這些步驟：

1、先定好三個坑位
2、讓前三個并發(fā)進(jìn)去坑位執(zhí)行
3、看哪個坑位并發(fā)先執(zhí)行完，就從剩余的并發(fā)中拿一個進(jìn)去補(bǔ)坑
4、一直重復(fù)第 3 步，一直到所有并發(fā)執(zhí)行完

Promise.all

在進(jìn)行多并發(fā)的時候，我們通常會使用Promise.all，但是Promise.all并不能控制并發(fā)，或者說它本來就沒這個能力，我們可以看下面的例子

const fetchFn = (delay, index) => {
  return new Promise(resolve => {
    console.log(index)
    setTimeout(() => {
      resolve(index)
    }, delay);
  })
}
const promises = [
  fetchFn(1000, 1),
  fetchFn(1000, 2),
  fetchFn(1000, 3),
  fetchFn(1000, 4),
  fetchFn(1000, 5),
  fetchFn(1000, 6)
]
Promise.all(promises)

最后是同時輸出，這說明這幾個并發(fā)是同時發(fā)生的

所以我們需要做一些改造，讓Promise.all執(zhí)行 promises 時支持控制并發(fā)，但是我們改造的不應(yīng)該是Promise.all，而是這一個個的fetchFn

期望效果

const limitFn = (limit) => {
  // ...coding
}
// 最大并發(fā)數(shù) 2
const generator = limitFn(2)
const promises = [
  generator(() => fetchFn(1000, 1)),
  generator(() => fetchFn(1000, 2)),
  generator(() => fetchFn(1000, 3)),
  generator(() => fetchFn(1000, 4)),
  generator(() => fetchFn(1000, 5)),
  generator(() => fetchFn(1000, 6))
]
Promise.all(promises)

實現(xiàn) limitFn

我們需要在函數(shù)內(nèi)部維護(hù)兩個變量：

queue：隊列，用來存每一個改造過的并發(fā)
activeCount：用來記錄正在執(zhí)行的并發(fā)數(shù)

并聲明函數(shù) generator ，這個函數(shù)返回一個 Promise，因為 Promise.all 最好是接收一個 Promise 數(shù)組

const limitFn = (concurrency) => {
  const queue = [];
  let activeCount = 0;
  const generator = (fn, ...args) =>
    new Promise((resolve) => {
      enqueue(fn, resolve, ...args);
    });
  return generator;
};

接下來我們來實現(xiàn) enqueue 這個函數(shù)做兩件事：

將每一個 fetchFn 放進(jìn)隊列里
將坑位里的 fetchFn 先執(zhí)行

const enqueue = (fn, resolve, ...args) => {
  queue.push(run.bind(null, fn, resolve, ...args));
  if (activeCount < limit && queue.length > 0) {
    queue.shift()();
  }
};

假如我設(shè)置最大并發(fā)數(shù)為 2，那么這一段代碼在一開始的時候只會執(zhí)行 2 次，因為一開始只會有 2 次符合 if 判斷，大家可以思考一下為什么~

if (activeCount < limit && queue.length > 0) {
    queue.shift()(); // 這段代碼
  }

一開始執(zhí)行 2 次，說明這時候兩個坑位已經(jīng)各自有一個 fetchFn 在執(zhí)行了

接下來我們實現(xiàn) run 函數(shù)，這個函數(shù)是用來包裝 fetch 的，他完成幾件事情：

1、將 activeCount++ ，這時候執(zhí)行中的并發(fā)數(shù) +1
2、將 fetchFn 執(zhí)行，并把結(jié)果 resolve 出去，說明這個并發(fā)執(zhí)行完了
3、將 activeCount--，這時候執(zhí)行中的并發(fā)數(shù) -1
4、從 queue 中取一個并發(fā)，拿來補(bǔ)坑執(zhí)行

const run = async (fn, resolve, ...args) => {
  activeCount++;
  const result = (async () => fn(...args))();
  try {
    const res = await result;
    resolve(res);
  } catch { }
  next();
};

其實第 3、4 步，是在 next 函數(shù)里面執(zhí)行的

const next = () => {
  activeCount--;
  if (queue.length > 0) {
    queue.shift()();
  }
};

完整代碼

const limitFn = (limit) => {
  const queue = [];
  let activeCount = 0;
  const next = () => {
    activeCount--;
    if (queue.length > 0) {
      queue.shift()();
    }
  };
  const run = async (fn, resolve, ...args) => {
    activeCount++;
    const result = (async () => fn(...args))();
    try {
      const res = await result;
      resolve(res);
    } catch { }
    next();
  };
  const enqueue = (fn, resolve, ...args) => {
    queue.push(run.bind(null, fn, resolve, ...args));
    if (activeCount < limit && queue.length > 0) {
      queue.shift()();
    }
  };
  const generator = (fn, ...args) =>
    new Promise((resolve) => {
      enqueue(fn, resolve, ...args);
    });
  return generator;
};

這不是我寫的

其實這是一個很出名的庫的源碼，就是p-limit，哈哈，但是重要嗎？知識嘛，讀懂了，它就是你的，到時跟面試官嘮嗑的時候，他哪知道是不是真的是你寫的~

到此這篇關(guān)于JavaScript Promise多并發(fā)問題的解決方法詳解的文章就介紹到這了,更多相關(guān)JavaScript Promise內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: