You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
1.1 KiB
Rust
47 lines
1.1 KiB
Rust
use flate2::read::ZlibEncoder;
|
|
use flate2::Compression;
|
|
use std::io::{Read, Result};
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub struct ByteArrayCache {
|
|
inner: Vec<u8>,
|
|
position: usize,
|
|
}
|
|
|
|
impl ByteArrayCache {
|
|
/// Creates a new byte array cache
|
|
pub fn new() -> Self {
|
|
Self {
|
|
inner: Vec::new(),
|
|
position: 0,
|
|
}
|
|
}
|
|
|
|
/// Creates a new byte array cache with capacity
|
|
pub fn with_capacity(capacity: usize) -> Self {
|
|
Self {
|
|
inner: Vec::with_capacity(capacity),
|
|
position: 0,
|
|
}
|
|
}
|
|
|
|
/// Write the data to the inner buffer by using compression
|
|
pub fn write<R: Read>(&mut self, reader: R) -> Result<()> {
|
|
let mut encoder = ZlibEncoder::new(reader, Compression::default());
|
|
let mut buffer = Vec::new();
|
|
encoder.read_to_end(&mut buffer)?;
|
|
self.inner.append(&mut buffer);
|
|
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl Read for ByteArrayCache {
|
|
fn read(&mut self, buf: &mut [u8]) -> Result<usize> {
|
|
let read = (&self.inner[self.position..]).read(buf)?;
|
|
self.position += read;
|
|
|
|
Ok(read)
|
|
}
|
|
}
|