Node.js 中如何使用 cluster 实现多进程并行处理

在 Node.js 中，单线程的模型使得处理大量并发请求时效率不高，因此使用多进程并行处理是一种常见的解决方案。Node.js 中提供了 cluster 模块，可以方便地实现多进程并行处理。

cluster 模块介绍

cluster 模块是 Node.js 的一个核心模块，用于创建子进程并将请求分发给子进程处理。cluster 模块可以将一个 Node.js 应用程序分成多个进程，每个进程独立运行，互不干扰，可以充分利用多核 CPU 的优势，提高应用程序的并发性能。

cluster 模块的使用

使用 cluster 模块实现多进程并行处理，需要以下步骤：

1. 引入 cluster 模块

const cluster = require('cluster');

1. 判断是否为主进程

if (cluster.isMaster) {
  // 主进程代码
} else {
  // 子进程代码
}

1. 创建子进程

if (cluster.isMaster) {
  // 创建子进程
  for (let i = 0; i < numCPUs; i++) {
    cluster.fork();
  }
} else {
  // 子进程代码
}

1. 监听子进程退出事件

cluster.on('exit', (worker, code, signal) => {
  console.log(`worker ${worker.process.pid} died`);
  cluster.fork();
});

1. 将请求分发给子进程

// javascriptcn.com 代码示例
if (cluster.isMaster) {
  // 主进程代码
  for (const id in cluster.workers) {
    cluster.workers[id].send({ msg: 'hello' });
  }
} else {
  // 子进程代码
  process.on('message', (msg) => {
    console.log(`worker ${process.pid} received message: ${msg}`);
  });
}

示例代码

下面是一个使用 cluster 模块实现多进程并行处理的示例代码：

// javascriptcn.com 代码示例
const http = require('http');
const cluster = require('cluster');
const numCPUs = require('os').cpus().length;

if (cluster.isMaster) {
  console.log(`Master ${process.pid} is running`);

  // 创建子进程
  for (let i = 0; i < numCPUs; i++) {
    cluster.fork();
  }

  // 监听子进程退出事件
  cluster.on('exit', (worker, code, signal) => {
    console.log(`worker ${worker.process.pid} died`);
    cluster.fork();
  });

  // 将请求分发给子进程
  http.createServer((req, res) => {
    for (const id in cluster.workers) {
      cluster.workers[id].send({ msg: 'hello' });
    }
    res.writeHead(200);
    res.end('Hello World\n');
  }).listen(8000);

} else {
  console.log(`Worker ${process.pid} started`);
  // 子进程代码
  process.on('message', (msg) => {
    console.log(`worker ${process.pid} received message: ${msg}`);
  });
}

总结

使用 cluster 模块实现多进程并行处理可以提高 Node.js 应用程序的并发性能，但也需要注意一些问题，比如进程间通信、子进程之间的负载均衡等。在实际应用中需要根据具体情况进行调整和优化，以达到最佳的性能和稳定性。

来源：JavaScript中文网 ，转载请注明来源 本文地址：https://www.javascriptcn.com/post/656d609ed2f5e1655d5a7c23