Add tracing (#1942)

[lemmy.git] / src / lib.rs

diff --git a/src/lib.rs b/src/lib.rs
index ea79cbb089fc1e1b0ac1ff1bd06ef46c57727100..98c2320d120544a825eeaa2f62b901f24e3bb867 100644 (file)
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -2,3 +2,27 @@
 pub mod api_routes;
 pub mod code_migrations;
 pub mod scheduled_tasks;
+
+use lemmy_utils::LemmyError;
+use tracing::subscriber::set_global_default;
+use tracing_error::ErrorLayer;
+use tracing_log::LogTracer;
+use tracing_subscriber::{fmt::format::FmtSpan, layer::SubscriberExt, EnvFilter, Registry};
+
+pub fn init_tracing() -> Result<(), LemmyError> {
+  LogTracer::init()?;
+
+  let env_filter = EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("info"));
+  let format_layer = tracing_subscriber::fmt::layer()
+    .with_span_events(FmtSpan::NEW | FmtSpan::CLOSE)
+    .pretty();
+
+  let subscriber = Registry::default()
+    .with(env_filter)
+    .with(format_layer)
+    .with(ErrorLayer::default());
+
+  set_global_default(subscriber)?;
+
+  Ok(())
+}