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
mod auto_increment;
mod base;
mod error;
mod mutable;
mod util;
#[cfg(not(feature = "alter-table"))]
impl crate::AlterTable for SledDatabase {}
#[cfg(not(feature = "auto-increment"))]
impl crate::AutoIncrement for SledDatabase {}
use {
crate::{DBFull, Database, Error, Result, Schema},
error::err_into,
sled::{self, Config, Db},
std::convert::TryFrom,
};
#[derive(Debug, Clone)]
pub struct SledDatabase {
tree: Db,
}
impl DBFull for SledDatabase {}
impl SledDatabase {
pub fn new(filename: &str) -> Result<Self> {
let tree = sled::open(filename).map_err(err_into)?;
Ok(Self { tree })
}
}
impl Database {
pub fn new_sled(sled: SledDatabase) -> Self {
Self::new(Box::new(sled))
}
}
impl TryFrom<Config> for SledDatabase {
type Error = Error;
fn try_from(config: Config) -> Result<Self> {
let tree = config.open().map_err(err_into)?;
Ok(Self { tree })
}
}
fn fetch_schema(tree: &Db, table_name: &str) -> Result<(String, Option<Schema>)> {
let key = format!("schema/{}", table_name);
let value = tree.get(&key.as_bytes()).map_err(err_into)?;
let schema = value
.map(|v| bincode::deserialize(&v))
.transpose()
.map_err(err_into)?;
Ok((key, schema))
}