-
-
Notifications
You must be signed in to change notification settings - Fork 35.4k
Expand file tree
/
Copy pathtest-cluster-port-reuse-between-workers.js
More file actions
93 lines (86 loc) · 2.43 KB
/
test-cluster-port-reuse-between-workers.js
File metadata and controls
93 lines (86 loc) · 2.43 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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
'use strict';
const common = require('../common');
const cluster = require('cluster');
const assert = require('assert');
const acts = {
WORKER1_SERVER1_CLOSED: { cmd: 'WORKER1_SERVER1_CLOSED' },
WORKER2_SERVER1_STARTED: { cmd: 'WORKER2_SERVER1_STARTED' },
WORKER1_SERVER2_CLOSED: { cmd: 'WORKER1_SERVER2_CLOSED' },
};
if (cluster.isMaster) {
const currentHost = '::';
const worker1 = cluster.fork({
WORKER_ID: 'worker1',
HOST: currentHost,
});
let worker2;
worker1.on('error', common.mustNotCall());
worker1.on('message', onMessage);
function createWorker2() {
worker2 = cluster.fork({
WORKER_ID: 'worker2',
HOST: currentHost,
});
worker2.on('error', common.mustNotCall());
worker2.on('message', onMessage);
}
function onMessage(msg) {
switch (msg.cmd) {
case acts.WORKER1_SERVER1_CLOSED.cmd:
createWorker2();
break;
case acts.WORKER2_SERVER1_STARTED.cmd:
worker1.send(acts.WORKER2_SERVER1_STARTED);
break;
case acts.WORKER1_SERVER2_CLOSED.cmd:
worker1.kill();
worker2.kill();
break;
default:
assert.fail(`Unexpected message ${msg.cmd}`);
}
}
} else {
const WORKER_ID = process.env.WORKER_ID;
function createServer() {
return new Promise((resolve, reject) => {
const net = require('net');
const PORT = 8000;
const server = net
.createServer((socket) => {
socket.end(
`Handled by worker ${process.env.WORKER_ID} (${process.pid})\n`
);
})
.on('error', (e) => {
reject(e);
});
server.listen(
{
port: PORT,
host: process.env.HOST,
},
() => resolve(server)
);
});
}
(async () => {
const server1 = await createServer();
if (WORKER_ID === 'worker2') {
process.send(acts.WORKER2_SERVER1_STARTED);
} else {
await createServer().catch(common.mustCall());
await new Promise((r) => server1.close(r));
process.send(acts.WORKER1_SERVER1_CLOSED);
process.on('message', async (msg) => {
if (msg.cmd === acts.WORKER2_SERVER1_STARTED.cmd) {
const server2 = await createServer();
await new Promise((r) => server2.close(r));
process.send(acts.WORKER1_SERVER2_CLOSED);
} else {
assert.fail(`Unexpected message ${msg.cmd}`);
}
});
}
})().then(common.mustCall());
}