1 |
3a515b92
|
cagy
|
module.exports = function makeKillable(server) {
|
2 |
|
|
var sockets = [];
|
3 |
|
|
|
4 |
|
|
server.on('connection', function (socket) {
|
5 |
|
|
//add socket to list
|
6 |
|
|
sockets.push(socket);
|
7 |
|
|
|
8 |
|
|
socket.once('close', function () {
|
9 |
|
|
//remove socket from list
|
10 |
|
|
sockets.splice(sockets.indexOf(socket), 1);
|
11 |
|
|
});
|
12 |
|
|
});
|
13 |
|
|
|
14 |
|
|
server.kill = function (cb) {
|
15 |
|
|
server.close(cb);
|
16 |
|
|
sockets.forEach(function (socket) {
|
17 |
|
|
socket.destroy();
|
18 |
|
|
});
|
19 |
|
|
// reset so the server can be restarted
|
20 |
|
|
sockets = [];
|
21 |
|
|
};
|
22 |
|
|
|
23 |
|
|
return server;
|
24 |
|
|
};
|