Compare commits
No commits in common. "bdcee41dcf67ecba377da7da02e76ae0ff0a4838" and "47b9a071a4995af96e9d9c6b4aea8fcde45781d5" have entirely different histories.
bdcee41dcf
...
47b9a071a4
@ -1,33 +1,17 @@
|
||||
pub mod error;
|
||||
|
||||
use async_std::{
|
||||
fs::create_dir,
|
||||
path::Path,
|
||||
sync::{Arc, Mutex},
|
||||
};
|
||||
use async_std::{fs::create_dir, path::Path};
|
||||
use error::DBError;
|
||||
use rand::{distributions::Alphanumeric, thread_rng, Rng};
|
||||
use std::{collections::HashMap, fmt};
|
||||
|
||||
const DATA: &str = "data";
|
||||
|
||||
#[derive(Clone)]
|
||||
enum CacheEntry {
|
||||
Raw(String),
|
||||
}
|
||||
|
||||
impl fmt::Display for CacheEntry {
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||
match self {
|
||||
CacheEntry::Raw(s) => write!(f, "{}", s),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Clone)]
|
||||
pub struct MoreThanText {
|
||||
cache: Arc<Mutex<HashMap<String, CacheEntry>>>,
|
||||
}
|
||||
pub struct MoreThanText;
|
||||
|
||||
impl MoreThanText {
|
||||
pub async fn new(dir: &str) -> Result<Self, DBError> {
|
||||
@ -42,22 +26,13 @@ impl MoreThanText {
|
||||
}
|
||||
}
|
||||
}
|
||||
Ok(Self {
|
||||
cache: Arc::new(Mutex::new(HashMap::new())),
|
||||
})
|
||||
Ok(Self {})
|
||||
}
|
||||
|
||||
async fn add_entry(&self, entry: CacheEntry) -> String {
|
||||
async fn add_entry(&self, _entry: CacheEntry) -> String {
|
||||
let id: String = thread_rng().sample_iter(&Alphanumeric).take(32).collect();
|
||||
let mut cache = self.cache.lock().await;
|
||||
cache.insert(id.clone(), entry);
|
||||
return id;
|
||||
}
|
||||
|
||||
async fn get_entry(&self, id: &str) -> CacheEntry {
|
||||
let cache = self.cache.lock().await;
|
||||
cache.get(id).unwrap().clone()
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
@ -135,14 +110,4 @@ mod cache {
|
||||
let id2 = mtt.db.add_entry(data2).await;
|
||||
assert_ne!(id1, id2, "Ids should be unique.")
|
||||
}
|
||||
|
||||
#[async_std::test]
|
||||
async fn retrieve_cache() {
|
||||
let mtt = MTT::new().await;
|
||||
let data = "something";
|
||||
let expected = CacheEntry::Raw(data.to_string());
|
||||
let id = mtt.db.add_entry(expected).await;
|
||||
let output = mtt.db.get_entry(&id).await;
|
||||
assert_eq!(output.to_string(), data);
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user