1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use crate::store::{BlobEntry, BlobTable, BlobTableInterface, StoreError};
use std::borrow::Cow;

/// The trait for data stores to implement. Allows accessing tables using a unique [`BlobTable`] identifier.
#[async_trait::async_trait]
pub trait BlobStoreInterface: Send + Sync {
    /// Gets a specific [`BlobEntry`] from the blob store.
    async fn get_in_table<'a>(&'a self, entry: &BlobEntry) -> Result<Cow<'a, [u8]>, StoreError> {
        let table = self.table(&entry.table).await?;
        table.get(&entry.key).await
    }

    /// Sets a specific [`BlobEntry`] in the data store.
    async fn set_in_table(&mut self, entry: &BlobEntry, value: &[u8]) -> Result<(), StoreError> {
        self.table(&entry.table).await?.set(&entry.key, value).await
    }

    /// Retrieve a specific [`BlobTable`], that implements [`BlobTableInterface`], inside the data store.
    async fn table<'a>(
        &'a self,
        table: &BlobTable,
    ) -> Result<Box<dyn BlobTableInterface<'a>>, StoreError>;

    /// Retrieve a specific [`BlobTable`], logging the error if it fails.
    async fn table_option(&self, table: &BlobTable) -> Option<Box<dyn BlobTableInterface>> {
        match self.table(table).await {
            Ok(t) => Some(t),
            Err(e) => {
                error!(
                    "blob_store: Failed to open table {} due to error: {:?}",
                    table, e
                );
                None
            }
        }
    }
}