use crate::{db::DBInner, ffi, ffi_util::to_cpath, DBCommon, Error, ThreadMode};
use std::{marker::PhantomData, path::Path};
const LOG_SIZE_FOR_FLUSH: u64 = 0_u64;
pub struct Checkpoint<'db> {
inner: *mut ffi::rocksdb_checkpoint_t,
_db: PhantomData<&'db ()>,
}
impl<'db> Checkpoint<'db> {
pub fn new<T: ThreadMode, I: DBInner>(db: &'db DBCommon<T, I>) -> Result<Self, Error> {
let checkpoint: *mut ffi::rocksdb_checkpoint_t;
unsafe {
checkpoint = ffi_try!(ffi::rocksdb_checkpoint_object_create(db.inner.inner()));
}
if checkpoint.is_null() {
return Err(Error::new("Could not create checkpoint object.".to_owned()));
}
Ok(Self {
inner: checkpoint,
_db: PhantomData,
})
}
pub fn create_checkpoint<P: AsRef<Path>>(&self, path: P) -> Result<(), Error> {
let cpath = to_cpath(path)?;
unsafe {
ffi_try!(ffi::rocksdb_checkpoint_create(
self.inner,
cpath.as_ptr(),
LOG_SIZE_FOR_FLUSH,
));
}
Ok(())
}
}
impl<'db> Drop for Checkpoint<'db> {
fn drop(&mut self) {
unsafe {
ffi::rocksdb_checkpoint_object_destroy(self.inner);
}
}
}