protocol : use send_timeout in fallback_function (#3980)

massalabs · May 25, 2023 · a138b93 · a138b93
1 parent 765fe8b
commit a138b93
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/massa-protocol-worker/src/handlers/peer_handler/mod.rs b/massa-protocol-worker/src/handlers/peer_handler/mod.rs
@@ -578,8 +578,9 @@ impl InitConnectionHandler<PeerId, Context, MessagesHandler> for MassaHandshake
                 warn!("Failed to serialize message: {}", err);
                 return;
             }
-            //TODO: Make it non blockable
-            if let Err(err) = endpoint.send::<PeerId>(buf.as_slice()) {
+            if let Err(err) =
+                endpoint.send_timeout::<PeerId>(buf.as_slice(), Duration::from_millis(200))
+            {
                 warn!("Failed to send message: {}", err);
                 return;
             }