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
use snap;
use std::convert::From;
use std::fs;
use std::io;
use std::io::{Read, Write};
use std::path::Path;
#[derive(Clone, Copy, Debug, Eq, Hash, Ord, PartialEq, PartialOrd)]
pub enum Error {
NotDir,
NotEmpty,
NotFound,
IoErrorKind(io::ErrorKind),
}
impl From<io::Error> for Error {
fn from(err: io::Error) -> Self {
Error::IoErrorKind(err.kind())
}
}
const MIN_COMPRESS_SIZE: usize = 2_048;
pub fn get_raw_file(dir: &Path, key: &str) -> Result<fs::File, Error> {
if !dir.is_dir() {
return Err(Error::NotDir);
}
match fs::File::open(dir.join(key)) {
Ok(file) => Ok(file),
Err(err) => Err(match err.kind() {
io::ErrorKind::NotFound => Error::NotFound,
_ => Error::from(err),
}),
}
}
pub fn write(dir: &Path, key: &str, value: &[u8]) -> Result<(), Error> {
if !dir.is_dir() {
return Err(Error::NotDir);
}
let mut file = fs::File::create(dir.join(key))?;
if value.len() >= MIN_COMPRESS_SIZE {
file.write_all(&[1])?;
let mut wtr = snap::Writer::new(file);
wtr.write_all(&value)?;
} else {
file.write_all(&[0])?;
file.write_all(&value)?;
}
Ok(())
}
pub fn read(dir: &Path, key: &str) -> Result<Vec<u8>, Error> {
let mut value: Vec<u8> = Vec::new();
read_into(&dir, &key, &mut value)?;
Ok(value)
}
pub fn read_into(dir: &Path, key: &str, mut buf: &mut Vec<u8>) -> Result<(), Error> {
if !buf.is_empty() {
return Err(Error::NotEmpty);
}
let mut file = get_raw_file(&dir, &key)?;
let mut is_compressed = [0];
file.read_exact(&mut is_compressed)?;
if is_compressed[0] == 0 {
file.read_to_end(&mut buf)?;
} else {
let mut rdr = snap::Reader::new(file);
rdr.read_to_end(&mut buf)?;
}
Ok(())
}
pub fn delete(dir: &Path, key: &str) -> Result<(), Error> {
get_raw_file(&dir, &key)?;
fs::remove_file(dir.join(key))?;
Ok(())
}