use std::io::{self, Read, Seek, Write};
use crate::{BlockCompression, MetadataEntry, Page};
mod v1;
pub(crate) mod datablock;
#[derive(thiserror::Error, Debug)]
#[non_exhaustive]
pub enum PersistenceError {
#[error("I/O error: {0}.")]
Io(#[from] io::Error),
#[error("Integer number: {0}.")]
InvalidNumber(#[from] std::num::TryFromIntError),
#[error("Invalid magic number.")]
InvalidMagic,
#[error("The book contains too many pages.")]
TooManyPages,
#[error("Unable to load page index.")]
PageError(#[from] crate::page::PageError),
}
const MAGIC_SIZE: usize = 8;
pub(crate) fn load<I>(mut input: I) -> Result<crate::Book<I>, PersistenceError>
where
I: Read + Seek,
{
let mut magic = [0; MAGIC_SIZE];
input
.read_exact(&mut magic)
.map_err(|_| PersistenceError::InvalidMagic)?;
match &magic {
v1::MAGIC => v1::load(input),
_ => Err(PersistenceError::InvalidMagic),
}
}
pub(crate) fn dump<O>(
output: O,
pages: &[Page],
metadata: &[MetadataEntry],
compression: BlockCompression,
) -> Result<(), PersistenceError>
where
O: Write + Seek,
{
v1::dump(output, pages, metadata, compression)
}