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

Check service readiness once per decoded item (#463)

Browse source
This commit is contained in:
Nikolay Kim 2024-11-10 15:24:00 +05:00 • committed by GitHub
parent 5700a50f56
commit a97059aa9b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 21 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ntex-io/CHANGES.md
View file

@ -1,5 +1,9 @@
# Changes
## [2.8.3] - 2024-11-10
* Check service readiness once per decoded item
## [2.8.2] - 2024-11-05
* Do not rely on not_ready(), always check service readiness

2
ntex-io/Cargo.toml
View file

@ -1,6 +1,6 @@
[package]
name = "ntex-io"
version = "2.8.2"
version = "2.8.3"
authors = ["ntex contributors <team@ntex.rs>"]
description = "Utilities for encoding and decoding frames"
keywords = ["network", "framework", "async", "futures"]

22
ntex-io/src/dispatcher.rs
View file

@ -126,12 +126,12 @@ pin_project_lite::pin_project! {
bitflags::bitflags! {
#[derive(Copy, Clone, Debug, Eq, PartialEq, Ord, PartialOrd, Hash)]
struct Flags: u8 {
const READY_ERR = 0b00001;
const IO_ERR = 0b00010;
const KA_ENABLED = 0b00100;
const KA_TIMEOUT = 0b01000;
const READ_TIMEOUT = 0b10000;
const READY = 0b100000;
const READY_ERR = 0b000001;
const IO_ERR = 0b000010;
const KA_ENABLED = 0b000100;
const KA_TIMEOUT = 0b001000;
const READ_TIMEOUT = 0b010000;
const READY = 0b100000;
}
}
@ -342,6 +342,7 @@ where
PollService::Continue => continue,
};
slf.flags.remove(Flags::READY);
slf.call_service(cx, item);
}
// handle write back-pressure
@ -471,9 +472,18 @@ where
}
fn poll_service(&mut self, cx: &mut Context<'_>) -> Poll<PollService<U>> {
if self.flags.contains(Flags::READY) {
if self.shared.service.poll_not_ready(cx).is_ready() {
self.flags.remove(Flags::READY);
} else {
return Poll::Ready(self.check_error());
}
}
// wait until service becomes ready
match self.shared.service.poll_ready(cx) {
Poll::Ready(Ok(_)) => {
self.flags.insert(Flags::READY);
let _ = self.shared.service.poll_not_ready(cx);
Poll::Ready(self.check_error())
}