您好,登錄后才能下訂單哦!
nodejs中怎么控制線程數,相信很多沒有經驗的人對此束手無策,為此本文總結了問題出現的原因和解決方法,通過這篇文章希望你能解決這個問題。
1 設計思路
設計的思路比較簡單,就是在用戶和work_threads模塊之前加一層,如果用戶直接調用work_threads模塊,那就可以創建任意數量的線程,控制就無從說起。通過加一層,可以緩存用戶提交的任務,等到有線程退出(有任務處理完)的時候,再開啟新的線程去處理緩存的任務。
2 具體實現
2.1 配置的實現
定義一些配置,比如最多能創建的線程數。使用Proxy,劫持一下對象,可以做一些事情,不過目前沒有。
// 線程池配置,使用Proxy,可以劫持屬性的讀寫,做點事情
const CONFIG = new Proxy({
MAX_THREAD: 3,
}, {
get(obj, key) {
return obj[key];
},
set(obj, key, value) {
obj[key] = value;
return true;
}
})
2.2 控制邏輯的實現
2.2.1 構造函數
線程池記錄當前的線程數和緩存的任務隊列。
constructor() {
// 任務隊列
this._workQueue = [];
// 當前線程數
this._count = 0;
}
2.2.2 新建一個線程
_newThread(...rest) {
// todo 新建失敗處理
const worker = new Worker(...rest);
// 線程數加一
this._count++;
// 退出后,如果有緩存的任務,則新建線程處理
worker.on('exit', () => {
this._count--;
// 有名額了并且有任務在等待
if (this._workQueue.length) {
const {
resolve,
reject,
params,
} = this._workQueue.shift();
// 開啟線程,并且通知用戶
resolve(this._newThread(...params));
}
});
return worker;
}
2.2.3 提交任務給線程池
// 給線程池提交一個任務
submit(...rest) {
return new Promise((resolve, reject) => {
// 還沒有達到閾值,則新建線程,否則緩存起來
if (this._count < CONFIG.MAX_THREAD) {
resolve(this._newThread(...rest));
} else {
this._workQueue.push({resolve, reject, params: rest});
}
});
}
}
思想很簡單,根據配置,最多創建n個線程,如果任務太多的話,就緩存起來,等待有線程退出的時候,再新建一個線程處理緩存起來的任務。API使用Promise方式。這樣可以實現等待操作。等到創建線程的時候可以通知用戶。而且用戶使用的時候,幾乎是透明的,沒有太多額外的成本,因為只是做了一些封裝,幾乎是透傳nodejs的線程功能。最后提供多種方式調用,包括一個默認的控制器、創建多個控制器。
const defaultThreadGate = new ThreadGate();
module.exports = {
ThreadGate,
defaultThreadGate,
submit: (...rest) => {
return defaultThreadGate.submit(...rest);
},
CONFIG,
}
看完上述內容,你們掌握nodejs中怎么控制線程數的方法了嗎?如果還想學到更多技能或想了解更多相關內容,歡迎關注億速云行業資訊頻道,感謝各位的閱讀!
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。