51 lines
1 KiB
Rust
51 lines
1 KiB
Rust
use std::{ops::Index, sync::Arc};
|
|
|
|
use conduit::{Result, Server};
|
|
|
|
use crate::{
|
|
cork::Cork,
|
|
maps,
|
|
maps::{Maps, MapsKey, MapsVal},
|
|
Engine, Map,
|
|
};
|
|
|
|
pub struct Database {
|
|
pub db: Arc<Engine>,
|
|
map: Maps,
|
|
}
|
|
|
|
impl Database {
|
|
/// Load an existing database or create a new one.
|
|
pub async fn open(server: &Arc<Server>) -> Result<Arc<Self>> {
|
|
let db = Engine::open(server)?;
|
|
Ok(Arc::new(Self {
|
|
db: db.clone(),
|
|
map: maps::open(&db)?,
|
|
}))
|
|
}
|
|
|
|
#[inline]
|
|
#[must_use]
|
|
pub fn cork(&self) -> Cork { Cork::new(&self.db, false, false) }
|
|
|
|
#[inline]
|
|
#[must_use]
|
|
pub fn cork_and_flush(&self) -> Cork { Cork::new(&self.db, true, false) }
|
|
|
|
#[inline]
|
|
#[must_use]
|
|
pub fn cork_and_sync(&self) -> Cork { Cork::new(&self.db, true, true) }
|
|
|
|
#[inline]
|
|
pub fn iter_maps(&self) -> impl Iterator<Item = (&MapsKey, &MapsVal)> + '_ { self.map.iter() }
|
|
}
|
|
|
|
impl Index<&str> for Database {
|
|
type Output = Arc<Map>;
|
|
|
|
fn index(&self, name: &str) -> &Self::Output {
|
|
self.map
|
|
.get(name)
|
|
.expect("column in database does not exist")
|
|
}
|
|
}
|