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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
use rmp_serde::{decode, encode};
use std::fmt::Debug;
use std::fs;
use std::marker::PhantomData;
use std::path::{Path, PathBuf};
extern crate serde;
use serde::{de::DeserializeOwned, Serialize};
pub struct Fsdb {
dir: PathBuf,
}
pub struct Bucket<V> {
dir: PathBuf,
_v: PhantomData<V>,
}
#[derive(Debug, thiserror::Error)]
pub enum Error {
#[error("io error: {0}")]
Io(#[from] std::io::Error),
#[error("encode error: {0}")]
Encode(#[from] rmp_serde::encode::Error),
#[error("dncode error: {0}")]
Decode(#[from] rmp_serde::decode::Error),
}
type Result<T> = std::result::Result<T, Error>;
impl Fsdb {
pub fn new(dir: &str) -> Result<Self> {
if !Path::new(dir).exists() {
fs::create_dir(dir)?;
}
Ok(Self { dir: dir.into() })
}
pub fn bucket<V: Serialize + DeserializeOwned>(&self, p: &str) -> Result<Bucket<V>> {
let mut dir = self.dir.clone();
dir.push::<PathBuf>(p.into());
if !Path::new(&dir).exists() {
fs::create_dir(dir.clone())?;
}
Ok(Bucket {
dir: dir.into(),
_v: PhantomData,
})
}
}
impl<V: Serialize + DeserializeOwned> Bucket<V> {
pub fn exists(&self, key: &str) -> bool {
let mut path = self.dir.clone();
path.push(key);
path.exists()
}
pub fn put(&self, key: &str, value: V) -> Result<()> {
let mut path = self.dir.clone();
path.push::<PathBuf>(key.into());
let mut f = fs::File::create(path.clone())?;
encode::write(&mut f, &value)?;
Ok(())
}
pub fn get(&self, key: &str) -> Result<V> {
let mut path = self.dir.clone();
path.push(key);
let f = fs::File::open(path)?;
Ok(decode::from_read(f)?)
}
pub fn remove(&self, key: &str) -> Result<()> {
let mut path = self.dir.clone();
path.push::<PathBuf>(key.into());
Ok(std::fs::remove_file(path)?)
}
pub fn list(&self) -> Result<Vec<String>> {
let path = self.dir.clone();
let paths = fs::read_dir(path)?;
let mut r = Vec::new();
paths.for_each(|name| {
if let Ok(na) = name {
if let Ok(n) = na.file_name().into_string() {
r.push(n);
}
}
});
Ok(r)
}
}
#[cfg(test)]
mod tests {
use crate::Fsdb;
use serde::{Deserialize, Serialize};
#[derive(Serialize, Deserialize, Debug, PartialEq, Eq, Clone)]
struct Thing {
n: u8,
}
#[test]
fn test_db() {
let db = Fsdb::new("testdb").expect("fail Fsdb::new");
let b = db.bucket("hi").expect("fail bucket");
let t1 = Thing { n: 1 };
b.put("key", t1.clone()).expect("failed to save");
let t2: Thing = b.get("key").expect("fail to load");
println!("t {:?}", t2.clone());
assert_eq!(t1, t2);
let list = b.list().expect("fail list");
assert_eq!(list, vec!["key".to_string()]);
}
}