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
98
99
100
101
102
103
104
105
|
var cluster = require('cluster')
, WebSocket = require('../')
, WebSocketServer = WebSocket.Server
, crypto = require('crypto')
, util = require('util')
, ansi = require('ansi');
require('tinycolor');
function roundPrec(num, prec) {
var mul = Math.pow(10, prec);
return Math.round(num * mul) / mul;
}
function humanSize(bytes) {
if (bytes >= 1048576) return roundPrec(bytes / 1048576, 2) + ' MB';
if (bytes >= 1024) return roundPrec(bytes / 1024, 2) + ' kB';
return roundPrec(bytes, 2) + ' B';
}
function generateRandomData(size) {
var buffer = new Buffer(size);
for (var i = 0; i < size; ++i) {
buffer[i] = ~~(Math.random() * 127);
}
return buffer;
}
if (cluster.isMaster) {
var wss = new WebSocketServer({port: 8181}, function() {
cluster.fork();
});
wss.on('connection', function(ws) {
ws.on('message', function(data, flags) {
ws.send(data, {binary: flags&&flags.binary});
});
ws.on('close', function() {});
});
cluster.on('death', function(worker) {
wss.close();
});
}
else {
var cursor = ansi(process.stdout);
var configs = [
[true, 10000, 64],
[true, 5000, 16*1024],
[true, 1000, 128*1024],
[true, 100, 1024*1024],
[true, 1, 500*1024*1024],
[false, 10000, 64],
[false, 5000, 16*1024],
[false, 1000, 128*1024],
[false, 100, 1024*1024],
];
var largest = configs[0][1];
for (var i = 0, l = configs.length; i < l; ++i) {
if (configs[i][2] > largest) largest = configs[i][2];
}
console.log('Generating %s of test data ...', humanSize(largest));
var randomBytes = generateRandomData(largest);
function roundtrip(useBinary, roundtrips, size, cb) {
var data = randomBytes.slice(0, size);
var prefix = util.format('Running %d roundtrips of %s %s data', roundtrips, humanSize(size), useBinary ? 'binary' : 'text');
console.log(prefix);
var client = new WebSocket('ws://localhost:' + '8181');
var dt;
var roundtrip = 0;
function send() {
client.send(data, {binary: useBinary});
}
client.on('error', function(e) {
console.error(e);
process.exit();
});
client.on('open', function() {
dt = Date.now();
send();
});
client.on('message', function(data, flags) {
if (++roundtrip == roundtrips) {
var elapsed = Date.now() - dt;
cursor.up();
console.log('%s:\t%ss\t%s'
, useBinary ? prefix.green : prefix.cyan
, roundPrec(elapsed / 1000, 1).toString().green.bold
, (humanSize((size * roundtrips) / elapsed * 1000) + '/s').blue.bold);
client.close();
cb();
return;
}
process.nextTick(send);
});
}
(function run() {
if (configs.length == 0) process.exit();
var config = configs.shift();
config.push(run);
roundtrip.apply(null, config);
})();
}
|
