Minor update

Improves error handling.
Aeron · Mar 3, 2024 · 5dfa03c · 5dfa03c
1 parent 3b60df6
commit 5dfa03c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -12,6 +12,7 @@ include = ["src/**/*", "LICENSE", "README.md"]
 edition = "2021"
 
 [dependencies]
+anyhow = { version = "1.0.80" }
 async-std = { version = "1.12", features = ["attributes"] }
 async-signals = { version = "0.4" }
 clap = { version = "4.0", features = ["cargo", "derive", "env", "wrap_help"] }

diff --git a/src/main.rs b/src/main.rs
@@ -3,6 +3,7 @@ mod pool;
 
 use std::time::{Duration, Instant};
 
+use anyhow::{bail, Result};
 use async_signals::Signals;
 use async_std::io::ErrorKind;
 use async_std::net::TcpListener;
@@ -27,7 +28,7 @@ async fn process(
     pool: ConnectionPoolProxy,
     length: usize,
     delay: Duration,
-) -> Result<(SocketAddr, Duration), std::io::Error> {
+) -> Result<(SocketAddr, Duration)> {
     let addr = stream.peer_addr()?;
     let cap = length + CRLF.len();
     let now = Instant::now();
@@ -56,6 +57,7 @@ async fn process(
         }
     } else {
         stream.shutdown(Shutdown::Both)?;
+        bail!("pool is not ready");
     }
 
     pool.remove(addr).await;