blob: 37aea29e1a4263f763e6982b939143cd321e1354 (
plain)
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
|
'use strict';
const common = require('../common');
const NUM_WORKERS = 4;
const PACKETS_PER_WORKER = 10;
const cluster = require('cluster');
const dgram = require('dgram');
if (common.isWindows) {
common.skip('dgram clustering is currently not supported ' +
'on windows.');
return;
}
if (cluster.isMaster)
master();
else
worker();
function master() {
let received = 0;
// Start listening on a socket.
const socket = dgram.createSocket('udp4');
socket.bind(common.PORT);
// Disconnect workers when the expected number of messages have been
// received.
socket.on('message', common.mustCall((data, info) => {
received++;
if (received === PACKETS_PER_WORKER * NUM_WORKERS) {
// Close the socket.
socket.close();
// Disconnect all workers.
cluster.disconnect();
}
}, NUM_WORKERS * PACKETS_PER_WORKER));
// Fork workers.
for (let i = 0; i < NUM_WORKERS; i++)
cluster.fork();
}
function worker() {
// Create udp socket and send packets to master.
const socket = dgram.createSocket('udp4');
const buf = Buffer.from('hello world');
// This test is intended to exercise the cluster binding of udp sockets, but
// since sockets aren't clustered when implicitly bound by at first call of
// send(), explicitly bind them to an ephemeral port.
socket.bind(0);
// There is no guarantee that a sent dgram packet will be received so keep
// sending until disconnect.
const interval = setInterval(() => {
socket.send(buf, 0, buf.length, common.PORT, '127.0.0.1');
}, 1);
cluster.worker.on('disconnect', () => {
clearInterval(interval);
});
}
|