Fix a bug of hanging forever while on ibd (mit-dci#42)

While we don't leave ibd, this code wouldn't shutdown if requested
Davidson-Souza · May 5, 2023 · b31cda0 · b31cda0
1 parent fd1b8ea
commit b31cda0
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/blockchain/p2p_blockchain/node.rs b/src/blockchain/p2p_blockchain/node.rs
@@ -650,11 +650,10 @@ impl UtreexoNode {
                 timeout(Duration::from_millis(10), self.node_rx.recv()).await
             {
                 try_and_log!(self.handle_notification(notification).await);
-                if *stop_signal.read().await {
-                    break;
-                }
             }
-
+            if *stop_signal.read().await {
+                break;
+            }
             periodic_job!(
                 self.maybe_open_connection().await,
                 self.last_connection,