Interface Repository

libs/orbit-essentials/src/repository.rs:73–135

A repository is a generic interface for storing and retrieving data.

/// A repository is a generic interface for storing and retrieving data.
pub trait Repository<Key, Value, Mem>: StableDb<Key, Value, Mem>
where
    Key: Eq + std::hash::Hash + Clone + Ord + Storable,
    Value: Clone + Storable,
    Mem: Memory,
{
    /// Returns the list of records from the repository.
    fn list(&self) -> Vec<Value> {
        let mut entries = Vec::with_capacity(self.len());

        Self::with_db(|db| db.iter().for_each(|(_, entry)| entries.push(entry)));

        entries
    }

    /// Returns whether a record exists in the repository.
    fn exists(&self, key: &Key) -> bool {
        Self::with_db(|db| db.contains_key(key))
    }

    /// Returns the record from the repository if it exists.
    fn get(&self, key: &Key) -> Option<Value> {
        Self::with_db(|db| db.get(key))
    }

    /// Inserts a record into the repository.
    fn insert(&self, key: Key, value: Value) -> Option<Value> {
        Self::with_db(|db| db.insert(key, value))
    }

    /// Removes a record from the repository and returns it if it exists.
    fn remove(&self, key: &Key) -> Option<Value> {
        Self::with_db(|db| db.remove(key))
    }

    /// Returns the number of records stored in the repository.
    fn len(&self) -> usize {
        Self::with_db(|db| db.len() as usize)
    }

    /// Returns whether the repository is empty or not.
    fn is_empty(&self) -> bool {
        self.len() == 0
    }

    fn find_with_filters<'a>(
        &self,
        filters: Vec<Box<dyn SelectionFilter<'a, IdType = UUID> + 'a>>,
    ) -> HashSet<UUID> {
        let mut found_ids = None;

        for filter in filters {
            found_ids = Some(filter.apply(found_ids.as_ref()));
        }

        found_ids.unwrap_or_default()
    }