fixup! feat: add a root frame for the callgraph generation

CodSpeedHQ · Jun 21, 2023 · d24465b · d24465b
1 parent 41be397
commit d24465b
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/packages/tinybench-plugin/src/index.ts b/packages/tinybench-plugin/src/index.ts
@@ -24,12 +24,14 @@ export function withCodSpeed(bench: Bench): Bench {
     console.log(`[CodSpeed] running with @codspeed/tinybench v${__VERSION__}`);
     for (const task of bench.tasks) {
       const uri = callingFile + "::" + task.name;
-      await (async function __codspeed_root_frame__() {
-        await optimizeFunction(task.fn);
-        measurement.startInstrumentation();
+      // eslint-disable-next-line no-inner-declarations
+      async function __codspeed_root_frame__() {
         await task.fn();
-        measurement.stopInstrumentation(uri);
-      })();
+      }
+      await optimizeFunction(__codspeed_root_frame__);
+      measurement.startInstrumentation();
+      await __codspeed_root_frame__();
+      measurement.stopInstrumentation(uri);
       console.log(`    ✔ Measured ${uri}`);
     }
     console.log(`[CodSpeed] Done running ${bench.tasks.length} benches.`);