From f6ccc7315133b7c01773894b43f3ae24d9492106 Mon Sep 17 00:00:00 2001
From: Riley <asonix@asonix.dog>
Date: Tue, 21 Sep 2021 03:32:52 -0500
Subject: [PATCH] Allow root_span to wrap the full middlware chain, not just
 the output future (#37)

---
 src/middleware.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/middleware.rs b/src/middleware.rs
index 9096f8bba..7477320af 100644
--- a/src/middleware.rs
+++ b/src/middleware.rs
@@ -133,7 +133,7 @@ where
         let root_span_wrapper = RootSpan::new(root_span.clone());
         req.extensions_mut().insert(root_span_wrapper);
 
-        let fut = self.service.call(req);
+        let fut = root_span.in_scope(|| self.service.call(req));
         Box::pin(
             async move {
                 let outcome = fut.await;