benchmark: include writev in benchmark

Currently we only consider write when benchmarking. PR-URL: #31066 Reviewed-By: Denys Otrishko <[email protected]> Reviewed-By: Anna Henningsen <[email protected]> Reviewed-By: Ruben Bridgewater <[email protected]>
nodejs · Jan 3, 2020 · 01fd3be · 01fd3be
1 parent 647f3c7
commit 01fd3be
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 4 deletions.
diff --git a/benchmark/streams/writable-manywrites.js b/benchmark/streams/writable-manywrites.js
@@ -5,25 +5,36 @@ const Writable = require('stream').Writable;
 
 const bench = common.createBenchmark(main, {
   n: [2e6],
-  sync: ['yes', 'no']
+  sync: ['yes', 'no'],
+  writev: ['yes', 'no'],
+  callback: ['yes', 'no']
 });
 
-function main({ n, sync }) {
+function main({ n, sync, writev, callback }) {
   const b = Buffer.allocUnsafe(1024);
   const s = new Writable();
   sync = sync === 'yes';
-  s._write = function(chunk, encoding, cb) {
+
+  const writecb = (cb) => {
     if (sync)
       cb();
     else
       process.nextTick(cb);
   };
 
+  if (writev === 'yes') {
+    s._writev = (chunks, cb) => writecb(cb);
+  } else {
+    s._write = (chunk, encoding, cb) => writecb(cb);
+  }
+
+  const cb = callback === 'yes' ? () => {} : null;
+
   bench.start();
 
   let k = 0;
   function run() {
-    while (k++ < n && s.write(b));
+    while (k++ < n && s.write(b, cb));
     if (k >= n)
       bench.end(n);
   }

diff --git a/test/benchmark/test-benchmark-streams.js b/test/benchmark/test-benchmark-streams.js
@@ -9,6 +9,8 @@ runBenchmark('streams',
                'kind=duplex',
                'n=1',
                'sync=no',
+               'writev=no',
+               'callback=no',
                'type=buffer',
              ],
              { NODEJS_BENCHMARK_ZERO_ALLOWED: 1 });