Add optional ability to log spans

This commit is contained in:
asonix 2023-12-11 14:58:53 -06:00
parent cd80ab16c3
commit 6daa745446
5 changed files with 22 additions and 2 deletions

View file

@ -14,6 +14,7 @@ concurrency = 32
[tracing.logging]
format = "normal"
targets = "info"
log_spans = false
[tracing.console]
buffer_capacity = 102400

View file

@ -17,6 +17,7 @@ impl Args {
old_repo_cache_capacity,
log_format,
log_targets,
log_spans,
console_address,
console_buffer_capacity,
opentelemetry_url,
@ -36,6 +37,7 @@ impl Args {
logging: Logging {
format: log_format,
targets: log_targets.map(Serde::new),
log_spans,
},
console: Console {
address: console_address,
@ -550,6 +552,8 @@ struct Logging {
format: Option<LogFormat>,
#[serde(skip_serializing_if = "Option::is_none")]
targets: Option<Serde<Targets>>,
#[serde(skip_serializing_if = "std::ops::Not::not")]
log_spans: bool,
}
#[derive(Debug, Default, serde::Serialize)]
@ -854,6 +858,9 @@ pub(super) struct Args {
/// Log levels to print to stdout, respects RUST_LOG formatting
#[arg(long)]
log_targets: Option<Targets>,
/// Whether to log openning and closing of tracing spans to stdout
#[arg(long)]
log_spans: bool,
/// Address and port to expose tokio-console metrics
#[arg(long)]

View file

@ -53,6 +53,7 @@ struct TracingDefaults {
struct LoggingDefaults {
format: LogFormat,
targets: Serde<Targets>,
log_spans: bool,
}
#[derive(Clone, Debug, serde::Serialize)]
@ -207,6 +208,7 @@ impl Default for LoggingDefaults {
LoggingDefaults {
format: LogFormat::Normal,
targets: "info".parse().expect("Valid targets string"),
log_spans: false,
}
}
}

View file

@ -153,6 +153,8 @@ pub(crate) struct Logging {
pub(crate) format: LogFormat,
pub(crate) targets: Serde<Targets>,
pub(crate) log_spans: bool,
}
#[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]

View file

@ -6,7 +6,9 @@ use opentelemetry_sdk::{propagation::TraceContextPropagator, Resource};
use tracing::subscriber::set_global_default;
use tracing_error::ErrorLayer;
use tracing_log::LogTracer;
use tracing_subscriber::{layer::SubscriberExt, registry::LookupSpan, Layer, Registry};
use tracing_subscriber::{
fmt::format::FmtSpan, layer::SubscriberExt, registry::LookupSpan, Layer, Registry,
};
pub(super) fn init_tracing(tracing: &Tracing) -> color_eyre::Result<()> {
color_eyre::install()?;
@ -15,7 +17,13 @@ pub(super) fn init_tracing(tracing: &Tracing) -> color_eyre::Result<()> {
opentelemetry::global::set_text_map_propagator(TraceContextPropagator::new());
let format_layer = tracing_subscriber::fmt::layer();
let fmt_span = if tracing.logging.log_spans {
FmtSpan::NEW | FmtSpan::CLOSE
} else {
FmtSpan::NONE
};
let format_layer = tracing_subscriber::fmt::layer().with_span_events(fmt_span);
match tracing.logging.format {
LogFormat::Compact => with_format(format_layer.compact(), tracing),