actix/actix-extras
1
0
Fork 0
mirror of https://github.com/actix/actix-extras.git synced 2024-11-28 09:42:40 +01:00
Code Issues Releases Wiki Activity

better keep-alive handling

Browse Source
This commit is contained in:
Nikolay Kim 2018-10-07 10:09:48 -07:00
parent 13193a0721
commit 8acf9eb98a
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/h1/dispatcher.rs
View File

@ -386,21 +386,13 @@ where
if let Some(ref mut timer) = self.ka_timer { if let Some(ref mut timer) = self.ka_timer {
match timer.poll() { match timer.poll() {
Ok(Async::Ready(_)) => { Ok(Async::Ready(_)) => {
if timer.deadline() >= self.ka_expire {
// check for any outstanding request handling
if self.state.is_empty() && self.messages.is_empty() {
// if we get timer during shutdown, just drop connection // if we get timer during shutdown, just drop connection
if self.flags.contains(Flags::SHUTDOWN) { if self.flags.contains(Flags::SHUTDOWN) {
return Err(DispatchError::DisconnectTimeout); return Err(DispatchError::DisconnectTimeout);
} else if !self.flags.contains(Flags::STARTED) { } else if timer.deadline() >= self.ka_expire {
// timeout on first request (slow request) return 408 // check for any outstanding response processing
trace!("Slow request timeout"); if self.state.is_empty() {
self.flags.insert(Flags::STARTED | Flags::DISCONNECTED); if self.flags.contains(Flags::STARTED) {
self.state =
State::SendResponse(Some(OutMessage::Response(
Response::RequestTimeout().finish(),
)));
} else {
trace!("Keep-alive timeout, close connection"); trace!("Keep-alive timeout, close connection");
self.flags.insert(Flags::SHUTDOWN); self.flags.insert(Flags::SHUTDOWN);
@ -412,6 +404,14 @@ where
} else { } else {
return Ok(()); return Ok(());
} }
} else {
// timeout on first request (slow request) return 408
trace!("Slow request timeout");
self.flags.insert(Flags::STARTED | Flags::DISCONNECTED);
self.state =
State::SendResponse(Some(OutMessage::Response(
Response::RequestTimeout().finish(),
)));
} }
} else if let Some(deadline) = self.config.keep_alive_expire() { } else if let Some(deadline) = self.config.keep_alive_expire() {
timer.reset(deadline) timer.reset(deadline)