Fine-Grained Latency Profiling with runtime/trace Regions and Tasks

package main
 
import (
	"context"
	"os"
	"runtime/trace"
	"time"
)
 
func ProcessDataBatch(ctx context.Context, batchID int) {
	ctx, task := trace.NewTask(ctx, "ProcessBatch")
	defer task.End()
 
	trace.Log(ctx, "batchID", string(rune(batchID)))
 
	region := trace.StartRegion(ctx, "ComputeStage")
	time.Sleep(10 * time.Millisecond)
	region.End()
 
	regionIO := trace.StartRegion(ctx, "IOFlushStage")
	time.Sleep(15 * time.Millisecond)
	regionIO.End()
}
 
func main() {
	f, err := os.Create("trace.out")
	if err != nil {
		panic(err)
	}
	defer f.Close()
 
	if err := trace.Start(f); err != nil {
		panic(err)
	}
	defer trace.Stop()
 
	ProcessDataBatch(context.Background(), 42)
}