use crate::{ migrate::{SledDb, SledIter, SledTree}, UploadError, }; use sled as sled034; use std::path::PathBuf; const SLED_034: &str = "db-0.34"; pub(crate) fn exists(mut base: PathBuf) -> bool { base.push("sled"); base.push(SLED_034); std::fs::metadata(base).is_ok() } pub(crate) fn open(mut base: PathBuf) -> Result { base.push("sled"); base.push(SLED_034); Ok(sled034::open(base)?) } impl SledDb for sled034::Db { type SledTree = sled034::Tree; fn open_tree(&self, name: &str) -> Result { Ok(sled034::Db::open_tree(self, name)?) } fn self_tree(&self) -> &Self::SledTree { &*self } } impl SledTree for sled034::Tree { fn get(&self, key: K) -> Result>, UploadError> where K: AsRef<[u8]>, { Ok(sled034::Tree::get(self, key)?.map(|v| Vec::from(v.as_ref()))) } fn insert(&self, key: K, value: V) -> Result<(), UploadError> where K: AsRef<[u8]>, V: AsRef<[u8]>, { Ok(sled034::Tree::insert(self, key, value.as_ref().to_vec()).map(|_| ())?) } fn iter(&self) -> SledIter { Box::new(sled034::Tree::iter(self).map(|res| { res.map(|(k, v)| (k.as_ref().to_vec(), v.as_ref().to_vec())) .map_err(UploadError::from) })) } fn range(&self, range: R) -> SledIter where K: AsRef<[u8]>, R: std::ops::RangeBounds, { Box::new(sled034::Tree::range(self, range).map(|res| { res.map(|(k, v)| (k.as_ref().to_vec(), v.as_ref().to_vec())) .map_err(UploadError::from) })) } }