pub mod batch;
pub mod bytes;
pub mod cache;
pub mod compaction;
pub mod comparator;
pub mod error;
pub mod iterator;
pub mod kv;
pub mod management;
pub mod options;
pub mod snapshots;
use comparator::{create_comparator, Comparator};
use leveldb_sys::*;
use libc::c_char;
use self::error::Error;
use self::options::{c_options, Options};
use std::{ffi::CString, ptr, sync::Arc, path::Path};
#[allow(missing_docs)]
struct RawDB {
ptr: *mut leveldb_t,
}
#[allow(missing_docs)]
impl Drop for RawDB {
fn drop(&mut self) {
unsafe {
leveldb_close(self.ptr);
}
}
}
#[allow(missing_docs)]
struct RawComparator {
ptr: *mut leveldb_comparator_t,
}
impl Drop for RawComparator {
fn drop(&mut self) {
unsafe {
leveldb_comparator_destroy(self.ptr);
}
}
}
#[derive(Clone)]
pub struct Database(pub(crate) Arc<DatabaseInner>);
pub(crate) struct DatabaseInner {
database: RawDB,
#[allow(dead_code)]
comparator: Option<RawComparator>,
#[allow(dead_code)]
options: Options,
}
unsafe impl Sync for DatabaseInner {}
unsafe impl Send for DatabaseInner {}
impl Database {
fn new(
database: *mut leveldb_t,
options: Options,
comparator: Option<*mut leveldb_comparator_t>,
) -> Database {
let raw_comp = match comparator {
Some(p) => Some(RawComparator { ptr: p }),
None => None,
};
Database(Arc::new(DatabaseInner {
database: RawDB { ptr: database },
comparator: raw_comp,
options,
}))
}
pub fn open(name: &Path, options: Options) -> Result<Database, Error> {
let mut error = ptr::null_mut();
unsafe {
let c_string = CString::new(name.to_str().unwrap()).unwrap();
let c_options = c_options(&options, None);
let db = leveldb_open(
c_options as *const leveldb_options_t,
c_string.as_bytes_with_nul().as_ptr() as *const c_char,
&mut error,
);
leveldb_options_destroy(c_options);
if error.is_null() {
Ok(Database::new(db, options, None))
} else {
Err(Error::new_from_char(error))
}
}
}
pub fn open_with_comparator<C: Comparator>(
name: &Path,
options: Options,
comparator: C,
) -> Result<Database, Error> {
let mut error = ptr::null_mut();
let comp_ptr = create_comparator(Box::new(comparator));
unsafe {
let c_string = CString::new(name.to_str().unwrap()).unwrap();
let c_options = c_options(&options, Some(comp_ptr));
let db = leveldb_open(
c_options as *const leveldb_options_t,
c_string.as_bytes_with_nul().as_ptr() as *const c_char,
&mut error,
);
leveldb_options_destroy(c_options);
if error.is_null() {
Ok(Database::new(db, options, Some(comp_ptr)))
} else {
Err(Error::new_from_char(error))
}
}
}
}