簡單介紹使用p-limit 限制併發數原始碼解析
併發是指在同一時間內處理的任務數量。例如,在一臺伺服器上同時執行多個 Web 伺服器執行緒,可以同時處理多個客戶端的請求。有時為了程式的穩定執行,我們需要限制併發的數量,p-limit 就是一個用js實現的控制併發數的庫。
原始碼地址:sindresorhus/p-limit
下面是官方提供的使用示例:
import pLimit from 'p-limit'; const limit = pLimit(1); const input = [ limit(() => fetchSomething('foo')), limit(() => fetchSomething('bar')), limit(() => doSomething()) ]; // Only one promise is run at once const result = await Promise.all(input); console.log(result);
在程式碼的第一行,使用了 pLimit(1) 來建立一個 p-limit 例項,並將併發限制設為 1。這意味著,在任意時刻,只能有一個 Promise 在執行。
在第四行,使用了 limit(() => fetchSomething('foo')) 來包裝一個非同步函式,並返回一個 Promise。同樣的方式,在第五、六行也包裝了其他兩個非同步函式。
最後,使用 Promise.all 方法來等待所有 Promise 的完成,並在完成後將結果輸出到控制檯。由於 p-limit 的限制,這些 Promise 只會按順序一個一個地執行,保證了併發的數量不會超過 1。
import Queue from 'yocto-queue'; export default function pLimit(concurrency) { if (!((Number.isInteger(concurrency) || concurrency === Number.POSITIVE_INFINITY) && concurrency > 0)) { throw new TypeError('Expected `concurrency` to be a number from 1 and up'); } const queue = new Queue(); let activeCount = 0; }
yocto-queue 是一種允許高效儲存和檢索資料的資料結構。前邊的章節分析過它的原始碼,詳情參見: 原始碼共讀|yocto-queue 佇列 連結串列
pLimit 函式接受一個引數,併發數,首先函式判斷引數是否是陣列型別,或者是否能夠轉換成數字型別,如果不能,丟擲一個錯誤。
之後定義了一個佇列來儲存待執行的函式,並使用一個計數器來記錄當前正在執行的函式的數量。
const next = () => { activeCount--; if (queue.size > 0) { queue.dequeue()(); } }; const run = async (fn, resolve, args) => { activeCount++; const result = (async () => fn(...args))(); resolve(result); try { await result; } catch {} next(); };
在程式碼的 next 函式中,如果佇列不為空,則從佇列中取出一個函式並執行。這個函式的執行會導致計數器的值減 1。
在程式碼的 run 函式中,使用了 async/await 語法來執行傳入的函式 fn。它還使用了 resolve 函式將函式的返回值包裝成一個 Promise,並將這個 Promise 返回給呼叫者。在函式執行完成後,呼叫 next 函式來執行下一個函式。
const enqueue = (fn, resolve, args) => { queue.enqueue(run.bind(undefined, fn, resolve, args)); (async () => { // This function needs to wait until the next microtask before comparing // `activeCount` to `concurrency`, because `activeCount` is updated asynchronously // when the run function is dequeued and called. The comparison in the if-statement // needs to happen asynchronously as well to get an up-to-date value for `activeCount`. await Promise.resolve(); if (activeCount < concurrency && queue.size > 0) { queue.dequeue()(); } })(); };
在程式碼的 enqueue 函式中,使用了 queue.enqueue 方法將傳入的函式 fn 加入佇列。然後,它使用了 async/await 語法來在下一個微任務中檢查當前的併發數量是否小於設定的併發限制。如果是,則從佇列中取出一個函式並執行。
const generator = (fn, ...args) => new Promise(resolve => { enqueue(fn, resolve, args); }); Object.defineProperties(generator, { activeCount: { get: () => activeCount, }, pendingCount: { get: () => queue.size, }, clearQueue: { value: () => { queue.clear(); }, }, }); return generator;
在程式碼的 generator 函式中,使用了 new Promise 語法來生成一個新的 Promise,並在其中呼叫了 enqueue 函式。這樣,每次呼叫生成的函式時,都會生成一個新的 Promise,並將函式加入佇列。
最後,使用了 Object.defineProperties 方法來為生成的函式新增屬性。這些屬性可以用來查詢當前的併發數量和等待佇列的大小,以及清空等待佇列。
控制併發的實現方式有很多種。例如,可以使用訊號量或佇列來控制併發請求的數量。也可以使用計數器或令牌桶演算法來限制請求的頻率。還可以使用拒絕服務(DoS)防護系統來檢測異常請求流量並進行限制
來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/69901823/viewspace-2933196/,如需轉載,請註明出處,否則將追究法律責任。
相關文章
- 控制請求併發數量:p-limit 原始碼解讀MIT原始碼
- Java併發:ThreadLocal的簡單介紹Javathread
- 簡單介紹nginx 變數使用Nginx變數
- flask 原始碼解析:簡介Flask原始碼
- 限制併發數
- Android IntentService使用全面介紹及原始碼解析AndroidIntent原始碼
- javascript 變數簡單介紹JavaScript變數
- margin引數簡單介紹
- margin 引數簡單介紹
- ReentrantLock介紹及原始碼解析ReentrantLock原始碼
- 短影片平臺原始碼,利用mq限制入庫併發數原始碼MQ
- 簡單介紹PostgreSQL解析URL的方法SQL
- Git發展簡單介紹Git
- javascript變數宣告簡單介紹JavaScript變數
- javascript變數汙染簡單介紹JavaScript變數
- httprunner3原始碼解讀(1)簡單介紹原始碼模組內容HTTP原始碼
- 第 23 期 Drone 簡單介紹和部分原始碼分析原始碼
- 簡單介紹Vue之vue.$set()方法原始碼案例Vue原始碼
- jquery解析json格式字串簡單介紹jQueryJSON字串
- 簡單介紹使用Nginx限制百度蜘蛛頻繁抓取的問題Nginx
- nginx 限制使用者併發連線數Nginx
- webstorm簡單介紹,webstrom基本使用WebORM
- javascript變數作用於簡單介紹JavaScript變數
- javascript變數宣告提升簡單介紹JavaScript變數
- TiKV 原始碼解析(五)fail-rs 介紹原始碼AI
- Guava併發:ListenableFuture使用介紹以及示例Guava
- 簡單介紹下各種 JavaScript 解析器JavaScript
- 簡單介紹Git合併分支的流程步驟Git
- Java併發之Semaphore原始碼解析(一)Java原始碼
- Java併發之Semaphore原始碼解析(二)Java原始碼
- SVG程式碼構成簡單介紹SVG
- css編碼規範簡單介紹CSS
- HTML編碼規範簡單介紹HTML
- javascript程式碼規範簡單介紹JavaScript
- Nginx limit模組限制併發數NginxMIT
- 簡單介紹nginx反向代理及使用Nginx
- POP簡單介紹與使用實踐
- SVG簡單介紹SVG