conduit/src/database/abstraction.rs

use super::Config;
use crate::Result;

use async_trait::async_trait;
use std::{future::Future, pin::Pin, sync::Arc};

#[cfg(feature = "sled")]
pub mod sled;

#[cfg(feature = "sqlite")]
pub mod sqlite;

#[cfg(feature = "heed")]
pub mod heed;

#[cfg(feature = "rocksdb")]
pub mod rocksdb;

#[cfg(feature = "persy")]
pub mod persy;

#[cfg(any(
    feature = "sqlite",
    feature = "rocksdb",
    feature = "heed",
    feature = "persy"
))]
pub mod watchers;

#[async_trait]
pub trait KeyValueDatabaseEngine: Send + Sync {
    async fn open(config: &Config) -> Result<Self>
    where
        Self: Sized;
    async fn open_tree(&self, name: &'static str) -> Result<Arc<dyn KvTree>>;
    async fn flush(&self) -> Result<()>;
    async fn cleanup(&self) -> Result<()> {
        Ok(())
    }
    async fn memory_usage(&self) -> Result<String> {
        Ok("Current database engine does not support memory usage reporting.".to_owned())
    }
}

#[async_trait]
pub trait KvTree: Send + Sync {
    fn get(&self, key: &[u8]) -> Result<Option<Vec<u8>>>;

    fn insert(&self, key: &[u8], value: &[u8]) -> Result<()>;
    fn insert_batch(&self, iter: &mut dyn Iterator<Item = (Vec<u8>, Vec<u8>)>) -> Result<()>;

    fn remove(&self, key: &[u8]) -> Result<()>;

    fn iter<'a>(&'a self) -> Box<dyn Send + Iterator<Item = (Vec<u8>, Vec<u8>)> + 'a>;

    fn iter_from<'a>(
        &'a self,
        from: &[u8],
        backwards: bool,
    ) -> Box<dyn Send + Iterator<Item = (Vec<u8>, Vec<u8>)> + 'a>;

    async fn increment(&self, key: &[u8]) -> Result<Vec<u8>>;
    fn increment_batch(&self, iter: &mut dyn Iterator<Item = Vec<u8>>) -> Result<()>;

    fn scan_prefix<'a>(
        &'a self,
        prefix: Vec<u8>,
    ) -> Box<dyn Send + Iterator<Item = (Vec<u8>, Vec<u8>)> + 'a>;

    fn watch_prefix<'a>(&'a self, prefix: &[u8]) -> Pin<Box<dyn Future<Output = ()> + Send + 'a>>;

    async fn clear(&self) -> Result<()> {
        for (key, _) in self.iter() {
            self.remove(&key)?;
        }

        Ok(())
    }
}