In process of separating cache from database.
This commit is contained in:
		@@ -1,10 +1,12 @@
 | 
			
		||||
use rand::{distributions::Alphanumeric, Rng, thread_rng};
 | 
			
		||||
use super::{DBError, Store, SessionData};
 | 
			
		||||
use super::{DBError, SessionData, Store};
 | 
			
		||||
use async_std::{fs::write, path::Path};
 | 
			
		||||
use rand::{distributions::Alphanumeric, thread_rng, Rng};
 | 
			
		||||
use std::{
 | 
			
		||||
    cell::Cell,
 | 
			
		||||
    time::{Duration, Instant},
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
#[derive(Clone)]
 | 
			
		||||
enum DataType {
 | 
			
		||||
    DBMap(Store),
 | 
			
		||||
}
 | 
			
		||||
@@ -40,40 +42,24 @@ impl SessionData for DataType {
 | 
			
		||||
 | 
			
		||||
struct Entry {
 | 
			
		||||
    data: DataType,
 | 
			
		||||
    filename: String,
 | 
			
		||||
    last_used: Cell<Instant>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Entry {
 | 
			
		||||
    fn new(data_type: &str) -> Result<Self, DBError> {
 | 
			
		||||
        let data = match DataType::new(data_type) {
 | 
			
		||||
            Ok(item) => item,
 | 
			
		||||
            Err(err) => return Err(err),
 | 
			
		||||
        };
 | 
			
		||||
    async fn new(filename: String, data: DataType) -> Result<Self, DBError> {
 | 
			
		||||
        if Path::new(&filename).exists().await {
 | 
			
		||||
            return Err(DBError::new("entry already exists"));
 | 
			
		||||
        }
 | 
			
		||||
        Ok(Self {
 | 
			
		||||
            data: data,
 | 
			
		||||
            filename: filename,
 | 
			
		||||
            last_used: Cell::new(Instant::now()),
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn elapsed(&self) -> Duration {
 | 
			
		||||
        self.last_used.get().elapsed()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl SessionData for Entry {
 | 
			
		||||
    fn add(&mut self, key: &str, value: &str, data: &str) -> Result<Vec<String>, DBError> {
 | 
			
		||||
        self.last_used.set(Instant::now());
 | 
			
		||||
        self.data.add(key, value, data)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn eq(&self, key: &str, value: &str) -> Result<Vec<String>, DBError> {
 | 
			
		||||
        self.last_used.set(Instant::now());
 | 
			
		||||
        self.data.eq(key, value)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn list(&self, keys: Vec<&str>) -> Result<Vec<String>, DBError> {
 | 
			
		||||
        self.last_used.set(Instant::now());
 | 
			
		||||
        self.data.list(keys)
 | 
			
		||||
    async fn get(&self) -> Result<DataType, DBError> {
 | 
			
		||||
        Ok(self.data.clone())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -86,7 +72,7 @@ impl Cache {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[cfg(test)]
 | 
			
		||||
mod datatype {
 | 
			
		||||
mod datatype_sesssion {
 | 
			
		||||
    use super::*;
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
@@ -119,74 +105,54 @@ mod datatype {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[cfg(test)]
 | 
			
		||||
mod entry {
 | 
			
		||||
mod datatype_file {
 | 
			
		||||
    use super::*;
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn invalid_cache_type() -> Result<(), DBError> {
 | 
			
		||||
        match Entry::new("uydg") {
 | 
			
		||||
            Ok(_) => Err(DBError::new("invalid data type should raise an error")),
 | 
			
		||||
    // Test file data traits here.
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[cfg(test)]
 | 
			
		||||
mod entry {
 | 
			
		||||
    use super::*;
 | 
			
		||||
    use tempfile::tempdir;
 | 
			
		||||
 | 
			
		||||
    #[async_std::test]
 | 
			
		||||
    async fn create() {
 | 
			
		||||
        let dir = tempdir().unwrap();
 | 
			
		||||
        let mut data = DataType::new("store").unwrap();
 | 
			
		||||
        data.add("database", "roger", "moore").unwrap();
 | 
			
		||||
        let filepath = dir.path().join("wiliam");
 | 
			
		||||
        let filename = filepath.to_str().unwrap();
 | 
			
		||||
        let item = Entry::new(filename.to_string(), data.clone())
 | 
			
		||||
            .await
 | 
			
		||||
            .unwrap();
 | 
			
		||||
        let output = item.get().await.unwrap();
 | 
			
		||||
        assert_eq!(
 | 
			
		||||
            data.list(["database"].to_vec()).unwrap(),
 | 
			
		||||
            output.list(["database"].to_vec()).unwrap()
 | 
			
		||||
        );
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[async_std::test]
 | 
			
		||||
    async fn no_over_writes() -> Result<(), DBError> {
 | 
			
		||||
        let dir = tempdir().unwrap();
 | 
			
		||||
        let id = "wicked";
 | 
			
		||||
        let file = dir.path().join(id);
 | 
			
		||||
        let filename = file.to_str().unwrap();
 | 
			
		||||
        write(&file, b"previous").await.unwrap();
 | 
			
		||||
        let data = DataType::new("store").unwrap();
 | 
			
		||||
        match Entry::new(filename.to_string(), data).await {
 | 
			
		||||
            Ok(_) => {
 | 
			
		||||
                return Err(DBError::new(
 | 
			
		||||
                    "Should produce an error for an existing Entry",
 | 
			
		||||
                ))
 | 
			
		||||
            }
 | 
			
		||||
            Err(err) => {
 | 
			
		||||
                assert_eq!(err.to_string(), "invalid data type");
 | 
			
		||||
                assert_eq!(err.to_string(), "entry already exists");
 | 
			
		||||
                Ok(())
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn create_storage() {
 | 
			
		||||
        let entry = Entry::new("store").unwrap();
 | 
			
		||||
        assert!(
 | 
			
		||||
            Duration::from_secs(1) > entry.elapsed(),
 | 
			
		||||
            "Entry last used should be now."
 | 
			
		||||
        );
 | 
			
		||||
        let expected: Vec<String> = Vec::new();
 | 
			
		||||
        assert_eq!(entry.list(["database"].to_vec()).unwrap(), expected);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn add_database_to_storage() {
 | 
			
		||||
        let mut entry = Entry::new("store").unwrap();
 | 
			
		||||
        entry
 | 
			
		||||
            .last_used
 | 
			
		||||
            .set(Instant::now() - Duration::from_secs(500));
 | 
			
		||||
        let name = "roger";
 | 
			
		||||
        let id = "cormick";
 | 
			
		||||
        entry.add("database", name, id).unwrap();
 | 
			
		||||
        assert!(
 | 
			
		||||
            Duration::from_secs(1) > entry.elapsed(),
 | 
			
		||||
            "Entry last used should be now."
 | 
			
		||||
        );
 | 
			
		||||
        assert_eq!(entry.eq("database", name).unwrap(), [id].to_vec());
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn get_database_from_storage() {
 | 
			
		||||
        let mut entry = Entry::new("store").unwrap();
 | 
			
		||||
        let name = "jessica";
 | 
			
		||||
        entry.add("database", name, "rabbit").unwrap();
 | 
			
		||||
        entry
 | 
			
		||||
            .last_used
 | 
			
		||||
            .set(Instant::now() - Duration::from_secs(500));
 | 
			
		||||
        entry.eq("database", name).unwrap();
 | 
			
		||||
        assert!(
 | 
			
		||||
            Duration::from_secs(1) > entry.elapsed(),
 | 
			
		||||
            "Entry last used should be now."
 | 
			
		||||
        );
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn list_database_from_storage() {
 | 
			
		||||
        let entry = Entry::new("store").unwrap();
 | 
			
		||||
        entry
 | 
			
		||||
            .last_used
 | 
			
		||||
            .set(Instant::now() - Duration::from_secs(500));
 | 
			
		||||
        entry.list(["database"].to_vec()).unwrap();
 | 
			
		||||
        assert!(
 | 
			
		||||
            Duration::from_secs(1) > entry.elapsed(),
 | 
			
		||||
            "Entry last used should be now."
 | 
			
		||||
        );
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[cfg(test)]
 | 
			
		||||
 
 | 
			
		||||
@@ -1,6 +1,6 @@
 | 
			
		||||
mod cache;
 | 
			
		||||
mod store;
 | 
			
		||||
pub mod error;
 | 
			
		||||
mod store;
 | 
			
		||||
 | 
			
		||||
use async_std::{
 | 
			
		||||
    fs::{create_dir, read, remove_file, write},
 | 
			
		||||
@@ -8,7 +8,6 @@ use async_std::{
 | 
			
		||||
    sync::{Arc, Mutex},
 | 
			
		||||
    task::{sleep, spawn},
 | 
			
		||||
};
 | 
			
		||||
use store::Store;
 | 
			
		||||
use error::DBError;
 | 
			
		||||
use rand::{distributions::Alphanumeric, thread_rng, Rng};
 | 
			
		||||
use std::{
 | 
			
		||||
@@ -16,6 +15,7 @@ use std::{
 | 
			
		||||
    fmt, slice, str,
 | 
			
		||||
    time::{Duration, Instant},
 | 
			
		||||
};
 | 
			
		||||
use store::Store;
 | 
			
		||||
 | 
			
		||||
const DATA: &str = "data";
 | 
			
		||||
const ENTRY: &str = "databases";
 | 
			
		||||
@@ -404,11 +404,13 @@ mod data {
 | 
			
		||||
        let id = "*gsdfg";
 | 
			
		||||
        let output = mtt.db.add("database", name, id).await;
 | 
			
		||||
        assert_eq!(output.session, [id], "should update session info.");
 | 
			
		||||
        /*
 | 
			
		||||
        assert_eq!(
 | 
			
		||||
            mtt.db.list(["database"].to_vec()).await.unwrap(),
 | 
			
		||||
            [name],
 | 
			
		||||
            "Should list the databases."
 | 
			
		||||
        );
 | 
			
		||||
        */
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user