-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathForkPool.js
More file actions
73 lines (59 loc) · 2 KB
/
ForkPool.js
File metadata and controls
73 lines (59 loc) · 2 KB
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
70
71
72
73
function ForkPool(maxSize) {
const self = ForkPool;
const queue = [];
const childProcess = require('child_process');
var size = 0;
ForkPool.maxSize = maxSize;
ForkPool.fork = function (forklet, callback) {
if (size >= maxSize) {
queue.push({
forklet: forklet,
callback: callback
});
if (forklet.onScheduledCallback) {
forklet.onScheduledCallback();
}
if (callback) {
callback({name: 'scheduled'});
}
return;
}
size++;
var cproc = childProcess.fork(forklet.moduleName, forklet.envars);
if (forklet.timeout && forklet.timeout > 0) {
(function (forklet, processObject, callback) {
setTimeout(function () {
processObject.kill('SIGKILL');
if (forklet.onTimedoutCallback) {
forklet.onTimedoutCallback(processObject);
}
if (callback) {
callback({name: 'timedout'}, processObject);
}
}, forklet.timeout);
})(forklet, cproc, callback);
}
cproc.on('exit', function (code, signal) {
size--;
if (forklet.onExitedCallback) {
forklet.onExitedCallback({code: code, signal: signal}, cproc);
}
if (callback) {
callback({name: 'exited', stateParams: {code: code, signal: signal}}, cproc);
}
if (queue.length > 0) {
var queueItem = queue[0];
queue.splice(0, 1);
self.fork(queueItem.forklet, queueItem.callback);
}
});
if (forklet.onStartedCallback) {
forklet.onStartedCallback(cproc);
}
if (callback) {
callback({name: 'started'}, cproc);
}
};
return ForkPool;
}
module.exports = ForkPool;