feat(tinybench-plugin): allow perf profiling in tinybench walltime

Author: GuillaumeLagrange

packages/tinybench-plugin/src/walltime.ts

@@ -1,5 +1,6 @@
 import {
   calculateQuantiles,
+  InstrumentHooks,
   mongoMeasurement,
   msToNs,
   msToS,
@@ -17,6 +18,7 @@ export function runWalltimeBench(bench: Bench, rootCallingFile: string): void {
     console.log(
       `[CodSpeed] running with @codspeed/tinybench v${__VERSION__} (walltime mode)`
     );
+    InstrumentHooks.setIntegration("codspeed-node", __VERSION__);
 
     // Store the original run method before we override it
     const originalRun = bench.run;
@@ -40,7 +42,9 @@ export function runWalltimeBench(bench: Bench, rootCallingFile: string): void {
         await task.warmup();
       }
       await mongoMeasurement.start(uri);
+      InstrumentHooks.startBenchmark();
       const taskResult = await task.run();
+      InstrumentHooks.stopBenchmark();
       await mongoMeasurement.stop(uri);
       results.push(taskResult);
 
@@ -67,8 +71,8 @@ export function runWalltimeBench(bench: Bench, rootCallingFile: string): void {
         };
 
         benchmarks.push(benchmark);
-
         console.log(`    ✔ Collected walltime data for ${uri}`);
+        InstrumentHooks.setExecutedBenchmark(process.pid, uri);
       } else {
         console.warn(`    ⚠ No result data available for ${uri}`);
       }