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

Add IoRef::is_wr_backpressure() method (#331)

Browse source
This commit is contained in:
Nikolay Kim 2024-03-31 13:40:03 +03:00 • committed by GitHub
parent d6feeec933
commit 7e2e3d1730
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 20 additions and 6 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 # Changes
## [1.0.2] - 2024-03-31
* Add IoRef::is_wr_backpressure() method
## [1.0.1] - 2024-02-05 ## [1.0.1] - 2024-02-05
* Add IoBoxed::take() method * Add IoBoxed::take() method

8
ntex-io/Cargo.toml
View file

@ -1,6 +1,6 @@
[package] [package]
name = "ntex-io" name = "ntex-io"
version = "1.0.1" version = "1.0.2"
authors = ["ntex contributors <team@ntex.rs>"] authors = ["ntex contributors <team@ntex.rs>"]
description = "Utilities for encoding and decoding frames" description = "Utilities for encoding and decoding frames"
keywords = ["network", "framework", "async", "futures"] keywords = ["network", "framework", "async", "futures"]
@ -18,10 +18,10 @@ path = "src/lib.rs"
[dependencies] [dependencies]
ntex-codec = "0.6.2" ntex-codec = "0.6.2"
ntex-bytes = "0.1.24" ntex-bytes = "0.1.24"
ntex-util = "1.0.0" ntex-util = "1.0"
ntex-service = "2.0.0" ntex-service = "2.0"
bitflags = "2.4" bitflags = "2"
log = "0.4" log = "0.4"
pin-project-lite = "0.2" pin-project-lite = "0.2"

8
ntex-io/src/filter.rs
View file

@ -134,12 +134,16 @@ impl Filter for Base {
s.with_write_destination(io, |buf| { s.with_write_destination(io, |buf| {
if let Some(buf) = buf { if let Some(buf) = buf {
let len = buf.len(); let len = buf.len();
if len > 0 && self.0.flags().contains(Flags::WR_PAUSED) { let flags = self.0.flags();
if len > 0 && flags.contains(Flags::WR_PAUSED) {
self.0 .0.remove_flags(Flags::WR_PAUSED); self.0 .0.remove_flags(Flags::WR_PAUSED);
self.0 .0.write_task.wake(); self.0 .0.write_task.wake();
} }
if len >= self.0.memory_pool().write_params_high() { if len >= self.0.memory_pool().write_params_high()
&& !flags.contains(Flags::WR_BACKPRESSURE)
{
self.0 .0.insert_flags(Flags::WR_BACKPRESSURE); self.0 .0.insert_flags(Flags::WR_BACKPRESSURE);
self.0 .0.dispatch_task.wake();
} }
} }
}); });

6
ntex-io/src/ioref.rs
View file

@ -41,6 +41,12 @@ impl IoRef {
.intersects(Flags::IO_STOPPING | Flags::IO_STOPPED) .intersects(Flags::IO_STOPPING | Flags::IO_STOPPED)
} }
#[inline]
/// Check if write back-pressure is enabled
pub fn is_wr_backpressure(&self) -> bool {
self.0.flags.get().contains(Flags::WR_BACKPRESSURE)
}
#[inline] #[inline]
/// Wake dispatcher task /// Wake dispatcher task
pub fn wake(&self) { pub fn wake(&self) {