From c5ea875f544824b0c042bf7c0a58b3134f9c0373 Mon Sep 17 00:00:00 2001
From: rtkay123 <dev@kanjala.com>
Date: Tue, 12 Aug 2025 05:13:32 +0200
Subject: feat(config): get active routing

---
 lib/warden-middleware/src/trace_layer.rs | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 lib/warden-middleware/src/trace_layer.rs

(limited to 'lib/warden-middleware/src/trace_layer.rs')

diff --git a/lib/warden-middleware/src/trace_layer.rs b/lib/warden-middleware/src/trace_layer.rs
new file mode 100644
index 0000000..5173e8d
--- /dev/null
+++ b/lib/warden-middleware/src/trace_layer.rs
@@ -0,0 +1,24 @@
+use axum::{Router, http::Request};
+use tower_http::trace::TraceLayer;
+use tracing::info_span;
+
+use super::REQUEST_ID_HEADER;
+
+pub fn apply_trace_context_middleware<T: Clone + Send + Sync + 'static>(
+    router: Router<T>,
+) -> Router<T> {
+    router.layer(
+        TraceLayer::new_for_http().make_span_with(|request: &Request<_>| {
+            let request_id = request
+                .headers()
+                .get(REQUEST_ID_HEADER)
+                .expect("should have been applied already");
+
+            info_span!(
+                "http_request",
+                request_id = ?request_id,
+                headers = ?request.headers()
+            )
+        }),
+    )
+}
-- 
cgit v1.2.3