convert_db: allow LMDB map size override #691
2 changed files with 24 additions and 21 deletions
|
@ -172,30 +172,31 @@ impl Db {
|
||||||
}
|
}
|
||||||
|
|
||||||
/// List of supported database engine types
|
/// List of supported database engine types
|
||||||
|
///
|
||||||
|
/// The `enum` holds list of *all* database engines that are are be supported by crate, no matter
|
||||||
|
/// if relevant feature is enabled or not. It allows us to distinguish between invalid engine
|
||||||
|
/// and valid engine, whose support is not enabled via feature flag.
|
||||||
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
|
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
|
||||||
pub enum Engine {
|
pub enum Engine {
|
||||||
#[cfg(feature = "lmdb")]
|
|
||||||
Lmdb,
|
Lmdb,
|
||||||
|
|
||||||
#[cfg(feature = "sqlite")]
|
|
||||||
Sqlite,
|
Sqlite,
|
||||||
|
|
||||||
#[cfg(feature = "sled")]
|
|
||||||
Sled,
|
Sled,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl Engine {
|
||||||
|
/// Return variant name as static `&str`
|
||||||
|
pub fn as_str(&self) -> &'static str {
|
||||||
|
match self {
|
||||||
|
Self::Lmdb => "lmdb",
|
||||||
|
Self::Sqlite => "sqlite",
|
||||||
|
Self::Sled => "sled",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl std::fmt::Display for Engine {
|
impl std::fmt::Display for Engine {
|
||||||
fn fmt(&self, fmt: &mut std::fmt::Formatter) -> std::fmt::Result {
|
fn fmt(&self, fmt: &mut std::fmt::Formatter) -> std::fmt::Result {
|
||||||
match self {
|
self.as_str().fmt(fmt)
|
||||||
#[cfg(feature = "lmdb")]
|
|
||||||
Self::Lmdb => fmt.write_str("lmdb"),
|
|
||||||
|
|
||||||
#[cfg(feature = "sqlite")]
|
|
||||||
Self::Sqlite => fmt.write_str("sqlite"),
|
|
||||||
|
|
||||||
#[cfg(feature = "sled")]
|
|
||||||
Self::Sled => fmt.write_str("sled"),
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -204,15 +205,9 @@ impl std::str::FromStr for Engine {
|
||||||
|
|
||||||
fn from_str(text: &str) -> Result<Engine> {
|
fn from_str(text: &str) -> Result<Engine> {
|
||||||
match text {
|
match text {
|
||||||
#[cfg(feature = "lmdb")]
|
|
||||||
"lmdb" | "heed" => Ok(Self::Lmdb),
|
"lmdb" | "heed" => Ok(Self::Lmdb),
|
||||||
|
|
||||||
#[cfg(feature = "sqlite")]
|
|
||||||
"sqlite" | "sqlite3" | "rusqlite" => Ok(Self::Sqlite),
|
"sqlite" | "sqlite3" | "rusqlite" => Ok(Self::Sqlite),
|
||||||
|
|
||||||
#[cfg(feature = "sled")]
|
|
||||||
"sled" => Ok(Self::Sled),
|
"sled" => Ok(Self::Sled),
|
||||||
|
|
||||||
kind => Err(Error(format!("Invalid DB engine: {}", kind).into())),
|
kind => Err(Error(format!("Invalid DB engine: {}", kind).into())),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -91,5 +91,13 @@ fn open_db(path: PathBuf, engine: Engine, open: &OpenDbOpt) -> Result<Db> {
|
||||||
let db = env_builder.open(&path)?;
|
let db = env_builder.open(&path)?;
|
||||||
Ok(lmdb_adapter::LmdbDb::init(db))
|
Ok(lmdb_adapter::LmdbDb::init(db))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Pattern is unreachable when all supported DB engines are compiled into binary. The allow
|
||||||
|
// attribute is added so that we won't have to change this match in case stop building
|
||||||
|
// support for one or more engines by default.
|
||||||
|
#[allow(unreachable_patterns)]
|
||||||
|
engine => Err(Error(
|
||||||
|
format!("Engine support not available in this build: {}", engine).into(),
|
||||||
|
)),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue