关于Nodejs cluster fork多线程任务的疑问

关于Node.js Cluster Fork 多线程任务的疑问

在Node.js中，cluster模块允许你在单个Node.js进程中利用多核CPU的能力。通过cluster.fork()方法，你可以创建多个工作进程（workers），每个工作进程都会运行相同的主进程代码。

示例代码

假设你有一个简单的Node.js应用，它使用socket.io来处理WebSocket连接。你希望使用cluster模块来分发这些连接到不同的工作进程。以下是一个基本的示例：

const cluster = require('cluster');
const http = require('http');
const numCPUs = require('os').cpus().length;
const io = require('socket.io');

if (cluster.isMaster) {
    // 主进程
    for (let i = 0; i < numCPUs; i++) {
        cluster.fork();
    }
} else {
    // 工作进程
    const server = http.createServer((req, res) => {
        res.writeHead(200);
        res.end('Hello World\n');
    });
    server.listen(8000);

    const socketServer = io(server);
    
    socketServer.on('connection', (socket) => {
        console.log(`Worker ${cluster.worker.id} got a connection`);
        socket.emit('message', `Hello from worker ${cluster.worker.id}`);
        
        // 业务逻辑
        logic(socket);
    });
}

function logic(socket) {
    // 在这里实现你的业务逻辑
}

解释

1. 主进程：

   • 使用cluster.fork()方法创建与CPU核心数量相等的工作进程。
   • 每个工作进程都会运行相同的代码，但它们会在不同的进程中运行。

2. 工作进程：

   • 创建一个HTTP服务器并监听端口8000。
   • 使用socket.io创建WebSocket服务器。
   • 监听连接事件，并打印日志信息。
   • 调用logic(socket)函数来处理具体的业务逻辑。

问题分析

根据你的描述，只有ID为4的工作进程在处理所有的连接。这可能是因为socket.io默认情况下将所有连接都路由到了同一个工作进程。为了公平地分配连接，你可以使用sticky-session库来确保连接被均匀地分发到各个工作进程。

解决方案

可以考虑使用sticky-session库来解决这个问题。sticky-session会确保WebSocket连接被正确地路由到不同的工作进程。以下是如何使用sticky-session的示例：

const sticky = require('sticky-session');
const http = require('http');
const io = require('socket.io');

if (!sticky.listen(http.createServer())) {
    const socketServer = io();

    socketServer.on('connection', (socket) => {
        console.log(`Worker ${cluster.worker.id} got a connection`);
        socket.emit('message', `Hello from worker ${cluster.worker.id}`);
        
        // 业务逻辑
        logic(socket);
    });

    function logic(socket) {
        // 在这里实现你的业务逻辑
    }

    return;
}

console.log('Sticky session running on master process.');

通过这种方式，sticky-session会确保WebSocket连接被均匀地分配到各个工作进程，从而避免某个进程负载过重的问题。

我现在基本放弃了多进程的打算。而是开多端口的方式。这样就是传统的负载均衡了