Add Read and Write timeouts (#2)

* Add Read and Write timeouts

What
--
Add Read and Write timeouts

Reading and writing data from/to a tcp stream can now timeout is it
takes longer than the set timeout. Default is 60 seconds.

Other
--
* Improve server creation in tests

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "seva"
-version = "0.1.1"
+version = "0.1.2"
 edition = "2021"
 authors = ["Matt Gathu <[email protected]>"]
 description = "Simple directory http server inspired by Python's http.server"

src/errors.rs

@@ -30,6 +30,10 @@ pub enum SevaError {
     UriTooLong,
     #[error("Missing value for header: {0}")]
     MissingHeaderValue(HeaderName),
+    #[error("timed out while reading data")]
+    ReadTimeOut,
+    #[error("timed out while writing data")]
+    WriteTimeOut,
 }
 
 #[derive(Error, Debug)]
@@ -53,10 +57,18 @@ impl fmt::Display for ParsingError {
 pub trait IoErrorUtils {
     fn kind(&self) -> io::ErrorKind;
 
+    fn is_addr_in_use(&self) -> bool {
+        self.kind() == io::ErrorKind::AddrInUse
+    }
+
     fn is_blocking(&self) -> bool {
         self.kind() == io::ErrorKind::WouldBlock
     }
 
+    fn is_timed_out(&self) -> bool {
+        self.kind() == io::ErrorKind::TimedOut
+    }
+
     fn is_not_found(&self) -> bool {
         self.kind() == io::ErrorKind::NotFound
     }

src/main.rs

@@ -43,6 +43,10 @@ struct Args {
     /// Log Level
     #[arg(long, default_value = "INFO")]
     log_level: Level,
+
+    /// Timeout duration in seconds
+    #[arg(long, default_value = "60")]
+    timeout: u64,
 }
 
 fn expand_tilde(path: &str) -> String {
@@ -81,7 +85,8 @@ fn main() -> errors::Result<()> {
         port = args.port,
     );
 
-    let mut server = server::HttpServer::new(&args.host, args.port, dir, false)?;
+    let mut server =
+        server::HttpServer::new(&args.host, args.port, dir, false, args.timeout)?;
     server.run()?;
     Ok(())
 }