rogalik_persist 0.3.0

A cross platform persistence backend for the Rogali engine
Documentation 
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

use anyhow::Result;
use serde::{de::DeserializeOwned, Serialize};
use std::{
    env::current_dir,
    fs::{File, remove_file},
    io::{Read, Write},
    path::PathBuf
};

use crate::KVStore;

pub struct FileStore;
impl FileStore {
    fn get_path(key: &str, base: Option<&str>) -> Result<PathBuf> {
        let dir = match base {
            Some(a) => a.into(),
            None => current_dir()?
        };
        Ok(dir.as_path().join(format!("{}.bin", key)))
    }
}

impl KVStore for FileStore {
    fn load<T: DeserializeOwned + Serialize>(key: &str, path: Option<&str>) -> Result<T> {
        let f = File::open(FileStore::get_path(key, path)?)?;
        Ok(bincode::deserialize_from(f)?)
    }
    fn store<T: DeserializeOwned + Serialize>(key: &str, value: &T, path: Option<&str>) -> Result<()> {
        let mut f = File::create(FileStore::get_path(key, path)?)?;
        bincode::serialize_into(&mut f, value)?;
        Ok(())
    }
    fn load_raw(key: &str, path: Option<&str>) -> Result<Vec<u8>> {
        let mut f = File::open(FileStore::get_path(key, path)?)?;
        let mut buf = Vec::new();
        f.read_to_end(&mut buf)?;
        Ok(buf)
    }
    fn store_raw(key: &str, value: &[u8], path: Option<&str>) -> Result<()> {
        let mut f = File::create(FileStore::get_path(key, path)?)?;
        f.write(value)?;
        Ok(())
    }
    fn remove(key: &str, path: Option<&str>) -> Result<()> {
        Ok(remove_file(FileStore::get_path(key, path)?)?)
    }
}