771733deacImplement parsing the binary format using the new parser library and the new encoding types. This is superior to the previous parsing implementation in that invalid data should never cause panics and it exposes and interface to construct an OpSet from a saved document much more efficiently. Signed-off-by: Alex Good <alex@memoryandthought.me>
51 lines
1.5 KiB
Rust
51 lines
1.5 KiB
Rust
use std::{borrow::Cow, io::Read};
|
|
|
|
use crate::storage::{Change, CheckSum, ChunkType, MAGIC_BYTES};
|
|
|
|
use super::OpReadState;
|
|
|
|
#[derive(Clone, Debug, PartialEq)]
|
|
pub(crate) struct Compressed<'a> {
|
|
checksum: CheckSum,
|
|
bytes: Cow<'a, [u8]>,
|
|
}
|
|
|
|
impl<'a> Compressed<'a> {
|
|
pub(crate) fn new(checksum: CheckSum, bytes: Cow<'a, [u8]>) -> Self {
|
|
Self { checksum, bytes }
|
|
}
|
|
|
|
pub(crate) fn compress<'b, O: OpReadState>(change: &'b Change<'b, O>) -> Compressed<'static> {
|
|
let mut result = Vec::with_capacity(change.bytes().len());
|
|
result.extend(MAGIC_BYTES);
|
|
result.extend(change.checksum().bytes());
|
|
result.push(u8::from(ChunkType::Compressed));
|
|
let mut deflater = flate2::bufread::DeflateEncoder::new(
|
|
change.body_bytes(),
|
|
flate2::Compression::default(),
|
|
);
|
|
let mut deflated = Vec::new();
|
|
let deflated_len = deflater.read_to_end(&mut deflated).unwrap();
|
|
leb128::write::unsigned(&mut result, deflated_len as u64).unwrap();
|
|
result.extend(&deflated[..]);
|
|
Compressed {
|
|
checksum: change.checksum(),
|
|
bytes: Cow::Owned(result),
|
|
}
|
|
}
|
|
|
|
pub(crate) fn bytes(&self) -> Cow<'a, [u8]> {
|
|
self.bytes.clone()
|
|
}
|
|
|
|
pub(crate) fn checksum(&self) -> CheckSum {
|
|
self.checksum
|
|
}
|
|
|
|
pub(crate) fn into_owned(self) -> Compressed<'static> {
|
|
Compressed {
|
|
checksum: self.checksum,
|
|
bytes: Cow::Owned(self.bytes.into_owned()),
|
|
}
|
|
}
|
|
}
|