NodeJs多進程和socket.io通訊

一、開啟多進程

const os = require('os');
const cp = require('child_process');

const forkList = {};
const forkPrefix = 'fork_';
let cpusLen = os.cpus().length;
let port = 9562;
for (var i = 1; i <= cpusLen; i++) {
    forkList[forkPrefix + i] = cp.fork(`${__dirname}/a.js`, [port]);
    var childPid = forkList[forkPrefix + i].pid;
    console.log("子進程Fork成功.Pid:" + childPid);
    port++;
}

// 監聽子進程分發消息
var pid = process.pid;
for (var key in forkList) {
    var forkObj = forkList[key];
    var childPid = forkObj.pid;
    forkObj.on('message', (m) => {
        console.log('主進程ID:' + pid + ',子進程ID:' + childPid + ',message:', m);
        send(m);
    });
}

// 輪詢所有子進程消息發送
function send(m) {
    for (var key in forkList) {
        forkList[key].send(m);
    }
}

 

二、啟動websocket服務-------（a.js）

var pid = process.pid;
console.log('starting socket.io server......', pid);

var argvs = process.argv.slice(2);
if (argvs.length <= 0) {
    console.log("fork child params bad...");
    return false;
}

var port = argvs[0];
var server = require('http').createServer();
var io = require('socket.io')(server);
io.on('connection', function (socket) {
    socket.on('router', function (data) {
        console.log(data);

        // 把消息發送給主進程，由主進程分發給所有子進程
        process.send(data);
    });

    socket.on('disconnect', function () {

    });
});
server.listen(port);

// 監聽由主進程分發消息
process.on('message', (m) => {
    console.log('pid:' + pid + ',message:', m);
    io.sockets.emit('router', m);
});

 

三、DEMO演示截圖

1、服務端啟動程序

 2、客戶端1

 

3、客戶端2

4、客戶端3

5、客戶端4