JavaScript中如何實(shí)現并發(fā)控制

這篇文章將為大家詳細講解有關(guān)JavaScript中如何實(shí)現并發(fā)控制，文章內容質(zhì)量較高，因此小編分享給大家做個(gè)參考，希望大家閱讀完這篇文章后對相關(guān)知識有一定的了解。

一、前言

在開(kāi)發(fā)過(guò)程中，有時(shí)會(huì )遇到需要控制任務(wù)并發(fā)執行數量的需求。

例如一個(gè)爬蟲(chóng)程序，可以通過(guò)限制其并發(fā)任務(wù)數量來(lái)降低請求頻率，從而避免由于請求過(guò)于頻繁被封禁問(wèn)題的發(fā)生。

接下來(lái)，本文介紹如何實(shí)現一個(gè)并發(fā)控制器。

二、示例

const task = timeout => new Promise((resolve) => setTimeout(() => {       resolve(timeout);     }, timeout))       const taskList = [1000, 3000, 200, 1300, 800, 2000];      async function startNoConcurrentControl() {       console.time(NO_CONCURRENT_CONTROL_LOG);       await Promise.all(taskList.map(item => task(item)));      console.timeEnd(NO_CONCURRENT_CONTROL_LOG);     }     startNoConcurrentControl();

上述示例代碼利用 Promise.all 方法模擬6個(gè)任務(wù)并發(fā)執行的場(chǎng)景，執行完所有任務(wù)的總耗時(shí)為 3000 毫秒。

下面會(huì )采用該示例來(lái)驗證實(shí)現方法的正確性。

三、實(shí)現

由于任務(wù)并發(fā)執行的數量是有限的，那么就需要一種數據結構來(lái)管理不斷產(chǎn)生的任務(wù)。

隊列的**「先進(jìn)先出」特性可以保證任務(wù)并發(fā)執行的順序，在 JavaScript 中可以通過(guò)「數組來(lái)模擬隊列」**：

class Queue {        constructor() {          this._queue = [];        }          push(value) {          return this._queue.push(value);        }          shift() {          return this._queue.shift();        }        isEmpty() {          return this._queue.length === 0;       }      }

對于每一個(gè)任務(wù)，需要管理其執行函數和參數：

class DelayedTask {        constructor(resolve, fn, args) {          this.resolve = resolve;          this.fn = fn;          this.args = args;        }      }

接下來(lái)實(shí)現核心的 TaskPool 類(lèi)，該類(lèi)主要用來(lái)控制任務(wù)的執行：

class TaskPool {        constructor(size) {          this.size = size;          this.queue = new Queue();       }          addTask(fn, args) {          return new Promise((resolve) => {            this.queue.push(new DelayedTask(resolve, fn, args));          if (this.size) {              this.size--;              const { resolve: taskResole, fn, args } = this.queue.shift();              taskResole(this.runTask(fn, args));            }          })        }        pullTask() {          if (this.queue.isEmpty()) {            return;          }          if (this.size === 0) {            return;          }          this.size++;          const { resolve, fn, args } = this.queue.shift();          resolve(this.runTask(fn, args));        }        runTask(fn, args) {          const result = Promise.resolve(fn(...args));           result.then(() => {           this.size--;            this.pullTask();          }).catch(() => {            this.size--;            this.pullTask();         })           return result;       }      }

TaskPool 包含三個(gè)關(guān)鍵方法：

•  addTask: 將新的任務(wù)放入隊列當中，并觸發(fā)任務(wù)池狀態(tài)檢測，如果當前任務(wù)池非滿(mǎn)載狀態(tài)，則從隊列中取出任務(wù)放入任務(wù)池中執行。

•  runTask: 執行當前任務(wù)，任務(wù)執行完成之后，更新任務(wù)池狀態(tài)，此時(shí)觸發(fā)主動(dòng)拉取新任務(wù)的機制。

•  pullTask: 如果當前隊列不為空，且任務(wù)池不滿(mǎn)載，則主動(dòng)取出隊列中的任務(wù)執行。

接下來(lái)，將前面示例的并發(fā)數控制為2個(gè)：

const cc = new ConcurrentControl(2);        async function startConcurrentControl() {        console.time(CONCURRENT_CONTROL_LOG);        await Promise.all(taskList.map(item => cc.addTask(task, [item])))        console.timeEnd(CONCURRENT_CONTROL_LOG);      }        startConcurrentControl();

執行流程如下：

最終執行任務(wù)的總耗時(shí)為 5000 毫秒。

四、高階函數優(yōu)化參數傳遞

await Promise.all(taskList.map(item => cc.addTask(task, [item])))

手動(dòng)傳遞每個(gè)任務(wù)的參數的方式顯得非常繁瑣，這里可以通過(guò)**「高階函數實(shí)現參數的自動(dòng)透傳」**：

addTask(fn) {       return (...args) => {          return new Promise((resolve) => {            this.queue.push(new DelayedTask(resolve, fn, args));            if (this.size) {              this.size--;              const { resolve: taskResole, fn: taskFn, args: taskArgs } = this.queue.shift();              taskResole(this.runTask(taskFn, taskArgs));            }          })        }      }

改造之后的代碼顯得簡(jiǎn)潔了很多：

await Promise.all(taskList.map(cc.addTask(task)))

五、優(yōu)化出隊操作

數組一般都是基于一塊**「連續內存」**來(lái)存儲，當調用數組的 shift 方法時(shí)，首先是刪除頭部元素（時(shí)間復雜度 O(1)），然后需要將未刪除元素左移一位（時(shí)間復雜度 O(n)），所以 shift 操作的時(shí)間復雜度為 O(n)。

由于 JavaScript 語(yǔ)言的特性，V8 在實(shí)現 JSArray 的時(shí)候給出了一種空間和時(shí)間權衡的解決方案，在不同的場(chǎng)景下，JSArray 會(huì )在 FixedArray 和 HashTable 兩種模式間切換。

在 hashTable 模式下，shift 操作省去了左移的時(shí)間復雜度，其時(shí)間復雜度可以降低為 O(1)，即使如此，shift 仍然是一個(gè)耗時(shí)的操作。

在數組元素比較多且需要頻繁執行 shift 操作的場(chǎng)景下，可以通過(guò) 「reverse + pop」 的方式優(yōu)化。

const Benchmark = require('benchmark');      const suite = new Benchmark.Suite;        suite.add('shift', function() {        let count = 10;        const arr = generateArray(count);        while (count--) {          arr.shift();        }     })     .add('reverse + pop', function() {        let count = 10;        const arr = generateArray(count);        arr.reverse();        while (count--) {          arr.pop();        }      })      .on('cycle', function(event) {        console.log(String(event.target));      })      .on('complete', function() {        console.log('Fastest is ' + this.filter('fastest').map('name'));        console.log('\n')      })      .run({        async: true      })

通過(guò) benchmark.js 跑出的基準測試數據，可以很容易地看出哪種方式的效率更高：

回顧之前 Queue 類(lèi)的實(shí)現，由于只有一個(gè)數組來(lái)存儲任務(wù)，直接使用 reverse + pop 的方式，必然會(huì )影響任務(wù)執行的次序。

這里就需要引入雙數組的設計，一個(gè)數組負責入隊操作，一個(gè)數組負責出隊操作。

class HighPerformanceQueue {       constructor() {         this.q1 = []; // 用于 push 數據         this.q2 = []; // 用于 shift 數據       }         push(value) {         return this.q1.push(value);      }       shift() {         let q2 = this.q2;        if (q2.length === 0) {           const q1 = this.q1;           if (q1.length === 0) {             return;           }           this.q1 = q2; // 感謝 @shaonialife 同學(xué)指正           q2 = this.q2 = q1.reverse();         }         return q2.pop();       }       isEmpty() {         if (this.q1.length === 0 && this.q2.length === 0) {         return true;        }         return false;       }     }

最后通過(guò)基準測試來(lái)驗證優(yōu)化的效果：