rings_core/storage/
memory.rs

1use async_trait::async_trait;
2use dashmap::DashMap;
3
4use crate::error::Result;
5use crate::storage::KvStorageInterface;
6
7#[derive(Debug, Default)]
8pub struct MemStorage<V>
9where V: Clone
10{
11    table: DashMap<String, V>,
12}
13
14impl<V> MemStorage<V>
15where V: Clone
16{
17    pub fn new() -> Self {
18        Self {
19            table: DashMap::default(),
20        }
21    }
22}
23
24#[cfg_attr(feature = "wasm", async_trait(?Send))]
25#[cfg_attr(not(feature = "wasm"), async_trait)]
26impl<V> KvStorageInterface<V> for MemStorage<V>
27where V: Clone + Send + Sync
28{
29    async fn get(&self, key: &str) -> Result<Option<V>> {
30        Ok(self.table.get(&key.to_string()).map(|v| v.value().clone()))
31    }
32
33    async fn put(&self, key: &str, value: &V) -> Result<()> {
34        self.table.insert(key.to_string(), value.clone());
35        Ok(())
36    }
37
38    async fn get_all(&self) -> Result<Vec<(String, V)>> {
39        Ok(self.table.clone().into_iter().collect())
40    }
41
42    async fn remove(&self, key: &str) -> Result<()> {
43        match self.get(key).await? {
44            Some(_) => self.table.remove(key),
45            None => None,
46        };
47        Ok(())
48    }
49
50    async fn clear(&self) -> Result<()> {
51        self.table.clear();
52        Ok(())
53    }
54
55    async fn count(&self) -> Result<u32> {
56        Ok(self.table.len() as u32)
57    }
58}
59
60#[cfg(not(feature = "wasm"))]
61#[cfg(test)]
62mod tests {
63    use super::*;
64    use crate::ecc::SecretKey;
65
66    #[tokio::test]
67    async fn memstorage_basic_interface_should_work() {
68        let store = MemStorage::new();
69        let addr = SecretKey::random().address().to_string();
70
71        assert_eq!(store.get(&addr).await.unwrap(), None);
72
73        store.put(&addr, &"value 1".to_string()).await.unwrap();
74        assert_eq!(store.get(&addr).await.unwrap(), Some("value 1".into()));
75
76        store.put(&addr, &"value 2".to_string()).await.unwrap();
77        assert_eq!(store.get(&addr).await.unwrap(), Some("value 2".into()));
78    }
79}
rings_core/storage/memory.rs

rings_core/storage/
memory.rs
