diff options
author | Simon Hauser <simon.hauser@helsinki-systems.de> | 2024-06-19T09·18+0200 |
---|---|---|
committer | clbot <clbot@tvl.fyi> | 2024-07-01T13·55+0000 |
commit | 6a9a4d56a4a99acb95c2b4ca1d590490c7732f79 (patch) | |
tree | 6f62121cd6bfb27f4a658d9a82eb3f173ead1bde /tvix/tracing/src/lib.rs | |
parent | 87f38cad6140a10f436ba4d60b52735cc545bc63 (diff) |
feat(tvix/tracing): expose stdout_writer and stderr_writer r/8333
Using std::io::{Stdout,StdErr} directly will clobber the output by an active progress bar. To resolve this issue the exposed writers should be prefered over `println!` and `eprintln!`. Change-Id: Ic79465cd4e8b9dad5a138f6b08c5f0de9dcf54a1 Reviewed-on: https://cl.tvl.fyi/c/depot/+/11860 Autosubmit: Simon Hauser <simon.hauser@helsinki-systems.de> Tested-by: BuildkiteCI Reviewed-by: flokli <flokli@flokli.de>
Diffstat (limited to 'tvix/tracing/src/lib.rs')
-rw-r--r-- | tvix/tracing/src/lib.rs | 37 |
1 files changed, 34 insertions, 3 deletions
diff --git a/tvix/tracing/src/lib.rs b/tvix/tracing/src/lib.rs index b965ca4a3dbc..b914408ff4c7 100644 --- a/tvix/tracing/src/lib.rs +++ b/tvix/tracing/src/lib.rs @@ -2,7 +2,7 @@ use indicatif::ProgressStyle; use lazy_static::lazy_static; use tokio::sync::{mpsc, oneshot}; use tracing::Level; -use tracing_indicatif::{filter::IndicatifFilter, IndicatifLayer}; +use tracing_indicatif::{filter::IndicatifFilter, writer, IndicatifLayer, IndicatifWriter}; use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt, EnvFilter, Layer}; #[cfg(feature = "otlp")] @@ -50,9 +50,30 @@ pub enum Error { #[derive(Clone)] pub struct TracingHandle { tx: Option<mpsc::Sender<Option<oneshot::Sender<()>>>>, + + stdout_writer: IndicatifWriter<writer::Stdout>, + stderr_writer: IndicatifWriter<writer::Stderr>, } impl TracingHandle { + /// Returns a writer for [std::io::Stdout] that ensures its output will not be clobbered by + /// active progress bars. + /// + /// Instead of `println!(...)` prefer `writeln!(handle.get_stdout_writer(), ...)` + pub fn get_stdout_writer(&self) -> IndicatifWriter<writer::Stdout> { + // clone is fine here because its only a wrapper over an `Arc` + self.stdout_writer.clone() + } + + /// Returns a writer for [std::io::Stderr] that ensures its output will not be clobbered by + /// active progress bars. + /// + /// Instead of `println!(...)` prefer `writeln!(handle.get_stderr_writer(), ...)`. + pub fn get_stderr_writer(&self) -> IndicatifWriter<writer::Stderr> { + // clone is fine here because its only a wrapper over an `Arc` + self.stderr_writer.clone() + } + /// This will flush possible attached tracing providers, e.g. otlp exported, if enabled. /// If there is none enabled this will result in a noop. /// @@ -167,6 +188,8 @@ impl TracingBuilder { pub fn build(self) -> Result<TracingHandle, Error> { // Set up the tracing subscriber. let indicatif_layer = IndicatifLayer::new().with_progress_style(PB_SPINNER_STYLE.clone()); + let stdout_writer = indicatif_layer.get_stdout_writer(); + let stderr_writer = indicatif_layer.get_stderr_writer(); let subscriber = tracing_subscriber::registry() .with( EnvFilter::builder() @@ -209,7 +232,11 @@ impl TracingBuilder { { subscriber.with(Some(layer)).try_init()?; } - return Ok(TracingHandle { tx: Some(tx) }); + return Ok(TracingHandle { + tx: Some(tx), + stdout_writer, + stderr_writer, + }); } } #[cfg(feature = "tracy")] @@ -221,7 +248,11 @@ impl TracingBuilder { subscriber.try_init()?; } - Ok(TracingHandle { tx: None }) + Ok(TracingHandle { + tx: None, + stdout_writer, + stderr_writer, + }) } } |