itsscb/tracing
1
0
Fork 0
mirror of https://github.com/tokio-rs/tracing.git synced 2025-10-02 15:24:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

examples: use tempfile in inferno-flame

Browse Source 
This commit updates the `inferno-flame` example to use the `tempfile`
crate as a replacement
for the unmaintained `tempdir` crate.

Also, the previous version of the example output the flamegraph inside
the temporary directory. Since the temporary directory is cleaned up
when the program exits, this means the user can't actually look at the
flamegraph when running this example. I've changed the example to put the
flamegraph in the current working dir instead, so the user can look at
it.

Signed-off-by: Eliza Weisman <eliza@buoyant.io>
This commit is contained in:
Eliza Weisman 2021-08-16 10:55:39 -07:00
parent 058c9e2519
commit 20e1588cfb
2 changed files with 26 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/Cargo.toml
View File

@ -48,7 +48,7 @@ log = "0.4"
# inferno example # inferno example
inferno = "0.10.0" inferno = "0.10.0"
tempdir = "0.3.7" tempfile = "3"
# opentelemetry example # opentelemetry example
opentelemetry = { version = "0.16", default-features = false, features = ["trace"] } opentelemetry = { version = "0.16", default-features = false, features = ["trace"] }

36
examples/examples/inferno-flame.rs
View File

@ -1,9 +1,11 @@
use std::fs::File; use std::{
use std::io::{BufReader, BufWriter}; env,
use std::path::Path; fs::File,
use std::thread::sleep; io::{BufReader, BufWriter},
use std::time::Duration; path::{Path, PathBuf},
use tempdir::TempDir; thread::sleep,
time::Duration,
};
use tracing::{span, Level}; use tracing::{span, Level};
use tracing_flame::FlameLayer; use tracing_flame::FlameLayer;
use tracing_subscriber::{prelude::*, registry::Registry}; use tracing_subscriber::{prelude::*, registry::Registry};
@ -20,11 +22,12 @@ fn setup_global_subscriber(dir: &Path) -> impl Drop {
_guard _guard
} }
fn make_flamegraph(dir: &Path) { fn make_flamegraph(tmpdir: &Path, out: &Path) {
let inf = File::open(dir.join(PATH)).unwrap(); println!("outputting flamegraph to {}", out.display());
let inf = File::open(tmpdir.join(PATH)).unwrap();
let reader = BufReader::new(inf); let reader = BufReader::new(inf);
let out = File::create(dir.join("inferno.svg")).unwrap(); let out = File::create(out).unwrap();
let writer = BufWriter::new(out); let writer = BufWriter::new(out);
let mut opts = inferno::flamegraph::Options::default(); let mut opts = inferno::flamegraph::Options::default();
@ -32,8 +35,19 @@ fn make_flamegraph(dir: &Path) {
} }
fn main() { fn main() {
let out = if let Some(arg) = env::args().nth(1) {
PathBuf::from(arg)
} else {
let mut path = env::current_dir().expect("failed to read current directory");
path.push("tracing-flame-inferno.svg");
path
};
// setup the flame layer // setup the flame layer
let tmp_dir = TempDir::new("flamegraphs").unwrap(); let tmp_dir = tempfile::Builder::new()
.prefix("flamegraphs")
.tempdir()
.expect("failed to create temporary directory");
let guard = setup_global_subscriber(tmp_dir.path()); let guard = setup_global_subscriber(tmp_dir.path());
// do a bunch of span entering and exiting to simulate a program running // do a bunch of span entering and exiting to simulate a program running
@ -52,5 +66,5 @@ fn main() {
// drop the guard to make sure the layer flushes its output then read the // drop the guard to make sure the layer flushes its output then read the
// output to create the flamegraph // output to create the flamegraph
drop(guard); drop(guard);
make_flamegraph(tmp_dir.path()); make_flamegraph(tmp_dir.path(), out.as_ref());
} }