From 75ca682efbe16a273826d4e7f884e81d740482fa Mon Sep 17 00:00:00 2001
From: Tethys Svensson <freaken@freaken.dk>
Date: Wed, 19 Jun 2024 16:09:02 +0200
Subject: [PATCH] Make child-spans work as expected when using the
 lambda-runtime

---
 lambda-runtime/src/layers/trace.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/lambda-runtime/src/layers/trace.rs b/lambda-runtime/src/layers/trace.rs
index 7a9f8370..35c74c12 100644
--- a/lambda-runtime/src/layers/trace.rs
+++ b/lambda-runtime/src/layers/trace.rs
@@ -43,7 +43,11 @@ where
 
     fn call(&mut self, req: LambdaInvocation) -> Self::Future {
         let span = request_span(&req.context);
-        self.inner.call(req).instrument(span)
+        let future = {
+            let _guard = span.enter();
+            self.inner.call(req)
+        };
+        future.instrument(span)
     }
 }