fix: propagate span into cache writing task

talzion12 · Apr 25, 2023 · e0c77ba · e0c77ba
1 parent d2352f1
commit e0c77ba
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/cache/layer.rs b/src/cache/layer.rs
@@ -11,6 +11,7 @@ use hyper::{
     Body,
 };
 use phf::phf_set;
+use tracing::Instrument;
 use url::Url;
 
 use crate::cache::metadata::CacheMetadata;
@@ -135,12 +136,15 @@ where
         let (sender, receiver) = channel::<Bytes>(10);
         let cache_cloned = self.cache.clone();
 
-        tokio::spawn(async move {
-            match cache_cloned.set(&uri, receiver, metadata).await {
-                Ok(()) => tracing::info!("Wrote to cache"),
-                Err(err) => tracing::error!("Failed to write to cache {err:?}"),
+        tokio::spawn(
+            async move {
+                match cache_cloned.set(&uri, receiver, metadata).await {
+                    Ok(()) => tracing::info!("Wrote to cache"),
+                    Err(err) => tracing::error!("Failed to write to cache {err:?}"),
+                }
             }
-        });
+            .in_current_span(),
+        );
 
         let res_body = Body::wrap_stream(body.then(move |part| {
             tracing::debug!("Received part");