Skip to content

Commit

Permalink
benchmark: add JSStreamWrap benchmark
Browse files Browse the repository at this point in the history
This is made to mirror the connection-bound net benchmarks.

PR-URL: #17983
Reviewed-By: Rich Trott <rtrott@gmail.com>
Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
Reviewed-By: Luigi Pinca <luigipinca@gmail.com>
Reviewed-By: James M Snell <jasnell@gmail.com>
  • Loading branch information
addaleax authored and rvagg committed Aug 16, 2018
1 parent 1942440 commit 0a3ebb0
Showing 1 changed file with 107 additions and 0 deletions.
107 changes: 107 additions & 0 deletions benchmark/net/net-wrap-js-stream-passthrough.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,107 @@
// test the speed of .pipe() with JSStream wrapping for PassThrough streams
'use strict';

const common = require('../common.js');
const { PassThrough } = require('stream');

const bench = common.createBenchmark(main, {
len: [102400, 1024 * 1024 * 16],
type: ['utf', 'asc', 'buf'],
dur: [5],
}, {
flags: ['--expose-internals']
});

var dur;
var len;
var type;
var chunk;
var encoding;
var JSStreamWrap; // Can only require internals inside main().

function main(conf) {
dur = +conf.dur;
len = +conf.len;
type = conf.type;
JSStreamWrap = require('internal/wrap_js_stream');

switch (type) {
case 'buf':
chunk = Buffer.alloc(len, 'x');
break;
case 'utf':
encoding = 'utf8';
chunk = 'ü'.repeat(len / 2);
break;
case 'asc':
encoding = 'ascii';
chunk = 'x'.repeat(len);
break;
default:
throw new Error(`invalid type: ${type}`);
}

doBenchmark();
}

function Writer() {
this.received = 0;
this.writable = true;
}

Writer.prototype.write = function(chunk, encoding, cb) {
this.received += chunk.length;

if (typeof encoding === 'function')
encoding();
else if (typeof cb === 'function')
cb();

return true;
};

// doesn't matter, never emits anything.
Writer.prototype.on = function() {};
Writer.prototype.once = function() {};
Writer.prototype.emit = function() {};
Writer.prototype.prependListener = function() {};


function flow() {
const dest = this.dest;
const res = dest.write(chunk, encoding);
if (!res)
dest.once('drain', this.flow);
else
process.nextTick(this.flow);
}

function Reader() {
this.flow = flow.bind(this);
this.readable = true;
}

Reader.prototype.pipe = function(dest) {
this.dest = dest;
this.flow();
return dest;
};


function doBenchmark() {
const reader = new Reader();
const writer = new Writer();

// the actual benchmark.
const fakeSocket = new JSStreamWrap(new PassThrough());
bench.start();
reader.pipe(fakeSocket);
fakeSocket.pipe(writer);

setTimeout(function() {
const bytes = writer.received;
const gbits = (bytes * 8) / (1024 * 1024 * 1024);
bench.end(gbits);
process.exit(0);
}, dur * 1000);
}

0 comments on commit 0a3ebb0

Please sign in to comment.