54 lines
1.8 KiB
Rust
54 lines
1.8 KiB
Rust
|
use tracing_subscriber::layer::SubscriberExt;
|
||
|
|
|
||
|
|
#[must_use]
|
||
|
|
pub fn get_subscriber(debug: bool) -> impl tracing::Subscriber + Send + Sync {
|
||
|
|
let env_filter = if debug { "debug" } else { "info" }.to_string();
|
||
|
|
let env_filter = tracing_subscriber::EnvFilter::try_from_default_env()
|
||
|
|
.unwrap_or_else(|_| tracing_subscriber::EnvFilter::new(env_filter));
|
||
|
|
let stdout_log = tracing_subscriber::fmt::layer().pretty();
|
||
|
|
let subscriber = tracing_subscriber::Registry::default()
|
||
|
|
.with(env_filter)
|
||
|
|
.with(stdout_log);
|
||
|
|
let json_log = if debug {
|
||
|
|
None
|
||
|
|
} else {
|
||
|
|
Some(tracing_subscriber::fmt::layer().json())
|
||
|
|
};
|
||
|
|
subscriber.with(json_log)
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn init_subscriber(subscriber: impl tracing::Subscriber + Send + Sync) {
|
||
|
|
tracing::subscriber::set_global_default(subscriber).expect("Failed to set subscriber");
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(test)]
|
||
|
|
mod tests {
|
||
|
|
use super::*;
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn get_subscriber_debug_mode() {
|
||
|
|
let subscriber = get_subscriber(true);
|
||
|
|
// If we can create the subscriber without panicking, the test passes
|
||
|
|
// We can't easily inspect the subscriber's internals, but we can verify it's created
|
||
|
|
let _ = subscriber;
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn get_subscriber_production_mode() {
|
||
|
|
let subscriber = get_subscriber(false);
|
||
|
|
// If we can create the subscriber without panicking, the test passes
|
||
|
|
let _ = subscriber;
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn get_subscriber_creates_valid_subscriber() {
|
||
|
|
// Test both debug and non-debug modes create valid subscribers
|
||
|
|
let debug_subscriber = get_subscriber(true);
|
||
|
|
let prod_subscriber = get_subscriber(false);
|
||
|
|
|
||
|
|
// Basic smoke test - if these are created without panicking, they're valid
|
||
|
|
let _ = debug_subscriber;
|
||
|
|
let _ = prod_subscriber;
|
||
|
|
}
|
||
|
|
}
|