mirror of https://github.com/lukechilds/node.git
Browse Source
This allow the server to be shared without the need to handle connection from masterv0.9.1-release
Andreas Madsen
12 years ago
committed by
isaacs
isaacs
4 changed files with 230 additions and 23 deletions
@ -0,0 +1,182 @@ |
|||
// Copyright Joyent, Inc. and other Node contributors.
|
|||
//
|
|||
// Permission is hereby granted, free of charge, to any person obtaining a
|
|||
// copy of this software and associated documentation files (the
|
|||
// "Software"), to deal in the Software without restriction, including
|
|||
// without limitation the rights to use, copy, modify, merge, publish,
|
|||
// distribute, sublicense, and/or sell copies of the Software, and to permit
|
|||
// persons to whom the Software is furnished to do so, subject to the
|
|||
// following conditions:
|
|||
//
|
|||
// The above copyright notice and this permission notice shall be included
|
|||
// in all copies or substantial portions of the Software.
|
|||
//
|
|||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|||
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|||
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
|
|||
// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
|
|||
// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
|
|||
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
|
|||
// USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|||
|
|||
var common = require('../common'); |
|||
var net = require('net'); |
|||
var fork = require('child_process').fork; |
|||
|
|||
if (process.argv[2] === 'worker') { |
|||
process.once('disconnect', function () { |
|||
process.exit(0); |
|||
}); |
|||
|
|||
process.on('message', function (msg, server) { |
|||
if (msg !== 'server') return; |
|||
|
|||
server.on('connection', function (socket) { |
|||
socket.end(process.argv[3]); |
|||
}); |
|||
|
|||
process.send('listen'); |
|||
}); |
|||
|
|||
process.send('online'); |
|||
return; |
|||
} |
|||
|
|||
var workers = []; |
|||
|
|||
function spawn(server, id, cb) { |
|||
// create a worker
|
|||
var worker = fork(__filename, ['worker', id]); |
|||
workers[id] = worker; |
|||
|
|||
// wait for ready
|
|||
worker.on('message', function (msg) { |
|||
switch (msg) { |
|||
case 'online': |
|||
worker.send('server', server); |
|||
break; |
|||
case 'listen': |
|||
cb(worker); |
|||
break; |
|||
default: |
|||
throw new Error('got wrong message' + msg); |
|||
} |
|||
}); |
|||
|
|||
return worker; |
|||
} |
|||
|
|||
// create a server instance there don't take connections
|
|||
var server = net.createServer().listen(common.PORT, function () { |
|||
|
|||
setTimeout(function() { |
|||
console.log('spawn worker#0'); |
|||
spawn(server, 0, function () { |
|||
console.log('worker#0 spawned'); |
|||
}); |
|||
}, 250); |
|||
|
|||
console.log('testing for standby, expect id 0'); |
|||
testResponse([0], function () { |
|||
|
|||
// test with two worker, expect id response to be 0 or 1
|
|||
testNewWorker(1, [0, 1], function () { |
|||
|
|||
// kill worker#0, expect id response to be 1
|
|||
testWorkerDeath(0, [1], function () { |
|||
|
|||
// close server, expect all connections to continue
|
|||
testServerClose(server, [1], function () { |
|||
|
|||
// killing worker#1, expect all connections to fail
|
|||
testWorkerDeath(1, [], function () { |
|||
console.log('test complete'); |
|||
}); |
|||
}); |
|||
}); |
|||
}); |
|||
}); |
|||
}); |
|||
|
|||
function testServerClose(server, expect, cb) { |
|||
console.log('closeing server'); |
|||
server.close(function () { |
|||
testResponse(expect, cb); |
|||
}); |
|||
} |
|||
|
|||
function testNewWorker(workerId, exptect, cb) { |
|||
console.log('spawning worker#' + workerId); |
|||
spawn(server, 1, function () { |
|||
testResponse(exptect, cb); |
|||
}); |
|||
} |
|||
|
|||
function testWorkerDeath(workerID, exptect, cb) { |
|||
// killing worker#1, expect all connections to fail
|
|||
console.log('killing worker#' + workerID); |
|||
workers[workerID].kill(); |
|||
workers[workerID].once('exit', function () { |
|||
testResponse(exptect, cb); |
|||
}); |
|||
} |
|||
|
|||
function testResponse(expect, cb) { |
|||
// make 25 connections
|
|||
var count = 25; |
|||
var missing = expect.slice(0); |
|||
|
|||
var i = count; |
|||
while (i--) { |
|||
makeConnection(function (error, id) { |
|||
if (expect.length === 0) { |
|||
if (error === null) { |
|||
throw new Error('connect should not be possible'); |
|||
} |
|||
} else { |
|||
if (expect.indexOf(id) === -1) { |
|||
throw new Error('got unexpected response: ' + id + |
|||
', expected: ' + expect.join(', ')); |
|||
} |
|||
|
|||
var index = missing.indexOf(id); |
|||
if (index !== -1) missing.splice(index, 1); |
|||
} |
|||
|
|||
count -= 1; |
|||
if (count === 0) { |
|||
if (missing.length !== 0) { |
|||
throw new Error('no connection responsed with the following id: ' + |
|||
missing.join(', ')); |
|||
} |
|||
cb(); |
|||
} |
|||
}); |
|||
} |
|||
} |
|||
|
|||
function makeConnection(cb) { |
|||
var client = net.connect({ port: common.PORT }); |
|||
|
|||
var id = null; |
|||
client.once('data', function(data) { |
|||
id = parseInt(data, 10); |
|||
}); |
|||
|
|||
var error = null; |
|||
client.once('error', function (e) { |
|||
error = e; |
|||
}); |
|||
|
|||
client.setTimeout(500); |
|||
|
|||
client.on('close', function () { |
|||
cb(error, id); |
|||
}); |
|||
} |
|||
|
|||
process.on('exit', function () { |
|||
workers.forEach(function (worker) { |
|||
try { worker.kill(); } catch(e) {} |
|||
}); |
|||
}); |
|||
Loading…
Reference in new issue