chore(vitest): emit benchmark markers

Author: not-matthias

packages/vitest-plugin/src/walltime/index.ts

@@ -1,5 +1,7 @@
 import {
   InstrumentHooks,
+  MARKER_TYPE_BENCHMARK_END,
+  MARKER_TYPE_BENCHMARK_START,
   setupCore,
   writeWalltimeResults,
 } from "@codspeed/core";
@@ -66,6 +68,7 @@ export class WalltimeRunner extends NodeBenchmarkRunner {
     this.isTinybenchHookedWithCodspeed = true;
 
     const originalRun = tinybench.Task.prototype.run;
+    const pid = process.pid;
 
     const getSuiteUri = (): string => {
       if (this.currentSuiteId === null) {
@@ -78,18 +81,37 @@ export class WalltimeRunner extends NodeBenchmarkRunner {
       const { fn } = this as { fn: Fn };
       const suiteUri = getSuiteUri();
 
-      function __codspeed_root_frame__() {
-        return fn();
+      const finishRound = (roundStart: bigint): void => {
+        const roundEnd = InstrumentHooks.currentTimestamp();
+        InstrumentHooks.addMarker(pid, MARKER_TYPE_BENCHMARK_START, roundStart);
+        InstrumentHooks.addMarker(pid, MARKER_TYPE_BENCHMARK_END, roundEnd);
+      };
+
+      function __codspeed_root_frame__(): unknown {
+        const roundStart = InstrumentHooks.currentTimestamp();
+        const result = fn();
+        if (
+          result !== null &&
+          typeof result === "object" &&
+          typeof (result as PromiseLike<unknown>).then === "function"
+        ) {
+          return (result as PromiseLike<unknown>).then((value) => {
+            finishRound(roundStart);
+            return value;
+          });
+        }
+        finishRound(roundStart);
+        return result;
       }
-      (this as { fn: Fn }).fn = __codspeed_root_frame__;
+      (this as { fn: Fn }).fn = __codspeed_root_frame__ as Fn;
 
       InstrumentHooks.startBenchmark();
       await originalRun.call(this);
       InstrumentHooks.stopBenchmark();
 
       // Look up the URI by task name
       const uri = `${suiteUri}::${this.name}`;
-      InstrumentHooks.setExecutedBenchmark(process.pid, uri);
+      InstrumentHooks.setExecutedBenchmark(pid, uri);
 
       return this;
     };