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中怎么控制线程数的方法了吗？如果还想学到更多技能或想了解更多相关内容，欢迎关注亿速云行业资讯频道，感谢各位的阅读！
亿速云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>
免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：is@yisu.com进行举报，并提供相关证据，一经查实，将立刻删除涉嫌侵权内容。
原文链接：https://my.oschina.net/u/4217331/blog/4464928
