feat: Tracing for DHT rpc requests

Author: lfbrehm

net/src/dht/driver.rs

@@ -454,10 +454,12 @@ impl DhtDriver {
     }
 
     fn dispatch_rpc_response(&mut self, inbound_id: InboundId, response: DhtResponse) {
+        let self_id = self.endpoint.id();
         let maybe_send = self.inbound_contexts.remove(&inbound_id);
         let io_tx = self.io_tx.clone();
         tokio::spawn(async move {
             if let Some((_conn, mut send)) = maybe_send {
+                trace!("RPC RESPONSE: from {} to {:?}", self_id, _conn.to_info());
                 let _ = write_response_to_stream(&mut send, &response).await;
             }
             let _ = io_tx.send(DhtIo::InboundDropped { inbound_id }).await;
@@ -687,12 +689,14 @@ async fn rpc_request(
     peer: NodeId,
     request: DhtRequest,
 ) -> Result<DhtResponse, DhtIoError> {
+    trace!("RPC_REQUEST: Connection to {} from {}", peer, endpoint.id());
     let conn = tokio::time::timeout(RPC_TIMEOUT, endpoint.connect(peer, DHT_ALPN))
         .await?
         .map_err(DhtIoError::network)?;
 
     let (mut send, mut recv) = conn.open_bi().await.map_err(DhtIoError::network)?;
 
+    trace!("RPC_REQUEST: Sending request");
     let request_bytes = encode_request(&request)?;
     let len = (request_bytes.len() as u32).to_be_bytes();
     send.write_all(&len).await.map_err(DhtIoError::network)?;
@@ -702,6 +706,8 @@ async fn rpc_request(
     send.finish().map_err(DhtIoError::network)?;
 
     let mut len_buf = [0u8; 4];
+
+    trace!("RPC_REQUEST: Receiving response");
     tokio::time::timeout(RPC_TIMEOUT, recv.read_exact(&mut len_buf))
         .await?
         .map_err(DhtIoError::network)?;

net/src/lib.rs

@@ -36,7 +36,7 @@ use parking_lot::RwLock;
 use tokio::sync::{Mutex, mpsc, oneshot};
 use tokio::task::JoinHandle;
 use tokio_util::sync::CancellationToken;
-use tracing::{Instrument, Span, debug, warn};
+use tracing::{Instrument, Span, debug, trace, warn};
 
 pub use dht::DhtHandle;
 pub use error::{NetError, Result};
@@ -457,6 +457,12 @@ impl NetHandle {
         peer_nodes.extend(peer_endpoints.iter().map(|endpoint| endpoint.id));
         let peer_nodes = unique_peer_nodes(peer_nodes, node_id);
         let dht_signed_authorized_nodes = Arc::new(RwLock::new(peer_nodes.clone()));
+
+        trace!(
+            "DHT SIGNED NODES: {:?}",
+            dht_signed_authorized_nodes.read()
+        );
+
         let peer_connectivity = Arc::new(Mutex::new(PeerConnectivityManagerState::new(
             &peer_nodes,
             "realm_config",