feat(testing): add benchmark markers

Author: not-matthias

testing/capi/instrument-hooks.go

@@ -8,9 +8,17 @@ typedef struct instruments_root_InstrumentHooks__547 InstrumentHooks;
 import "C"
 import (
 	"runtime"
+	"time"
 	"unsafe"
 )
 
+const (
+	MarkerTypeSampleStart    = 0
+	MarkerTypeSampleEnd      = 1
+	MarkerTypeBenchmarkStart = 2
+	MarkerTypeBenchmarkEnd   = 3
+)
+
 // This will be set in the go-runner
 var integrationVersion = "dev"
 
@@ -79,3 +87,16 @@ func (i *InstrumentHooks) IsInstrumented() bool {
 	}
 	return bool(C.instrument_hooks_is_instrumented(i.hooks))
 }
+
+func CurrentTimestamp() uint64 {
+	return uint64(C.instrument_hooks_current_timestamp())
+}
+
+func (i *InstrumentHooks) AddBenchmarkTimestamps(startTimestamp, endTimestamp uint64) {
+	if i.hooks == nil {
+		return
+	}
+	pid := uint32(C.getpid())
+	C.instrument_hooks_add_marker(i.hooks, C.uint32_t(pid), C.uint8_t(MarkerTypeBenchmarkStart), C.uint64_t(startTimestamp))
+	C.instrument_hooks_add_marker(i.hooks, C.uint32_t(pid), C.uint8_t(MarkerTypeBenchmarkEnd), C.uint64_t(endTimestamp))
+}

testing/testing/benchmark.go

@@ -90,6 +90,7 @@ type codspeed struct {
 
 	codspeedTimePerRoundNs []time.Duration
 	codspeedItersPerRound  []int64
+	startTimestamp         uint64
 }
 
 // B is a type passed to [Benchmark] functions to manage benchmark
@@ -155,6 +156,7 @@ func (b *B) StartTimer() {
 		// runtime.ReadMemStats(&memStats)
 		// b.startAllocs = memStats.Mallocs
 		// b.startBytes = memStats.TotalAlloc
+		b.codspeed.startTimestamp = capi.CurrentTimestamp()
 		b.start = highPrecisionTimeNow()
 		b.timerOn = true
 		// b.loop.i &^= loopPoisonTimer
@@ -165,6 +167,9 @@ func (b *B) StartTimer() {
 // while performing steps that you don't want to measure.
 func (b *B) StopTimer() {
 	if b.timerOn {
+		endTimestamp := capi.CurrentTimestamp()
+		b.codspeed.instrument_hooks.AddBenchmarkTimestamps(b.codspeed.startTimestamp, endTimestamp)
+
 		// For b.N loops: This will be called in runN which sets b.N to the number of iterations.
 		// For b.Loop() loops: loopSlowPath sets b.N to 0 to prevent b.N loops within b.Loop. However, since
 		// we're starting/stopping the timer for each iteration in the b.Loop() loop, we can use 1 as