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

Periodically check readiness

Browse source
This commit is contained in:
Nikolay Kim 2024-11-04 21:17:24 +05:00
parent 48ce0ea2bb
commit 9ae3d06aa3
2 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ntex-io/Cargo.toml
View file

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

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

@ -11,6 +11,8 @@ use crate::{Decoded, DispatchItem, IoBoxed, IoStatusUpdate, RecvError};
type Response<U> = <U as Encoder>::Item;
const READY_COUNT: u8 = 32;
#[derive(Clone, Debug)]
/// Shared dispatcher configuration
pub struct DispatcherConfig(Rc<DispatcherConfigInner>);
@ -146,6 +148,7 @@ where
shared: Rc<DispatcherShared<S, U>>,
response: Option<PipelineCall<S, DispatchItem<U>>>,
cfg: DispatcherConfig,
ready_count: u8,
read_remains: u32,
read_remains_prev: u32,
read_max_timeout: Seconds,
@ -473,7 +476,8 @@ where
fn poll_service(&mut self, cx: &mut Context<'_>) -> Poll<PollService<U>> {
// check service readiness
if self.flags.contains(Flags::READY) {
if self.shared.service.poll_not_ready(cx).is_pending() {
if self.ready_count != 0 && self.shared.service.poll_not_ready(cx).is_pending() {
self.ready_count -= 1;
return Poll::Ready(self.check_error());
}
self.flags.remove(Flags::READY);
@ -482,6 +486,7 @@ where
// wait until service becomes ready
match self.shared.service.poll_ready(cx) {
Poll::Ready(Ok(_)) => {
self.ready_count = READY_COUNT;
self.flags.insert(Flags::READY);
Poll::Ready(self.check_error())
}