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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

#[cfg(feature = "wasm")]
pub mod idb;
#[cfg(not(feature = "wasm"))]
pub mod kv;
use async_trait::async_trait;

#[cfg(feature = "wasm")]
pub use self::idb::IDBStorage;
#[cfg(not(feature = "wasm"))]
pub use self::kv::KvStorage;
use crate::err::Result;

/// Persistence Storage read and write functions
#[cfg_attr(feature = "wasm", async_trait(?Send))]
#[cfg_attr(not(feature = "wasm"), async_trait)]
pub trait PersistenceStorageReadAndWrite<K, V>: PersistenceStorageOperation {
    /// Get a cache entry by `key`.
    async fn get(&self, key: &K) -> Result<V>;

    /// Put `entry` in the cache under `key`.
    async fn put(&self, key: &K, entry: &V) -> Result<()>;

    async fn get_all(&self) -> Result<Vec<(K, V)>>;
}

/// Persistence Storage remove functions
#[cfg_attr(feature = "wasm", async_trait(?Send))]
#[cfg_attr(not(feature = "wasm"), async_trait)]
pub trait PersistenceStorageRemove<K>: PersistenceStorageOperation {
    /// Remove an `entry` by `key`.
    async fn remove(&self, key: &K) -> Result<()>;
}

/// Persistence Storage Operations
#[cfg_attr(feature = "wasm", async_trait(?Send))]
#[cfg_attr(not(feature = "wasm"), async_trait)]
pub trait PersistenceStorageOperation {
    /// Clear Storage.
    /// All `Entry` will be deleted.
    async fn clear(&self) -> Result<()>;

    /// Get the current storage usage, if applicable.
    async fn count(&self) -> Result<u64>;

    /// Get the maximum storage size, if applicable.
    async fn max_size(&self) -> Result<usize>;

    /// Get the storage size, if applicable.
    async fn total_size(&self) -> Result<usize>;

    /// Prune database storage
    async fn prune(&self) -> Result<()>;

    /// Close db instance
    async fn close(self) -> Result<()>;
}