From 79e0bf9b646397303a27231031681815559e51ba Mon Sep 17 00:00:00 2001
From: Rafael Gonzaga <rafael.nunu@hotmail.com>
Date: Wed, 21 Dec 2022 13:09:43 -0300
Subject: [PATCH] benchmark: include webstreams benchmark

Signed-off-by: RafaelGSS <rafael.nunu@hotmail.com>
PR-URL: https://github.com/nodejs/node/pull/45876
Reviewed-By: Yagiz Nizipli <yagiz@nizipli.com>
Reviewed-By: Paolo Insogna <paolo@cowtech.it>
---
 benchmark/webstreams/creation.js              | 49 +++++++++++++++++++
 benchmark/webstreams/pipe-to.js               | 36 ++++++++++++++
 .../webstreams/readable-async-iterator.js     | 31 ++++++++++++
 test/benchmark/test-benchmark-webstreams.js   |  7 +++
 4 files changed, 123 insertions(+)
 create mode 100644 benchmark/webstreams/creation.js
 create mode 100644 benchmark/webstreams/pipe-to.js
 create mode 100644 benchmark/webstreams/readable-async-iterator.js
 create mode 100644 test/benchmark/test-benchmark-webstreams.js

diff --git a/benchmark/webstreams/creation.js b/benchmark/webstreams/creation.js
new file mode 100644
index 00000000000000..085faba8b89864
--- /dev/null
+++ b/benchmark/webstreams/creation.js
@@ -0,0 +1,49 @@
+'use strict';
+const common = require('../common.js');
+const {
+  ReadableStream,
+  TransformStream,
+  WritableStream,
+} = require('node:stream/web');
+const assert = require('assert');
+
+const bench = common.createBenchmark(main, {
+  n: [50e3],
+  kind: ['ReadableStream', 'TransformStream', 'WritableStream']
+});
+
+let rs, ws, ts;
+
+function main({ n, kind }) {
+  switch (kind) {
+    case 'ReadableStream':
+      bench.start();
+      for (let i = 0; i < n; ++i)
+        rs = new ReadableStream();
+      bench.end(n);
+
+      // Avoid V8 deadcode (elimination)
+      assert.ok(rs);
+      break;
+    case 'WritableStream':
+      bench.start();
+      for (let i = 0; i < n; ++i)
+        ws = new WritableStream();
+      bench.end(n);
+
+      // Avoid V8 deadcode (elimination)
+      assert.ok(ws);
+      break;
+    case 'TransformStream':
+      bench.start();
+      for (let i = 0; i < n; ++i)
+        ts = new TransformStream();
+      bench.end(n);
+
+      // Avoid V8 deadcode (elimination)
+      assert.ok(ts);
+      break;
+    default:
+      throw new Error('Invalid kind');
+  }
+}
diff --git a/benchmark/webstreams/pipe-to.js b/benchmark/webstreams/pipe-to.js
new file mode 100644
index 00000000000000..a41b31b5e127ec
--- /dev/null
+++ b/benchmark/webstreams/pipe-to.js
@@ -0,0 +1,36 @@
+'use strict';
+const common = require('../common.js');
+const {
+  ReadableStream,
+  WritableStream,
+} = require('node:stream/web');
+
+const bench = common.createBenchmark(main, {
+  n: [5e6],
+  highWaterMarkR: [512, 1024, 2048, 4096],
+  highWaterMarkW: [512, 1024, 2048, 4096],
+});
+
+
+async function main({ n, highWaterMarkR, highWaterMarkW }) {
+  const b = Buffer.alloc(1024);
+  let i = 0;
+  const rs = new ReadableStream({
+    highWaterMark: highWaterMarkR,
+    pull: function(controller) {
+      if (i++ === n) {
+        controller.enqueue(b);
+      } else {
+        controller.close();
+      }
+    }
+  });
+  const ws = new WritableStream({
+    highWaterMark: highWaterMarkW,
+    write(chunk, controller) {},
+    close() { bench.end(n); },
+  });
+
+  bench.start();
+  rs.pipeTo(ws);
+}
diff --git a/benchmark/webstreams/readable-async-iterator.js b/benchmark/webstreams/readable-async-iterator.js
new file mode 100644
index 00000000000000..0d7e4737e3a15a
--- /dev/null
+++ b/benchmark/webstreams/readable-async-iterator.js
@@ -0,0 +1,31 @@
+'use strict';
+const common = require('../common.js');
+const {
+  ReadableStream,
+} = require('node:stream/web');
+
+const bench = common.createBenchmark(main, {
+  n: [1e5],
+});
+
+
+async function main({ n }) {
+  const rs = new ReadableStream({
+    pull: function(controller) {
+      controller.enqueue(1);
+    }
+  });
+
+  let x = 0;
+
+  bench.start();
+  for await (const chunk of rs) {
+    x += chunk;
+    if (x > n) {
+      break;
+    }
+  }
+  // Use x to ensure V8 does not optimize away the loop as a noop.
+  console.assert(x);
+  bench.end(n);
+}
diff --git a/test/benchmark/test-benchmark-webstreams.js b/test/benchmark/test-benchmark-webstreams.js
new file mode 100644
index 00000000000000..c4146c859a98b9
--- /dev/null
+++ b/test/benchmark/test-benchmark-webstreams.js
@@ -0,0 +1,7 @@
+'use strict';
+
+require('../common');
+
+const runBenchmark = require('../common/benchmark');
+
+runBenchmark('webstreams', { NODEJS_BENCHMARK_ZERO_ALLOWED: 1 });