1 設(shè)計(jì)思路
設(shè)計(jì)的思路比較簡(jiǎn)單，就是在用戶和work_threads模塊之前加一層，如果用戶直接調(diào)用work_threads模塊，那就可以創(chuàng)建任意數(shù)量的線程，控制就無(wú)從說(shuō)起。通過(guò)加一層，可以緩存用戶提交的任務(wù)，等到有線程退出（有任務(wù)處理完）的時(shí)候，再開(kāi)啟新的線程去處理緩存的任務(wù)。

2 具體實(shí)現(xiàn)
2.1 配置的實(shí)現(xiàn)
定義一些配置，比如最多能創(chuàng)建的線程數(shù)。使用Proxy，劫持一下對(duì)象，可以做一些事情，不過(guò)目前沒(méi)有。

// 線程池配置，使用Proxy，可以劫持屬性的讀寫(xiě)，做點(diǎn)事情
const CONFIG = new Proxy({
    MAX_THREAD: 3,
}, {
    get(obj, key) {
        return obj[key];
    },
    set(obj, key, value) {
        obj[key] = value;
        return true;
    }
})

2.2 控制邏輯的實(shí)現(xiàn)

2.2.1 構(gòu)造函數(shù)
線程池記錄當(dāng)前的線程數(shù)和緩存的任務(wù)隊(duì)列。

    constructor() {
        // 任務(wù)隊(duì)列
        this._workQueue = [];
        // 當(dāng)前線程數(shù)
        this._count = 0;
    }

2.2.2 新建一個(gè)線程

_newThread(...rest) {
        // todo 新建失敗處理
        const worker = new Worker(...rest);
        // 線程數(shù)加一
        this._count++;
        // 退出后，如果有緩存的任務(wù)，則新建線程處理
        worker.on('exit', () => {
            this._count--;
            // 有名額了并且有任務(wù)在等待
            if (this._workQueue.length) {
               const {
                   resolve,
                   reject,
                   params,
               } = this._workQueue.shift();
               // 開(kāi)啟線程，并且通知用戶
               resolve(this._newThread(...params));
            }
        });
        return worker;
    }

2.2.3 提交任務(wù)給線程池

  // 給線程池提交一個(gè)任務(wù)
    submit(...rest) {
        return new Promise((resolve, reject) => {
            // 還沒(méi)有達(dá)到閾值，則新建線程，否則緩存起來(lái)
            if (this._count < CONFIG.MAX_THREAD) {
                resolve(this._newThread(...rest));
            } else {
                this._workQueue.push({resolve, reject, params: rest});
            }
        });
    }
}

思想很簡(jiǎn)單，根據(jù)配置，最多創(chuàng)建n個(gè)線程，如果任務(wù)太多的話，就緩存起來(lái)，等待有線程退出的時(shí)候，再新建一個(gè)線程處理緩存起來(lái)的任務(wù)。API使用Promise方式。這樣可以實(shí)現(xiàn)等待操作。等到創(chuàng)建線程的時(shí)候可以通知用戶。而且用戶使用的時(shí)候，幾乎是透明的，沒(méi)有太多額外的成本，因?yàn)橹皇亲隽艘恍┓庋b，幾乎是透?jìng)鱪odejs的線程功能。最后提供多種方式調(diào)用，包括一個(gè)默認(rèn)的控制器、創(chuàng)建多個(gè)控制器。

const defaultThreadGate = new ThreadGate();
module.exports = {
    ThreadGate,
    defaultThreadGate,
    submit: (...rest) => {
        return defaultThreadGate.submit(...rest);
    },
    CONFIG,
}