end-to-end flamegraphs

hydro-project · Aug 2, 2024 · 1146268 · 1146268
1 parent a214786
commit 1146268
Show file tree

Hide file tree

Showing 5 changed files with 307 additions and 74 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/hydro_deploy/core/Cargo.toml b/hydro_deploy/core/Cargo.toml
@@ -22,6 +22,7 @@ futures = "0.3.26"
 futures-core = "0.3.26"
 hydroflow_cli_integration = { path = "../hydroflow_cli_integration", version = "^0.5.2" }
 indicatif = "0.17.8"
+inferno = "0.11.20"
 memo-map = "0.3.2"
 nanoid = "0.4.0"
 nix = "0.26.2"
@@ -31,4 +32,4 @@ shell-escape = "0.1.5"
 tempfile = "3.3.0"
 tokio = { version = "1.16", features = [ "full" ] }
 tokio-stream = { version = "0.1.15", default-features = false }
-tokio-util = { version = "0.7.7" }
+tokio-util = { version = "0.7.7", features = [ "io-util" ] }
diff --git a/hydro_deploy/core/src/hydroflow_crate/perf_options.rs b/hydro_deploy/core/src/hydroflow_crate/perf_options.rs
@@ -1,7 +1,11 @@
 use std::path::PathBuf;
 
-#[derive(Clone)]
+#[derive(Clone, buildstructor::Builder)]
+#[non_exhaustive] // Prevent direct construction.
 pub struct PerfOptions {
     pub output_file: PathBuf,
     pub frequency: u32,
+    pub fold_options: Option<inferno::collapse::perf::Options>,
+    // This type is super annoying and isn't `clone` and has a lifetime... so wrap in fn pointer for now.
+    pub flamegraph_options: Option<fn() -> inferno::flamegraph::Options<'static>>,
 }