-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathindex.js
43 lines (33 loc) · 915 Bytes
/
index.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
var cluster = require('cluster');
var os = require('os');
function clustastic(main, options) {
options = Object.assign({
logging: console.log,
workers: os.cpus().length
}, options);
if (!Number.isInteger(options.workers) ||
options.workers < 0 ||
options.workers > os.cpus().length) {
throw new Error('Invalid workers amount');
}
if (cluster.isMaster) {
for (var i = 0; i < options.workers; i++) {
cluster.fork();
}
cluster.on('exit', function (worker) {
if (typeof options.logging == 'function') {
options.logging('Worker %d exited', worker.id);
}
cluster.fork();
});
} else {
main();
if (typeof options.logging == 'function') {
options.logging('Worker %d running', cluster.worker.id);
}
};
};
clustastic.workerId = function getWorkerId() {
return cluster.isMaster ? null : cluster.worker.id;
};
module.exports = clustastic;