高并发下的Node.js与负载均衡

 

新兴的Node.js已经吸引了很多开发人员的眼光，它提供给我们一个快速构建高性能的网络应用的平台。我也开始逐步投入node.js的怀抱，在学习和使用的过程中，遇到了一些问题，也有一些经验，我觉得有必要写出来，作为总结，也用作分享。

众所周知，node.js基于v8引擎，所以它本身并不支持多线程（有多线程的Module哦），那么为了充分利用server的Multi-core，就必须使用多进程的方式。那么进程之间如何负载均衡就会是一个关键所在。

多进程共享监听socket

Node.js与进程相关的模块有process，child_process，cluster，这其中cluster用于方便的创建共享端口的多进程模式（The cluster module allows you to easily create a network of processes that all share server ports）,这种模式使多个进程间共享一个监听状态的socket，并由系统将accept的connection分配给不同的子进程，而且实现起来也非常简单，cluster为你做了大部分事情，这里有一个test case：

 1 var cluster = require('cluster');
 2 var http = require('http');
 3 var numCPUs = require('os').cpus().length;
 4 
 5 if (cluster.isMaster) {
 6   // Fork workers.
 7   for (var i = 0; i < numCPUs; i++) {
 8     cluster.fork();
 9   }
10 
11   cluster.on('exit', function(worker, code, signal) {
12     console.log('worker ' + worker.process.pid + ' died');
13   });
14 } else {
15   // Workers can share any TCP connection
16   // In this case its a HTTP server
17   http.createServer(function(req, res) {
18     res.writeHead(200);
19     res.end("hello world
");
20   }).listen(8000);
21 }

但是这种完全依赖于系统的负载均衡存在着一个重要缺陷：在linux和Solaris上，只要某个子进程的accept queue为空（通常为最后创建的那个子进程），系统就会将多个connetion分配到同一个子进程上，这会造成进程间负载极为不均衡。特别是在使用长连接的时候，单位时间内的new coming connection并不高，子进程的accept queue往往均为空，就会导致connection会不停的分配给同一个进程。所以这种负载均衡完全依赖于accept queue的空闲程度，只有在使用短连接，而且并发非常高的情况下，才能达到负载均衡，但是这个时候系统的load会非常高，系统也会变得不稳定起来。

Nginx是怎么做的?

如果你了解nginx，那么你可能第一时间会想到使用nginx的处理方式，nginx有一个master和多个worker进程，master进程监听端口，负责accept connection，并把accept 的socket发送给各worker进程，由worker进程接收数据并处理。linux下，nginx是使用socketpair建立master和worker进程间的通信，并使用sendmsg、recvmsg等api来传输命令和文件描述符的。那么node.js是否支持这种方案呢？

答案是肯定的，作出这个回答的依据在于node.js的child_process和cluster模块均有一个send方法：child.send(message, [sendHandle])

这个方法的第二个参数就是我们想要传递的socket，而且node.js文档上还给出了一个test case：

 1 var normal = require('child_process').fork('child.js', ['normal']);
 2 var special = require('child_process').fork('child.js', ['special']);
 3 // Open up the server and send sockets to child
 4 var server = require('net').createServer();
 5 server.on('connection', function (socket) {
 6   // if this is a VIP
 7   if (socket.remoteAddress === '74.125.127.100') {
 8     special.send('socket', socket);
 9     return;
10   }
11   // just the usual dudes
12   normal.send('socket', socket);
13 });
14 server.listen(1337);

child.js

1 process.on('message', function(m, socket) {
2   if (m === 'socket') {
3     socket.end('You were handled as a ' + process.argv[2] + ' person');
4   }
5 });

简单，精炼！似乎是一个完美的解决方案。我们稍微加工一下，让他成为一个可以正常运行的http server:

master.js

 1 var http = require('http'),
 2     numCPUs = require('os').cpus().length;
 3     cp = require('child_process'), 
 4     net = require('net');
 5 var workers = [];
 6 for (var i = 0; i < numCPUs; i++) {
 7     workers.push(cp.fork('app.js', ['normal']));
 8 }
 9 
10 net.createServer(function(s) {
11     s.pause();
12     var worker = worker.shift();
13     worker.send('c',s);
14     workers.push(worker);
15 }).listen(80);

 1 var http = require('http'),
 2     cp = require('child_process'),
 3     net = require('net');
 4 var server = http.createServer(function(req,res){
 5     res.writeHead(200, {"Content-Type": "text/plain",         "Connection": "close"});
 6     res.end("hello, world");
 7 });
 8 console.log("webServer started on " + process.pid);
 9 process.on("message", function(msg,socket) {
10     process.nextTick(function(){
11         if(msg == 'c' && socket) {
12             socket.readable = socket.writable = true;
13             socket.resume();
14             server.connections++;
15             socket.server = server;
16             server.emit("connection", socket);
17             socket.emit("connect");
18         }
19     });
20 });
21                                         

我们在worker进程中创建了一个http server，但是这个http server并不监听，也不绑定端口，在收到master传输过来的socket时，调用server.emit("connection", socket);就可以触发server的connection事件了。看起来很不错，简单的测试之后可以正常工作，这个方案几近于完美。在经历过共享监听socket方案的失败后，我们把服务迁移到这种架构上来。

但是，我们遇到了问题。 我们发现master进程的cpu和内存在逐渐增长，并最终到达100%，或者node.js崩溃（Assertion `fd_to_send >= 0' failed），这令我很抓狂，百般无奈之下我们求助于node.js的开发人员，在github上报告了我们遇到的问题（Issue #4587）。就在当天晚上，node.js的开发人员indutny找到了问题所在，主要在于主进程在将socket发送给子进程之后，并没有销毁，而是保留在socketList中，这会导致socket在主进程中逐步累积，并最终达到上限。

indutny很快解决了这个问题，于第二天提交了这个commit，按照这个commit，indutny给send函数增加了第三个可选参数，修改后的send函数将变为：

child.send(message,[socket], [{ track: false, process: false }])

我们的master进程不需要track每个socket状态，所以我们将它设为false即可。到此，这个问题得到了完美的解决，希望这个改进可以随node.js的下一个版本一起发布。

-------------------------------------------

node.js官方于1月18日发布了0.9.7(Unstable)新版本，已经包含了此改进。