-
Notifications
You must be signed in to change notification settings - Fork 0
/
clusters.js
97 lines (77 loc) · 2.33 KB
/
clusters.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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
'use strict';
const cluster = require('cluster');
const fs = require('fs');
const path = require('path');
class Workers {
static get watchFileExtensions() {
return ['.js', '.json'];
}
static get ignoreFiles() {
return [];
}
static get ignoreDirectories() {
return ['node_modules', '.git', 'data'];
}
static restart (file) {
console.log(`Restarting workers due to file change: ${file}`);
Object.keys(cluster.workers).forEach((workerID) => {
cluster.workers[workerID].send({
type: 'shutdown',
from: 'master'
});
setTimeout((workerID) => {
if (cluster.workers[workerID])
cluster.workers[workerID].kill('SIGKILL');
}, 6000);
});
}
static watchPathForChanges (currentPath) {
fs.readdir(currentPath, (err, files) => {
if (!err) {
for (let file of files) {
let currentFile = `${currentPath}/${file}`;
if (fs.lstatSync(currentFile).isDirectory() && !this.ignoreDirectories.includes(file)) {
this.watchPathForChanges(currentFile);
} else if (this.watchFileExtensions.includes(path.extname(file)) && !this.ignoreFiles.includes(file)) {
fs.watch(currentFile, () => {
clearTimeout(this.restartTimeout);
this.restartTimeout = setTimeout((currentFile) => {
this.restart(currentFile);
}, 2000, currentFile);
});
}
}
}
});
}
constructor() {
if (cluster.isMaster) {
const cpus = [1];
console.log(`Setting up ${cpus.length} workers.`);
for (let _ of cpus)
this.fork();
Workers.watchPathForChanges('./');
cluster.on('exit', (deadWorker, code, signal) => {
console.log(`Worker ${deadWorker.process.pid} died: ${signal || code || 'shutdown'}`);
this.fork();
});
} else {
process.on('message', (message) => {
switch (message.type) {
case 'shutdown':
process.exit(0);
}
});
console.log(`Worker ${process.pid} is alive.`);
}
}
fork() {
console.log('Starting new worker.');
cluster.fork().on('message', (message) => {
if (message.relay)
for (let [, worker] of Object.entries(cluster.workers))
worker.send(message);
});
}
}
module.exports = Workers;