105 lines
2.8 KiB
JavaScript
105 lines
2.8 KiB
JavaScript
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);
|
|
})();
|
|
} |