feat: Sender timout for DHT-RPC-msgs & rcv timout increase

Author: lfbrehm

net/src/dht/driver.rs

@@ -337,6 +337,7 @@ impl DhtDriver {
             tokio::spawn(async move {
                 if let Some((_conn, mut send)) = maybe_send {
                     let _ = write_response_to_stream(
+                        inbound_id,
                         &mut send,
                         &DhtResponse::Error {
                             code: ErrorCode::InvalidRequest,
@@ -465,7 +466,7 @@ impl DhtDriver {
                     self_id,
                     _conn.to_info()
                 );
-                let _ = write_response_to_stream(&mut send, &response).await;
+                let _ = write_response_to_stream(inbound_id, &mut send, &response).await;
             }
             let _ = io_tx.send(DhtIo::InboundDropped { inbound_id }).await;
         });
@@ -733,6 +734,11 @@ async fn rpc_request(
         .await?
         .map_err(DhtIoError::network)?;
 
+    let timeout = tokio::time::timeout(Duration::from_millis(100), send.stopped()).await;
+    if let Err(timer) = timeout {
+        warn!(op_id, "RPC_REQUEST: Send timer elapsed {}", timer);
+    }
+
     trace!(op_id, "RPC_REQUEST: Decode response");
 
     Ok(decode_response(&response_bytes)?)
@@ -766,6 +772,7 @@ async fn read_request_from_stream(recv: &mut RecvStream) -> Result<DhtRequest, D
 }
 
 async fn write_response_to_stream(
+    inbound_id: InboundId,
     send: &mut SendStream,
     response: &DhtResponse,
 ) -> Result<(), DhtIoError> {
@@ -782,7 +789,10 @@ async fn write_response_to_stream(
         .map_err(DhtIoError::network)?;
     send.finish().map_err(DhtIoError::network)?;
 
-    let _ = tokio::time::timeout(Duration::from_millis(100), send.stopped()).await;
+    let timeout = tokio::time::timeout(Duration::from_millis(500), send.stopped()).await;
+    if let Err(timer) = timeout {
+        warn!(inbound_id, "RPC_RESPONSE: Send timer elapsed {}", timer);
+    }
 
     Ok(())
 }