YGGverse/agate
1
0
Fork 0
mirror of https://github.com/YGGverse/agate.git synced 2026-04-08 20:45:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

handle errors in handle_request

Browse source
This commit is contained in:
Johann150 2021-01-24 20:45:36 +01:00 • committed by Matt Brubeck
parent 3353989e7e
commit 74853799c7
1 changed files with 22 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

38
src/main.rs
View file

@ -34,11 +34,7 @@ fn main() -> Result {
log::info!("Listening on {:?}...", ARGS.addrs); log::info!("Listening on {:?}...", ARGS.addrs);
loop { loop {
let (stream, _) = listener.accept().await?; let (stream, _) = listener.accept().await?;
tokio::spawn(async { tokio::spawn(async { handle_request(stream).await });
if let Err(e) = handle_request(stream).await {
log::error!("{:?}", e);
}
});
} }
}) })
} }
@ -123,8 +119,9 @@ struct RequestHandle {
pub log_line: String, pub log_line: String,
} }
/// Handle a single client session (request + response). /// Handle a single client session (request + response) and any errors that
async fn handle_request(stream: TcpStream) -> Result { /// may occur while processing it.
async fn handle_request(stream: TcpStream) {
let log_line = format!( let log_line = format!(
"{} {}", "{} {}",
stream.local_addr().unwrap(), stream.local_addr().unwrap(),
@ -139,19 +136,28 @@ async fn handle_request(stream: TcpStream) -> Result {
} }
); );
let stream = TLS.accept(stream).await?; let stream = match TLS.accept(stream).await {
Ok(stream) => stream,
Err(e) => {
log::warn!("{} error:{}", log_line, e);
return;
}
};
let mut handle = RequestHandle { stream, log_line }; let mut handle = RequestHandle { stream, log_line };
match parse_request(&mut handle).await { let mut result = match parse_request(&mut handle).await {
Ok(url) => send_response(url, &mut handle).await?, Ok(url) => send_response(url, &mut handle).await,
Err((status, msg)) => send_header(&mut handle, status, msg).await?, Err((status, msg)) => send_header(&mut handle, status, msg).await,
};
if let Err(e) = result {
log::warn!("{} error:{}", handle.log_line, e);
} else if let Err(e) = handle.stream.shutdown().await {
log::warn!("{} error:{}", handle.log_line, e);
} else {
log::info!("{}", handle.log_line);
} }
handle.stream.shutdown().await?;
log::info!("{}", handle.log_line);
Ok(())
} }
/// TLS configuration. /// TLS configuration.