Break27/imposter
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: client shutdown

Browse Source
This commit is contained in:
break27 2024-06-07 17:56:43 +08:00
parent 47fe240aac
commit 8ebba7271c
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/server.rs
View File

@ -21,16 +21,18 @@ impl Server {
let agent = agent.clone(); let agent = agent.clone();
let (mut inbound, addr) = listener.accept().await?; let (mut inbound, addr) = listener.accept().await?;
log::info!("*** Incoming connection from {}", addr); log::info!("*** Incoming connection from {addr}");
async_std::task::spawn(async move { async_std::task::spawn(async move {
if let Err(e) = agent.handle(&mut inbound).await { if let Err(e) = agent.handle(&mut inbound).await {
log::error!("Agent: {}", e); log::error!("Agent: {e}");
let resp = http::Response::from_err(e); let resp = http::Response::from_err(e);
inbound.write(resp.to_string().as_bytes()).await.unwrap(); inbound.write(resp.to_string().as_bytes()).await.unwrap();
inbound.flush().await.unwrap(); inbound.flush().await.unwrap();
} }
let _ = inbound.shutdown(std::net::Shutdown::Both);
}); });
} }
} }