Fix success message bootstrap on server side

MissKrabtri · Jun 27, 2022 · e2d674f · e2d674f
1 parent aabc030
commit e2d674f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/massa-bootstrap/src/client.rs b/massa-bootstrap/src/client.rs
@@ -94,6 +94,7 @@ async fn stream_final_state(
                     });
                 }
                 BootstrapServerMessage::FinalStateFinished => {
+                    info!("State bootstrap complete");
                     *next_bootstrap_message = Some(BootstrapClientMessage::AskBootstrapPeers);
                     return Ok(());
                 }
@@ -296,7 +297,7 @@ async fn bootstrap_from_server(
             }
         };
     }
-    info!("Successful state bootstrap");
+    info!("Successful bootstrap");
     Ok(())
 }
 

diff --git a/massa-bootstrap/src/server.rs b/massa-bootstrap/src/server.rs
@@ -432,7 +432,12 @@ async fn manage_bootstrap(
     loop {
         match tokio::time::timeout(bootstrap_settings.read_timeout.into(), server.next()).await {
             Err(_) => return Ok(()),
-            Ok(Err(e)) => return Err(e),
+            Ok(Err(e)) => {
+                match e {
+                    BootstrapError::IoError(_) => return Ok(()),
+                    _ => return Err(e)
+                }
+            },
             Ok(Ok(msg)) => match msg {
                 BootstrapClientMessage::AskBootstrapPeers => {
                     match tokio::time::timeout(