Add network and commit to write task logs

ZcashFoundation · Feb 3, 2023 · 9327468 · 9327468
1 parent fc3c637
commit 9327468
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/zebra-state/src/service.rs b/zebra-state/src/service.rs
@@ -338,16 +338,19 @@ impl StateService {
             tokio::sync::mpsc::unbounded_channel();
 
         let finalized_state_for_writing = finalized_state.clone();
+        let span = Span::current();
         let block_write_task = std::thread::spawn(move || {
-            write::write_blocks_from_channels(
-                finalized_block_write_receiver,
-                non_finalized_block_write_receiver,
-                finalized_state_for_writing,
-                non_finalized_state,
-                invalid_block_reset_sender,
-                chain_tip_sender,
-                non_finalized_state_sender,
-            )
+            span.in_scope(move || {
+                write::write_blocks_from_channels(
+                    finalized_block_write_receiver,
+                    non_finalized_block_write_receiver,
+                    finalized_state_for_writing,
+                    non_finalized_state,
+                    invalid_block_reset_sender,
+                    chain_tip_sender,
+                    non_finalized_state_sender,
+                )
+            })
         });
         let block_write_task = Arc::new(block_write_task);