nodejs express+socket.io多线程原始实现

nodejs express+socket.io多线程原始实现

本文主要使用到express + socket.io 实现session共享，以及开启多线程。

session共享请参考上一片文章 http://blog.csdn.net/u012251421/article/details/76207034，里面详细介绍了如何使用express和socket.io 开启多线程。

下面介绍一下如何使用nodejs 内部的cluster实现多线程。如果是单纯的express,没有socket.io的情况是很好实现的,具体代码如下：

var app = require('../app'),
session = require('../session'),
cluster = require('cluster'),
net = require('net'),
numCPUs = require('os').cpus().length,
http = require('http'),
fs = require('fs'),
ServerPort = '80',
server = http.createServer(app),
config = JSON.parse(fs.readFileSync("./config.json")),
if (cluster.isMaster) {
console.log('Master is running');
for (var i = 0; i < numCPUs; i++) {
cluster.fork();
}
cluster.on('exit', function (worker, code, signal) {
console.log('worker ${worker.process.pid} died');
});
} else {
server.listen(ServerPort);
server.on('error', onError);
server.on('listening', onListening);
}

使用上面的代码就可以实现单纯express多线程，上面使用到redis来实现session共享，代码如下：

var sess = require('express-session');
var RedisStore = require('connect-redis')(sess);
var redisConn = require('./routes/redis/redisConn');
// var session = sess({
//     secret: 'magustek.com',
//     name: 'magustek.com', //这里的name值得是cookie的name，默认cookie的name是：connect.sid
//     cookie: {maxAge: 30 * 60 * 1000}, //设置maxAge是30分钟，ession和相应的cookie失效过期
//     resave: false,
//     rolling: true,
//     saveUninitialized: true
// });

var session = sess({
secret: 'magustek.com',
store: new RedisStore({client: redisConn.Client, host: redisConn.IP, port: redisConn.PORT, ttl: 30 * 60, db: 15})
});
module.exports = session;

redisConn 内容：

/**
* Created by se7en on 2017/7/17.
*/

var redis = require('redis');
var fs = require('fs');

var logger = require('log4js').getLogger('system');
var dbConfig = JSON.parse(fs.readFileSync("./config.json"));

var IP = "192.168.2.64";
var PORT = 6379;
var USER = "root";
var PWD = "123456";

if (dbConfig && dbConfig.redis) {
var dbCon = dbConfig.redis;
if (dbCon.IP) {
IP = dbCon.IP;
}
if (dbCon.PORT) {
PORT = dbCon.PORT;
}
if (dbCon.USER) {
USER = dbCon.USER;
}
if (dbCon.PWD) {
PWD = dbCon.PWD;
}
}

var client = redis.createClient(PORT, IP);
client.on('error', function (error) {
client.quit();
logger.error('redis conn error:', error);
client = redis.createClient(PORT, IP);
});

var redisConn = {
name: 'redis',
PORT: PORT,
IP: IP,
Client: client
}

module.exports = redisConn;

使用以上就可以实现多线程的nodejs以及session共享了。

下面介绍一下express+socket.io实现多线程，具体代码如下：

var app = require('../app'),
session = require('../session'),
cluster = require('cluster'),
net = require('net'),
numCPUs = require('os').cpus().length,
http = require('http'),
fs = require('fs'),
api = require('../routes/api/apiV1'),
ServerPort = '80',
server = http.createServer(app),
io = require('socket.io').listen(server),
config = JSON.parse(fs.readFileSync("./config.json")),

if (config) {
if (config && config.server && config.server.PORT) {
ServerPort = config.server.PORT;
}
}
ServerPort = normalizePort(ServerPort);

if (cluster.isMaster) {
var workers = [];
var spawn = function (i) {
workers[i] = cluster.fork();
workers[i].on('exit', function (code, signal) {
console.log('respawning worker', i);
spawn(i);
});
};
for (var i = 0; i < numCPUs; i++) {
spawn(i);
}
var worker_index = function (ip, len) {
var s = '';
for (var i = 0, _len = ip.length; i < _len; i++) {
if (!isNaN(ip[i])) {
s += ip[i];
}
}
return Number(s) % len;
};
var server = net.createServer({pauseOnConnect: true}, function (connection) {
var worker = workers[worker_index(connection.remoteAddress, numCPUs)];
worker.send('sticky-session:connection', connection);
}).listen(ServerPort);
} else {
api.init(io);
server.listen(0);
process.on('message', function (message, connection) {
if (message !== 'sticky-session:connection') {
return;
}
server.emit('connection', connection);
connection.resume();
});
}

function normalizePort(val) {
var port = parseInt(val, 10);
if (isNaN(port)) {
return val;
}
if (port >= 0) {
return port;
}
return false;
}

可以看出和单纯的express相比主要修改了cluster.isMaster里面的处理逻辑，不同系统下处理逻辑不太一样，不过可以参考以上的实现。

参考文章：

https://github.com/elad/node-cluster-socket.io

https://my.oschina.net/swingcoder/blog/527648