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

Tune compio integration (#412)

Browse source
This commit is contained in:
Nikolay Kim 2024-09-05 08:53:00 +05:00 • committed by GitHub
parent 432791356c
commit 8a3a8f1df8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 38 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ntex-compio/CHANGES.md
View file

@ -1,5 +1,9 @@
# Changes
## [0.1.1] - 2024-09-05
* Tune write task
## [0.1.0] - 2024-08-29
* Initial release

2
ntex-compio/Cargo.toml
View file

@ -1,6 +1,6 @@
[package]
name = "ntex-compio"
version = "0.1.0"
version = "0.1.1"
authors = ["ntex contributors <team@ntex.rs>"]
description = "compio runtime intergration for ntex framework"
keywords = ["network", "framework", "async", "futures"]

36
ntex-compio/src/io.rs
View file

@ -15,8 +15,10 @@ impl IoStream for crate::TcpStream {
compio::runtime::spawn(async move {
run(&mut io, &read, write).await;
let res = io.close().await;
log::debug!("{} Stream is closed, {:?}", read.tag(), res);
match io.close().await {
Ok(_) => log::debug!("{} Stream is closed", read.tag()),
Err(e) => log::error!("{} Stream is closed, {:?}", read.tag(), e),
}
})
.detach();
@ -31,8 +33,10 @@ impl IoStream for crate::UnixStream {
compio::runtime::spawn(async move {
run(&mut io, &read, write).await;
let res = io.close().await;
log::debug!("{} Stream is closed, {:?}", read.tag(), res);
match io.close().await {
Ok(_) => log::debug!("{} Unix stream is closed", read.tag()),
Err(e) => log::error!("{} Unix stream is closed, {:?}", read.tag(), e),
}
})
.detach();
@ -221,21 +225,23 @@ async fn write<T: AsyncWrite>(io: &mut T, state: &WriteContext) -> io::Result<()
let BufResult(result, buf1) = io.write(buf).await;
buf = buf1;
match result {
return match result {
Ok(0) => Err(io::Error::new(
io::ErrorKind::WriteZero,
"failed to write frame to transport",
)),
Ok(size) => {
if buf.0.len() == size {
return io.flush().await;
// return io.flush().await;
state.memory_pool().release_write_buf(buf.0);
Ok(())
} else {
buf.0.advance(size);
continue;
}
if size == 0 {
return Err(io::Error::new(
io::ErrorKind::WriteZero,
"failed to write frame to transport",
));
}
buf.0.advance(size);
}
Err(e) => return Err(e),
}
Err(e) => Err(e),
};
}
})
.await