mirrors/ntex
1
0
Fork 0
mirror of https://github.com/ntex-rs/ntex.git synced 2025-04-03 21:07:39 +03:00
Code Issues Projects Releases Packages Wiki Activity

Handle io flush during write back-pressure (#246)

Browse source
This commit is contained in:
Nikolay Kim 2023-11-12 21:19:33 +06:00 • committed by GitHub
parent 2a19b7f995
commit c6b26123ca
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 53 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

11
ntex-tokio/src/io.rs
View file

@ -132,8 +132,8 @@ impl Future for WriteTask {
match this.st {
IoWriteState::Processing(ref mut delay) => {
match this.state.poll_ready(cx) {
Poll::Ready(WriteStatus::Ready) => {
match ready!(this.state.poll_ready(cx)) {
WriteStatus::Ready => {
if let Some(delay) = delay {
if delay.poll_elapsed(cx).is_ready() {
this.state.close(Some(io::Error::new(
@ -157,14 +157,14 @@ impl Future for WriteTask {
}
}
}
Poll::Ready(WriteStatus::Timeout(time)) => {
WriteStatus::Timeout(time) => {
log::trace!("initiate timeout delay for {:?}", time);
if delay.is_none() {
*delay = Some(sleep(time));
}
self.poll(cx)
}
Poll::Ready(WriteStatus::Shutdown(time)) => {
WriteStatus::Shutdown(time) => {
log::trace!("write task is instructed to shutdown");
let timeout = if let Some(delay) = delay.take() {
@ -176,7 +176,7 @@ impl Future for WriteTask {
this.st = IoWriteState::Shutdown(timeout, Shutdown::None);
self.poll(cx)
}
Poll::Ready(WriteStatus::Terminate) => {
WriteStatus::Terminate => {
log::trace!("write task is instructed to terminate");
if !matches!(
@ -191,7 +191,6 @@ impl Future for WriteTask {
this.state.close(None);
Poll::Ready(())
}
Poll::Pending => Poll::Pending,
}
}
IoWriteState::Shutdown(ref mut delay, ref mut st) => {