refactor: wip

Author: BioPhoton

packages/utils/mocks/multiprocess-profiling/profiler-worker-child.mjs

@@ -0,0 +1,23 @@
+import { NodejsProfiler } from '../../src/lib/profiler/profiler-node.js';
+import { entryToTraceEvents } from '../../src/lib/profiler/trace-file-utils.js';
+import { traceEventWalFormat } from '../../src/lib/profiler/wal-json-trace.js';
+
+(async () => {
+  const profiler = new NodejsProfiler({
+    format: {
+      ...traceEventWalFormat(),
+      encodePerfEntry: entryToTraceEvents,
+    },
+  });
+
+  profiler.marker(`process-${processId}:process-start`, {
+    tooltipText: `Process ${processId} started`,
+  });
+
+  profiler.measure(`process-${processId}:work`, () => {
+    const arr = Array.from({ length: 1000 }, (_, i) => i);
+    return arr.reduce((sum, x) => sum + x, 0);
+  });
+
+  profiler.close();
+})();

packages/utils/mocks/multiprocess-profiling/profiler-worker.mjs

@@ -0,0 +1,56 @@
+import { spawn } from 'node:child_process';
+import path from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { NodejsProfiler } from '../../src/lib/profiler/profiler-node.js';
+import { entryToTraceEvents } from '../../src/lib/profiler/trace-file-utils.js';
+import { traceEventWalFormat } from '../../src/lib/profiler/wal-json-trace.js';
+
+const [numProcesses] = process.argv.slice(2);
+
+if (!numProcesses) {
+  console.error('Usage: node profiler-worker.mjs <numProcesses>');
+  process.exit(1);
+}
+
+const numProcs = parseInt(numProcesses, 10);
+if (isNaN(numProcs) || numProcs < 1) {
+  console.error('numProcesses must be a positive integer');
+  process.exit(1);
+}
+
+const workerScriptPath = path.join(
+  fileURLToPath(path.dirname(import.meta.url)),
+  './profiler-worker-child.mjs',
+);
+
+const profiler = new NodejsProfiler({
+  format: {
+    ...traceEventWalFormat(),
+    encodePerfEntry: entryToTraceEvents,
+  },
+});
+
+(async () => {
+  const processes = Array.from({ length: numProcs }, (_, i) => {
+    return new Promise((resolve, reject) => {
+      const child = spawn('npx', ['tsx', workerScriptPath], {
+        stdio: 'pipe',
+      });
+
+      child.on('close', code => {
+        if (code === 0) {
+          resolve(code);
+        } else {
+          reject(new Error(`Process ${i + 1} exited with code ${code}`));
+        }
+      });
+
+      child.on('error', reject);
+    });
+  });
+
+  await Promise.all(processes);
+
+  profiler.close();
+  console.log(JSON.stringify(profiler.stats, null, 2));
+})();