asonix/pict-rs
1
1
Fork 1
Code Issues 2 Pull requests 3 Actions Projects Releases 133 Wiki Activity

Improve AyncRead impl

Browse source
This commit is contained in:
asonix 2024-02-22 16:21:31 -06:00
parent de356c1f12
commit 00a08a8bc9
1 changed files with 14 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
src/bytes_stream.rs
View file

@ -110,25 +110,26 @@ impl Stream for IoStream {
impl AsyncRead for BytesReader {
fn poll_read(
mut self: Pin<&mut Self>,
_cx: &mut Context<'_>,
_: &mut Context<'_>,
buf: &mut tokio::io::ReadBuf<'_>,
) -> Poll<std::io::Result<()>> {
while buf.remaining() > 0 {
if self.index == self.inner[0].len() {
self.inner.pop_front();
self.index = 0;
}
if let Some(bytes) = self.inner.front() {
if self.index == bytes.len() {
self.inner.pop_front();
self.index = 0;
continue;
}
if self.inner.is_empty() {
let upper_bound = (self.index + buf.remaining()).min(bytes.len());
let slice = &bytes[self.index..upper_bound];
buf.put_slice(slice);
self.index += slice.len();
} else {
break;
}
let upper_bound = (self.index + buf.remaining()).min(self.inner[0].len());
let slice = &self.inner[0][self.index..upper_bound];
buf.put_slice(slice);
self.index += slice.len();
}
Poll::Ready(Ok(()))