Add some missing tracing spans (#4660)

Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>

zebra-network/src/address_book_updater.rs

@@ -59,9 +59,7 @@ impl AddressBookUpdater {
         let address_book = Arc::new(std::sync::Mutex::new(address_book));
 
         let worker_address_book = address_book.clone();
-        let span = Span::current();
         let worker = move || {
-            span.in_scope(|| {
             info!("starting the address book updater");
 
             while let Some(event) = worker_rx.blocking_recv() {
@@ -80,12 +78,13 @@ impl AddressBookUpdater {
             let error = Err(AllAddressBookUpdaterSendersClosed.into());
             info!(?error, "stopping address book updater");
             error
-            })
         };
 
         // Correctness: spawn address book accesses on a blocking thread,
         //              to avoid deadlocks (see #1976)
-        let address_book_updater_task_handle = tokio::task::spawn_blocking(worker);
+        let span = Span::current();
+        let address_book_updater_task_handle =
+            tokio::task::spawn_blocking(move || span.in_scope(worker));
 
         (
             address_book,

zebra-network/src/peer_set/candidate_set.rs

@@ -4,6 +4,7 @@ use chrono::Utc;
 use futures::stream::{FuturesUnordered, StreamExt};
 use tokio::time::{sleep_until, timeout, Instant};
 use tower::{Service, ServiceExt};
+use tracing::Span;
 
 use zebra_chain::serialization::DateTime32;
 
@@ -333,7 +334,10 @@ where
         //
         // Extend handles duplicate addresses internally.
         let address_book = self.address_book.clone();
-        tokio::task::spawn_blocking(move || address_book.lock().unwrap().extend(addrs))
+        let span = Span::current();
+        tokio::task::spawn_blocking(move || {
+            span.in_scope(|| address_book.lock().unwrap().extend(addrs))
+        })
         .await
         .expect("panic in new peers address book update task");
     }
@@ -386,7 +390,8 @@ where
         };
 
         // Correctness: Spawn address book accesses on a blocking thread, to avoid deadlocks (see #1976).
-        let next_peer = tokio::task::spawn_blocking(next_peer)
+        let span = Span::current();
+        let next_peer = tokio::task::spawn_blocking(move || span.in_scope(next_peer))
             .await
             .expect("panic in next peer address book task")?;
 
@@ -406,7 +411,10 @@ where
         // Spawn address book accesses on a blocking thread,
         // to avoid deadlocks (see #1976).
         let address_book = self.address_book.clone();
-        tokio::task::spawn_blocking(move || address_book.lock().unwrap().update(addr))
+        let span = Span::current();
+        tokio::task::spawn_blocking(move || {
+            span.in_scope(|| address_book.lock().unwrap().update(addr))
+        })
         .await
         .expect("panic in peer failure address book update task");
     }