use flate2::{Compress, Compression, FlushCompress, Status};use std::io; const BUF_SIZE: usize = 4096 * 8; pub struct DeflateWriter<W> { compressor: Compress, inner: W, buf: [u8; BUF_SIZE],} impl<W> DeflateWriter<W>where W: io::Write,{ pub fn new(inner: W) -> DeflateWriter<W> { DeflateWriter { compressor: Compress::new(Compression::fast(), true), inner, buf: [0; BUF_SIZE], } } pub fn reset(&mut self) { self.compressor.reset(); } pub fn into_inner(self) -> W { self.inner } fn write_inner(&mut self, mut buf: &[u8], flush: FlushCompress) -> io::Result<usize> { let total_in_when_start = self.compressor.total_in(); loop { let last_total_in = self.compressor.total_in(); let last_total_out = self.compressor.total_out(); let status = self .compressor .compress(buf, &mut self.buf, flush) .map_err(|err| io::Error::new(io::ErrorKind::Other, err))?; let written = self.compressor.total_out() - last_total_out; if written > 0 { self.inner.write_all(&self.buf[..written as usize])?; } match status { Status::StreamEnd => return Ok((self.compressor.total_in() - total_in_when_start) as usize), Status::Ok | Status::BufError => { let consumed = self.compressor.total_in() - last_total_in; buf = &buf[consumed as usize..]; // output buffer still makes progress if self.compressor.total_out() > last_total_out { continue; } // input still makes progress if self.compressor.total_in() > last_total_in { continue; } // input also makes no progress anymore, need more so leave with what we have return Ok((self.compressor.total_in() - total_in_when_start) as usize); } } } }} impl<W: io::Write> io::Write for DeflateWriter<W> { fn write(&mut self, buf: &[u8]) -> io::Result<usize> { self.write_inner(buf, FlushCompress::None) } fn flush(&mut self) -> io::Result<()> { self.write_inner(&[], FlushCompress::Finish).map(|_| ()) }} #[cfg(test)]mod tests;
use flate2::{Compress, Compression, FlushCompress, Status};
use std::io;
const BUF_SIZE: usize = 4096 * 8;
pub struct DeflateWriter<W> {
compressor: Compress,
inner: W,
buf: [u8; BUF_SIZE],
}
impl<W> DeflateWriter<W>
where
W: io::Write,
{
pub fn new(inner: W) -> DeflateWriter<W> {
DeflateWriter {
compressor: Compress::new(Compression::fast(), true),
inner,
buf: [0; BUF_SIZE],
}
}
pub fn reset(&mut self) {
self.compressor.reset();
}
pub fn into_inner(self) -> W {
self.inner
}
fn write_inner(&mut self, mut buf: &[u8], flush: FlushCompress) -> io::Result<usize> {
let total_in_when_start = self.compressor.total_in();
loop {
let last_total_in = self.compressor.total_in();
let last_total_out = self.compressor.total_out();
let status = self
.compressor
.compress(buf, &mut self.buf, flush)
.map_err(|err| io::Error::new(io::ErrorKind::Other, err))?;
let written = self.compressor.total_out() - last_total_out;
if written > 0 {
self.inner.write_all(&self.buf[..written as usize])?;
}
match status {
Status::StreamEnd => return Ok((self.compressor.total_in() - total_in_when_start) as usize),
Status::Ok | Status::BufError => {
let consumed = self.compressor.total_in() - last_total_in;
buf = &buf[consumed as usize..];
// output buffer still makes progress
if self.compressor.total_out() > last_total_out {
continue;
}
// input still makes progress
if self.compressor.total_in() > last_total_in {
continue;
}
// input also makes no progress anymore, need more so leave with what we have
return Ok((self.compressor.total_in() - total_in_when_start) as usize);
}
}
}
}
}
impl<W: io::Write> io::Write for DeflateWriter<W> {
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
self.write_inner(buf, FlushCompress::None)
}
fn flush(&mut self) -> io::Result<()> {
self.write_inner(&[], FlushCompress::Finish).map(|_| ())
}
}
#[cfg(test)]
mod tests;