feat(chain, file_store): add is_empty
method to PersistBackend
trait
This commit is contained in:
parent
5998a22819
commit
7d5f31f6cc
@ -80,6 +80,15 @@ pub trait PersistBackend<C> {
|
|||||||
|
|
||||||
/// Return the aggregate changeset `C` from persistence.
|
/// Return the aggregate changeset `C` from persistence.
|
||||||
fn load_from_persistence(&mut self) -> Result<C, Self::LoadError>;
|
fn load_from_persistence(&mut self) -> Result<C, Self::LoadError>;
|
||||||
|
|
||||||
|
/// Returns whether the persistence backend contains no data.
|
||||||
|
fn is_empty(&mut self) -> Result<bool, Self::LoadError>
|
||||||
|
where
|
||||||
|
C: Append,
|
||||||
|
{
|
||||||
|
self.load_from_persistence()
|
||||||
|
.map(|changeset| changeset.is_empty())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<C: Default> PersistBackend<C> for () {
|
impl<C: Default> PersistBackend<C> for () {
|
||||||
@ -94,4 +103,8 @@ impl<C: Default> PersistBackend<C> for () {
|
|||||||
fn load_from_persistence(&mut self) -> Result<C, Self::LoadError> {
|
fn load_from_persistence(&mut self) -> Result<C, Self::LoadError> {
|
||||||
Ok(C::default())
|
Ok(C::default())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn is_empty(&mut self) -> Result<bool, Self::LoadError> {
|
||||||
|
Ok(true)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
@ -37,6 +37,14 @@ where
|
|||||||
let (changeset, result) = self.aggregate_changesets();
|
let (changeset, result) = self.aggregate_changesets();
|
||||||
result.map(|_| changeset)
|
result.map(|_| changeset)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn is_empty(&mut self) -> Result<bool, Self::LoadError> {
|
||||||
|
let init_pos = self.db_file.stream_position()?;
|
||||||
|
let stream_len = self.db_file.seek(io::SeekFrom::End(0))?;
|
||||||
|
let magic_len = self.magic.len() as u64;
|
||||||
|
self.db_file.seek(io::SeekFrom::Start(init_pos))?;
|
||||||
|
Ok(stream_len == magic_len)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'a, C> Store<'a, C>
|
impl<'a, C> Store<'a, C>
|
||||||
@ -182,6 +190,19 @@ mod test {
|
|||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
struct TestTracker;
|
struct TestTracker;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn is_empty() {
|
||||||
|
let mut file = NamedTempFile::new().unwrap();
|
||||||
|
file.write_all(&TEST_MAGIC_BYTES).expect("should write");
|
||||||
|
|
||||||
|
let mut db = Store::<TestChangeSet>::new(&TEST_MAGIC_BYTES, file.reopen().unwrap())
|
||||||
|
.expect("must open");
|
||||||
|
assert!(db.is_empty().expect("must read"));
|
||||||
|
db.write_changes(&vec!["hello".to_string(), "world".to_string()])
|
||||||
|
.expect("must write");
|
||||||
|
assert!(!db.is_empty().expect("must read"));
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn new_fails_if_file_is_too_short() {
|
fn new_fails_if_file_is_too_short() {
|
||||||
let mut file = NamedTempFile::new().unwrap();
|
let mut file = NamedTempFile::new().unwrap();
|
||||||
|
Loading…
x
Reference in New Issue
Block a user