util: upgrade tokio-util to bytes 0.6 (#3052)

This commit is contained in:
Dirkjan Ochtman
2020-10-27 09:30:29 +01:00
committed by GitHub
parent 6d0ba19af5
commit fe2b997675
11 changed files with 112 additions and 238 deletions
+3 -5
View File
@@ -3,7 +3,7 @@ use futures_core::stream::Stream;
use pin_project_lite::pin_project;
use std::pin::Pin;
use std::task::{Context, Poll};
use tokio::io::AsyncRead;
use tokio::io::{AsyncRead, AsyncReadExt};
const CAPACITY: usize = 4096;
@@ -70,11 +70,9 @@ impl<R: AsyncRead> ReaderStream<R> {
impl<R: AsyncRead> Stream for ReaderStream<R> {
type Item = std::io::Result<Bytes>;
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
use crate::util::poll_read_buf;
let mut this = self.as_mut().project();
let reader = match this.reader.as_pin_mut() {
let mut reader = match this.reader.as_pin_mut() {
Some(r) => r,
None => return Poll::Ready(None),
};
@@ -83,7 +81,7 @@ impl<R: AsyncRead> Stream for ReaderStream<R> {
this.buf.reserve(CAPACITY);
}
match poll_read_buf(cx, reader, &mut this.buf) {
match reader.poll_read_buf(&mut this.buf, cx) {
Poll::Pending => Poll::Pending,
Poll::Ready(Err(err)) => {
self.project().reader.set(None);