async_compression_issue_150_workaround/codec/zstd/
decoder.rs

1use std::io::Result;
2
3use crate::{codec::Decode, unshared::Unshared, util::PartialBuffer};
4use libzstd::stream::raw::{Decoder, Operation};
5
6#[derive(Debug)]
7pub struct ZstdDecoder {
8    decoder: Unshared<Decoder<'static>>,
9}
10
11impl ZstdDecoder {
12    pub fn new() -> Self {
13        Self {
14            decoder: Unshared::new(Decoder::new().unwrap()),
15        }
16    }
17}
18
19impl Decode for ZstdDecoder {
20    fn reinit(&mut self) -> Result<()> {
21        self.decoder.get_mut().reinit()?;
22        Ok(())
23    }
24
25    fn decode(
26        &mut self,
27        input: &mut PartialBuffer<impl AsRef<[u8]>>,
28        output: &mut PartialBuffer<impl AsRef<[u8]> + AsMut<[u8]>>,
29    ) -> Result<bool> {
30        let status = self
31            .decoder
32            .get_mut()
33            .run_on_buffers(input.unwritten(), output.unwritten_mut())?;
34        input.advance(status.bytes_read);
35        output.advance(status.bytes_written);
36        Ok(status.remaining == 0)
37    }
38
39    fn flush(
40        &mut self,
41        output: &mut PartialBuffer<impl AsRef<[u8]> + AsMut<[u8]>>,
42    ) -> Result<bool> {
43        let mut out_buf = zstd_safe::OutBuffer::around(output.unwritten_mut());
44        let bytes_left = self.decoder.get_mut().flush(&mut out_buf)?;
45        let len = out_buf.as_slice().len();
46        output.advance(len);
47        Ok(bytes_left == 0)
48    }
49
50    fn finish(
51        &mut self,
52        output: &mut PartialBuffer<impl AsRef<[u8]> + AsMut<[u8]>>,
53    ) -> Result<bool> {
54        let mut out_buf = zstd_safe::OutBuffer::around(output.unwritten_mut());
55        let bytes_left = self.decoder.get_mut().finish(&mut out_buf, true)?;
56        let len = out_buf.as_slice().len();
57        output.advance(len);
58        Ok(bytes_left == 0)
59    }
60}
async_compression_issue_150_workaround/codec/zstd/decoder.rs

async_compression_issue_150_workaround/codec/zstd/
decoder.rs
