一、啥是 Node.js 集群模式
咱先聊聊啥是 Node.js 集群模式。大家都知道，Node.js 是单线程运行的，就像一个人干活，一次只能做一件事。要是遇到很多请求同时过来，这一个人就忙不过来了，处理速度就会变慢。而集群模式呢，就好比叫来了好多人一起干活，充分利用电脑多核 CPU 的能力，让每个核心都能处理请求，这样就能大大提升服务性能。
举个简单例子，假如你开了一家餐厅，只有一个厨师，客人多了，做菜速度就跟不上。要是你多雇几个厨师，每个厨师负责做一部分菜，那效率肯定就提高了。这就是 Node.js 集群模式的原理，把任务分配给多个工作进程，让它们并行处理请求。
二、为啥要用 Node.js 集群模式
应用场景
Node.js 集群模式在很多场景下都能发挥大作用。比如说，你做了一个电商网站，在促销活动的时候，会有大量用户同时访问，这时候单线程的 Node.js 就很难应对了。用集群模式，就可以把这些请求分配给多个工作进程处理，保证网站的响应速度。
再比如，你做的是实时聊天应用，会有很多用户同时发送消息，要是用单线程处理，消息的收发就可能会有延迟。用集群模式，就能让多个工作进程同时处理消息，提高消息的处理速度。
技术优缺点
优点
- 提升性能：前面也说过了，利用多核 CPU，让多个工作进程并行处理请求，能大大提高服务的处理能力。
- 高可用性：如果一个工作进程挂了，其他工作进程还能继续工作，不会影响整个服务的运行。就像餐厅里有一个厨师生病了，其他厨师还能接着做菜，客人还是能吃到饭。
缺点
- 管理复杂度增加：多个工作进程需要管理，比如进程的启动、停止、监控等，这就增加了管理的复杂度。
- 资源消耗变大：多个工作进程会占用更多的系统资源，比如内存、CPU 等。
注意事项
在使用 Node.js 集群模式的时候，有一些注意事项。首先，要合理分配工作进程的数量。如果进程数量太多，会占用过多的系统资源；如果进程数量太少，又不能充分利用多核 CPU 的能力。一般来说，工作进程的数量可以设置为 CPU 的核心数。
其次，要注意进程间的通信。多个工作进程之间可能需要共享数据或者传递消息，这就需要使用合适的通信机制，比如 IPC（进程间通信）。
三、怎么用 Node.js 集群模式
下面我们就来看看怎么用 Node.js 集群模式。这里我们用 Node.js 来实现一个简单的 HTTP 服务器，并且使用集群模式来提升性能。
示例代码（Node.js 技术栈）
// 引入 cluster 和 http 模块
const cluster = require('cluster');
const http = require('http');
// 获取 CPU 核心数
const numCPUs = require('os').cpus().length;
if (cluster.isMaster) {
// 主进程代码
console.log(`主进程 ${process.pid} 正在运行`);
// 为每个 CPU 核心创建一个工作进程
for (let i = 0; i < numCPUs; i++) {
cluster.fork();
}
// 监听工作进程退出事件
cluster.on('exit', (worker, code, signal) => {
console.log(`工作进程 ${worker.process.pid} 已退出`);
});
} else {
// 工作进程代码
http.createServer((req, res) => {
res.writeHead(200);
res.end('你好，世界！\n');
}).listen(8000);
console.log(`工作进程 ${process.pid} 已启动`);
}
代码解释
- 引入模块：首先引入了
cluster和http模块，cluster模块用于实现集群模式，http模块用于创建 HTTP 服务器。 - 主进程代码：通过
cluster.isMaster判断当前进程是否为主进程。如果是主进程，就根据 CPU 核心数创建相应数量的工作进程，并且监听工作进程的退出事件。 - 工作进程代码：如果是工作进程，就创建一个 HTTP 服务器，监听 8000 端口，当有请求过来时，返回 “你好，世界！”。
四、进程间通信示例
多个工作进程之间可能需要共享数据或者传递消息，下面我们来看一个进程间通信的示例。
// 引入 cluster 和 http 模块
const cluster = require('cluster');
const http = require('http');
const numCPUs = require('os').cpus().length;
if (cluster.isMaster) {
// 主进程代码
console.log(`主进程 ${process.pid} 正在运行`);
// 为每个 CPU 核心创建一个工作进程
for (let i = 0; i < numCPUs; i++) {
const worker = cluster.fork();
// 监听工作进程发送的消息
worker.on('message', (msg) => {
console.log(`主进程收到来自工作进程 ${worker.process.pid} 的消息: ${msg}`);
});
}
// 监听工作进程退出事件
cluster.on('exit', (worker, code, signal) => {
console.log(`工作进程 ${worker.process.pid} 已退出`);
});
} else {
// 工作进程代码
http.createServer((req, res) => {
res.writeHead(200);
res.end('你好，世界！\n');
// 向主进程发送消息
process.send('处理了一个请求');
}).listen(8000);
console.log(`工作进程 ${process.pid} 已启动`);
}
代码解释
在这个示例中，工作进程在处理完请求后，会向主进程发送一条消息 “处理了一个请求”。主进程会监听工作进程发送的消息，并打印出来。这样就实现了工作进程和主进程之间的通信。
五、负载均衡
在集群模式中，负载均衡是很重要的。负载均衡就是把请求均匀地分配给多个工作进程，避免某个工作进程过于繁忙，而其他工作进程闲置。
Node.js 的 cluster 模块默认使用了一种简单的负载均衡算法，叫做轮询算法。轮询算法就是依次把请求分配给每个工作进程，循环往复。
自定义负载均衡
除了使用默认的轮询算法，我们还可以自定义负载均衡算法。下面是一个自定义负载均衡的示例。
const cluster = require('cluster');
const http = require('http');
const numCPUs = require('os').cpus().length;
if (cluster.isMaster) {
// 主进程代码
console.log(`主进程 ${process.pid} 正在运行`);
const workers = [];
// 为每个 CPU 核心创建一个工作进程
for (let i = 0; i < numCPUs; i++) {
const worker = cluster.fork();
workers.push(worker);
}
// 自定义负载均衡算法
let currentWorkerIndex = 0;
http.createServer((req, res) => {
const worker = workers[currentWorkerIndex];
// 把请求转发给工作进程
worker.send({ cmd: 'forward', req: req.url });
currentWorkerIndex = (currentWorkerIndex + 1) % numCPUs;
// 监听工作进程的响应
worker.on('message', (msg) => {
if (msg.cmd === 'response') {
res.writeHead(200);
res.end(msg.data);
}
});
}).listen(8000);
// 监听工作进程退出事件
cluster.on('exit', (worker, code, signal) => {
console.log(`工作进程 ${worker.process.pid} 已退出`);
});
} else {
// 工作进程代码
process.on('message', (msg) => {
if (msg.cmd === 'forward') {
const responseData = `处理请求: ${msg.req}`;
// 向主进程发送响应
process.send({ cmd: 'response', data: responseData });
}
});
}
代码解释
在这个示例中，主进程创建了一个 HTTP 服务器，并且自定义了负载均衡算法。每次有请求过来时，主进程会把请求转发给一个工作进程，然后更新当前工作进程的索引。工作进程处理完请求后，会把响应发送给主进程，主进程再把响应返回给客户端。
六、文章总结
通过本文，我们了解了 Node.js 集群模式的原理、应用场景、优缺点和注意事项。我们知道了 Node.js 集群模式可以充分利用多核 CPU 的能力，提升服务性能，适用于高并发的场景。同时，我们也学习了如何使用 Node.js 的 cluster 模块来实现集群模式，包括创建工作进程、进程间通信和负载均衡等。
不过，使用集群模式也会增加管理的复杂度和资源消耗，所以在实际使用中，要根据具体情况合理使用。
评论