-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathnombo-balancer.node.js
69 lines (62 loc) · 1.42 KB
/
nombo-balancer.node.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
var cluster = require('cluster');
var LoadBalancer = require('loadbalancer');
var balancer;
if (cluster.isMaster) {
process.on('message', function (m) {
var balancers;
if (m.type == 'init') {
var balancerCount = m.data.balancerCount;
balancers = [];
var launchBalancer = function (i) {
balancer = cluster.fork();
balancers[i] = balancer;
balancer.on('error', function (err) {
process.send({
message: err.message,
stack: err.stack
});
});
balancer.on('message', process.send.bind(process));
balancer.on('exit', function () {
launchBalancer(i);
})
balancer.send(m);
};
for (var i=0; i<balancerCount; i++) {
launchBalancer(i);
}
} else {
for (var i in balancers) {
balancers[i].send(m);
}
}
});
} else {
var handleError = function (err, notice) {
var error;
if (err.stack) {
error = {
message: err.message,
stack: err.stack
};
} else {
error = err;
}
process.send({type: notice ? 'notice' : 'error', data: error});
if (err.code != 'ECONNRESET') {
process.exit();
}
};
var handleNotice = function (err) {
handleError(err, true);
};
process.on('message', function (m) {
if (m.type == 'init') {
balancer = new LoadBalancer(m.data);
balancer.on('error', handleError);
balancer.on('notice', handleNotice);
} else if (m.type == 'setWorkers') {
balancer.setWorkers(m.data);
}
});
}