From b0e6f10530e5ae2168ff7edd7fb9c6ffa8a8fbe5 Mon Sep 17 00:00:00 2001
From: Anna Henningsen <anna@addaleax.net>
Date: Sat, 14 Apr 2018 18:26:57 +0200
Subject: [PATCH] benchmark: add bench for zlib gzip + gunzip cycle
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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>
---
 benchmark/zlib/pipe.js               | 39 ++++++++++++++++++++++++++++
 test/parallel/test-benchmark-zlib.js |  9 +++++--
 2 files changed, 46 insertions(+), 2 deletions(-)
 create mode 100644 benchmark/zlib/pipe.js

diff --git a/benchmark/zlib/pipe.js b/benchmark/zlib/pipe.js
new file mode 100644
index 00000000000000..9b05749bbb824d
--- /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);
+}
diff --git a/test/parallel/test-benchmark-zlib.js b/test/parallel/test-benchmark-zlib.js
index 350d05552cda39..25b7d1a4d5f905 100644
--- a/test/parallel/test-benchmark-zlib.js
+++ b/test/parallel/test-benchmark-zlib.js
@@ -9,5 +9,10 @@ runBenchmark('zlib',
                'method=deflate',
                'n=1',
                'options=true',
-               'type=Deflate'
-             ]);
+               'type=Deflate',
+               'inputLen=1024',
+               'duration=0.001'
+             ],
+             {
+               'NODEJS_BENCHMARK_ZERO_ALLOWED': 1
+             });