initial import

bench/index.js

@@ -0,0 +1,84 @@
+import { bench, run } from './benchlib.js';
+import takeUntil from '../index.js';
+
+function createSource() {
+  let index = 0;
+  let sink = null;
+  return {
+    pipe(s) { sink = s; return s; },
+    resume() {
+      while (sink && !sink.paused && index < 10000) {
+        sink.write(index++);
+      }
+    },
+    write() {},
+    end() {},
+    abort() {},
+    get sink() { return sink; },
+    get paused() { return sink ? sink.paused : true; }
+  };
+}
+
+function createCollectSink() {
+  const items = [];
+  return {
+    items,
+    write(data) { items.push(data); },
+    end() {},
+    abort() {},
+    paused: false,
+    ended: false
+  };
+}
+
+function createSignalStream() {
+  let sink = null;
+  return {
+    pipe(s) { sink = s; return s; },
+    write() {},
+    end() { if (sink) sink.end(); },
+    abort() {},
+    get paused() { return false; },
+    trigger() { if (sink) sink.end(); }
+  };
+}
+
+async function benchmark() {
+  await bench('takeUntil with early signal (10000 items)', () => {
+    const sink = createCollectSink();
+    const signal = createSignalStream();
+    const stream = takeUntil(signal);
+    stream.pipe(sink);
+    const source = createSource();
+    source.pipe(stream);
+    source.resume();
+    signal.trigger();
+    return sink.items.length;
+  }, { iterations: 20, warmup: 2, itemsPerIter: 10000, unit: 'items/s' });
+
+  await bench('takeUntil with late signal (10000 items)', () => {
+    const sink = createCollectSink();
+    const signal = createSignalStream();
+    const stream = takeUntil(signal);
+    stream.pipe(sink);
+    const source = createSource();
+    source.pipe(stream);
+    source.resume();
+    return sink.items.length;
+  }, { iterations: 20, warmup: 2, itemsPerIter: 10000, unit: 'items/s' });
+
+  await bench('takeUntil immediate signal (0 items)', () => {
+    const sink = createCollectSink();
+    const signal = createSignalStream();
+    const stream = takeUntil(signal);
+    stream.pipe(sink);
+    const source = createSource();
+    source.pipe(stream);
+    signal.trigger();
+    return sink.items.length;
+  }, { iterations: 100, warmup: 5 });
+
+  await run({ output: './benchmark-results.json' });
+}
+
+benchmark().catch(console.error);