From bcd2ea858e50e8a1575903730637e2c3377eaa99 Mon Sep 17 00:00:00 2001
From: Shaun Cox <shaunco@microsoft.com>
Date: Mon, 19 Jun 2023 15:28:35 -0500
Subject: [PATCH] Enable flame graphs and reports for benchmarks

---
 Cargo.toml       | 3 ++-
 benches/trace.rs | 7 ++++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index b2165f5..d683b72 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -38,8 +38,9 @@ thiserror = { version = "1.0.31", optional = true }
 
 [dev-dependencies]
 async-trait = "0.1.56"
-criterion = { version = "0.5.0", default-features = false }
+criterion = { version = "0.4.0", default-features = false, features = ["html_reports"] }
 opentelemetry-jaeger = "0.18.0"
+pprof = { version = "0.11.1", features = ["flamegraph", "criterion"] }
 futures-util = { version = "0.3", default-features = false }
 tokio = { version = "1", features = ["full"] }
 tokio-stream = "0.1"
diff --git a/benches/trace.rs b/benches/trace.rs
index 8dbc96e..c683ffb 100644
--- a/benches/trace.rs
+++ b/benches/trace.rs
@@ -4,6 +4,7 @@ use opentelemetry::{
     trace::{SpanBuilder, Tracer as _, TracerProvider as _},
     Context,
 };
+use pprof::criterion::{Output, PProfProfiler};
 use std::time::SystemTime;
 use tracing::trace_span;
 use tracing_subscriber::prelude::*;
@@ -122,5 +123,9 @@ fn tracing_harness() {
     dummy();
 }
 
-criterion_group!(benches, many_children);
+criterion_group! {
+    name = benches;
+    config = Criterion::default().with_profiler(PProfProfiler::new(100, Output::Flamegraph(None)));
+    targets = many_children
+}
 criterion_main!(benches);