网站建设资讯

NEWS

网站建设资讯

nodejs中怎么控制线程数

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中怎么控制线程数的方法了吗?如果还想学到更多技能或想了解更多相关内容,欢迎关注创新互联行业资讯频道,感谢各位的阅读!


本文标题:nodejs中怎么控制线程数
转载注明:http://cdweb.net/article/jhhgcc.html