Verify database path parameter on command-line
This commit is contained in:
parent
53a4f76516
commit
643344151b
1 changed files with 18 additions and 0 deletions
|
@ -30,6 +30,23 @@ fn file_valid(v: String) -> Result<(), String> {
|
||||||
))
|
))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn path_valid(v: String) -> Result<(), String> {
|
||||||
|
if v == "memory" {
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
if v.starts_with("file://") {
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
if v.starts_with("tikv://") {
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
Err(String::from(
|
||||||
|
"\
|
||||||
|
Provide a valid database path paramater\
|
||||||
|
",
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
fn conn_valid(v: String) -> Result<(), String> {
|
fn conn_valid(v: String) -> Result<(), String> {
|
||||||
if v.starts_with("https://") {
|
if v.starts_with("https://") {
|
||||||
return Ok(());
|
return Ok(());
|
||||||
|
@ -116,6 +133,7 @@ pub fn init() {
|
||||||
Arg::with_name("path")
|
Arg::with_name("path")
|
||||||
.index(1)
|
.index(1)
|
||||||
.required(true)
|
.required(true)
|
||||||
|
.validator(path_valid)
|
||||||
.default_value("memory")
|
.default_value("memory")
|
||||||
.help("Database path used for storing data"),
|
.help("Database path used for storing data"),
|
||||||
)
|
)
|
||||||
|
|
Loading…
Reference in a new issue