benchmark: add bench for zlib gzip + gunzip cycle

Originally wrote this for some work that is going to take a while
longer before it’s ready to be PR’ed, so it seems fine to start
with this on its own.

PR-URL: https://github.com/nodejs/node/pull/20034
Reviewed-By: James M Snell <jasnell@gmail.com>
Reviewed-By: Luigi Pinca <luigipinca@gmail.com>
Reviewed-By: Ruben Bridgewater <ruben@bridgewater.de>

1 files changed, 39 insertions, 0 deletions

diff --git a/benchmark/zlib/pipe.js b/benchmark/zlib/pipe.js
new file mode 100644
index 0000000000..9b05749bbb
--- /dev/null
+++ b/benchmark/zlib/pipe.js
@@ -0,0 +1,39 @@
+'use strict';
+const common = require('../common.js');
+const fs = require('fs');
+const zlib = require('zlib');
+
+const bench = common.createBenchmark(main, {
+  inputLen: [1024],
+  duration: [5],
+  type: ['string', 'buffer']
+});
+
+function main({ inputLen, duration, type }) {
+  const buffer = Buffer.alloc(inputLen, fs.readFileSync(__filename));
+  const chunk = type === 'buffer' ? buffer : buffer.toString('utf8');
+
+  const input = zlib.createGzip();
+  const output = zlib.createGunzip();
+
+  let readFromOutput = 0;
+  input.pipe(output);
+  if (type === 'string')
+    output.setEncoding('utf8');
+  output.on('data', (chunk) => readFromOutput += chunk.length);
+
+  function write() {
+    input.write(chunk, write);
+  }
+
+  bench.start();
+  write();
+
+  setTimeout(() => {
+    // Give result in GBit/s, like the net benchmarks do
+    bench.end(readFromOutput * 8 / (1024 ** 3));
+
+    // Cut off writing the easy way.
+    input.write = () => {};
+  }, duration * 1000);
+}