This commit is contained in:
Rob Ede 2021-06-03 18:36:40 +01:00
parent 9b7bba1048
commit 9bab04ae36
No known key found for this signature in database
GPG Key ID: 97C636207D3EF933
1 changed files with 13 additions and 10 deletions

View File

@ -37,15 +37,15 @@ where
#[inline] #[inline]
pub fn new(stream: S, encoding: ContentEncoding) -> Decoder<S> { pub fn new(stream: S, encoding: ContentEncoding) -> Decoder<S> {
let decoder = match encoding { let decoder = match encoding {
ContentEncoding::Br => Some(ContentDecoder::Br(Box::new(BrotliDecoder::new( ContentEncoding::Br => Some(ContentDecoder::Br(Box::new(
Writer::new(), BrotliDecoder::new(Writer::new()),
)))), ))),
ContentEncoding::Deflate => Some(ContentDecoder::Deflate(Box::new(ZlibDecoder::new( ContentEncoding::Deflate => Some(ContentDecoder::Deflate(Box::new(
Writer::new(), ZlibDecoder::new(Writer::new()),
)))), ))),
ContentEncoding::Gzip => Some(ContentDecoder::Gzip(Box::new(GzDecoder::new( ContentEncoding::Gzip => Some(ContentDecoder::Gzip(Box::new(
Writer::new(), GzDecoder::new(Writer::new()),
)))), ))),
ContentEncoding::Zstd => Some(ContentDecoder::Zstd(Box::new( ContentEncoding::Zstd => Some(ContentDecoder::Zstd(Box::new(
ZstdDecoder::new(Writer::new()).expect( ZstdDecoder::new(Writer::new()).expect(
"Failed to create zstd decoder. This is a bug. \ "Failed to create zstd decoder. This is a bug. \
@ -83,7 +83,10 @@ where
{ {
type Item = Result<Bytes, PayloadError>; type Item = Result<Bytes, PayloadError>;
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> { fn poll_next(
mut self: Pin<&mut Self>,
cx: &mut Context<'_>,
) -> Poll<Option<Self::Item>> {
loop { loop {
if let Some(ref mut fut) = self.fut { if let Some(ref mut fut) = self.fut {
let (chunk, decoder) = let (chunk, decoder) =