Revision 2dd1c1ca9b05a07f3fd96c0093bc356fca457868 authored by einaros on 10 May 2012, 19:24:45 UTC, committed by einaros on 10 May 2012, 19:24:45 UTC
1 parent e4f6d07
speed.js
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);
})();
}
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...