From d6118657a83a139b6a0e90314bef2ded68dfcd22 Mon Sep 17 00:00:00 2001
From: Bastian Gruber <foreach@me.com>
Date: Wed, 3 May 2023 08:19:25 +0200
Subject: [PATCH] Break reading loop as soon as client closes the connection

---
 problem_05/bin/server.rs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/problem_05/bin/server.rs b/problem_05/bin/server.rs
index 885db80..4adffdb 100644
--- a/problem_05/bin/server.rs
+++ b/problem_05/bin/server.rs
@@ -69,7 +69,7 @@ pub async fn handle_request(socket: TcpStream, upstream: TcpStream) -> Result<()
                     }
                     None => {
                         info!("Client closed the connection");
-                        return Ok(())
+                        break;
                     }
                 }
             }
@@ -90,12 +90,14 @@ pub async fn handle_request(socket: TcpStream, upstream: TcpStream) -> Result<()
                     }
                     None => {
                         info!("Server closed the connection");
-                        return Ok(())
+                        break;
                     }
                 }
             }
         }
     }
+
+    Ok(())
 }
 
 fn replace_address(message: String) -> String {