Ignore ConnectionAborted error

Neo-Ciber94 · Sep 13, 2024 · c12ab09 · c12ab09
1 parent f0b6700
commit c12ab09
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/src/server.rs b/src/server.rs
@@ -1,5 +1,5 @@
 use std::{
-    io::{BufRead, BufReader, Write},
+    io::{BufRead, BufReader, ErrorKind, Write},
     net::{SocketAddr, TcpListener, TcpStream},
     str::FromStr,
     sync::{Arc, Mutex},
@@ -33,7 +33,11 @@ impl Server {
     ) {
         let request = read_request(&mut stream).expect("Failed to read request");
         let response = handler.handle(request);
-        write_response(response, &mut stream).unwrap()
+        match write_response(response, &mut stream) {
+            Ok(_) => {}
+            Err(err) if err.kind() == ErrorKind::ConnectionAborted => {}
+            Err(err) => eprintln!("{err}"),
+        }
     }
 
     pub fn listen<H: RequestHandler + Send + Sync + 'static>(
@@ -194,10 +198,5 @@ fn write_response(response: Response<Body>, stream: &mut TcpStream) -> std::io::
         }
     }
 
-    let last_bytes =  body.read_last().map_err(std::io::Error::other)?;
-    if let Some(bytes) = last_bytes{
-        stream.write_all(&bytes)?;
-    }
-
     Ok(())
 }