From a244fb2e6d3720576088ceb44c92baca37abf99d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ahmet=20Kaan=20G=C3=9CM=C3=9C=C5=9E?=
 <96421894+Tahinli@users.noreply.github.com>
Date: Wed, 5 Jun 2024 23:52:19 +0300
Subject: [PATCH] fix: :ambulance: exit when a client disconnects

---
 src/main.rs           | 7 ++++---
 src/server_network.rs | 3 +--
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 8165ebd..1ddc91e 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -19,7 +19,8 @@ async fn main() {
         },
         None => return,
     };
-    todo!("Limbo Block: Not in chain, but processing by others or none. Sync it also")
+    //todo!("Limbo Block: Not in chain, but processing by others or none. Sync it also");
+    //todo!("Consensus should be notified for new block, should forget old blocks");
 }
 
 async fn server() {
@@ -29,13 +30,13 @@ async fn server() {
     };
 
     let blockchain = BlockChain::new(server_config.difficulty.into());
-    let blockhain_thread_safe = Arc::new(Mutex::new(blockchain));
+    let blockchain_thread_safe = Arc::new(Mutex::new(blockchain));
 
     let block_data_channel_sender = broadcast::channel(1).0;
 
     server_network::start_network(
         server_config,
-        blockhain_thread_safe,
+        blockchain_thread_safe,
         block_data_channel_sender.subscribe(),
     )
     .await;
diff --git a/src/server_network.rs b/src/server_network.rs
index 1bac762..78e8ca4 100644
--- a/src/server_network.rs
+++ b/src/server_network.rs
@@ -35,12 +35,11 @@ pub async fn start_network(
         blockchain_thread_safe.clone(),
         consensus_data_channels.clone(),
     ));
-    //todo!("Consensus should be notified for new block, should forget old blocks");
     loop {
         if let Ok(connection) = listener_socket.accept().await {
             let ws_stream = match accept_async(connection.0).await {
                 Ok(ws_stream) => ws_stream,
-                Err(_) => return,
+                Err(_) => continue,
             };
             let (ws_stream_sender, ws_stream_receiver) = ws_stream.split();
             let blockchain_thread_safe = blockchain_thread_safe.clone();