fix: uniteTraceEvents behavior (#3)

src/__tests__/end-to-end.ts

@@ -215,6 +215,8 @@ describe('end-to-end', () => {
     `);
   });
 
+  // TODO: write a test which will fail on the default highWaterMark
+
   test('should unite trace events', async () => {
     const file3 = tempy.file();
 

src/streams/trace-merge/transforms/TraceAnalyze.ts

@@ -1,17 +1,21 @@
-import { Transform } from 'node:stream';
+import { Writable } from 'node:stream';
 import type { TraceEvent } from 'trace-event-lib';
 import type { Resolver } from '../resolvers';
 import type { JSONLEntry } from '../../jsonl';
 
-export class TraceAnalyze extends Transform {
+export class TraceAnalyze extends Writable {
   readonly #resolver: Resolver;
 
   constructor(resolver: Resolver) {
-    super({ objectMode: true });
+    super({
+      objectMode: true,
+      highWaterMark: Number.MAX_SAFE_INTEGER,
+    });
+
     this.#resolver = resolver;
   }
 
-  _transform(
+  _write(
     chunk: unknown,
     _encoding: string,
     callback: (error?: Error | null, data?: unknown) => void,

src/streams/trace-merge/transforms/TraceMerge.ts

@@ -9,7 +9,11 @@ export class TraceMerge extends Transform {
   #resolver?: Resolver;
 
   constructor(resolverPromise: Promise<Resolver>) {
-    super({ objectMode: true });
+    super({
+      objectMode: true,
+      highWaterMark: Number.MAX_SAFE_INTEGER,
+    });
+
     this.#resolverPromise = resolverPromise;
   }