blob: e43bbf6f21cbc19cb7e9c9b4cf7624cf7f46b7a0 (
plain)
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
|
use std::{env, sync::Arc};
#[macro_use]
pub mod macros;
pub mod rdb;
pub mod resp_commands;
pub mod resp_parser;
pub mod shared_cache;
#[derive(Debug, Default)]
pub struct Config {
pub dir: Option<String>,
pub dbfilename: Option<String>,
pub port: Option<String>,
}
pub type SharedConfig = Arc<Option<Config>>;
impl Config {
pub fn new() -> Result<Option<Config>, String> {
let args: Vec<String> = env::args().collect();
if args.len() == 1 {
return Ok(None);
}
let mut dir = None;
let mut dbfilename = None;
let mut port = None;
let mut i = 1; // Skip program name
while i < args.len() {
match args[i].as_str() {
"--dir" => {
if i + 1 >= args.len() {
return Err("--dir requires a value".to_string());
}
dir = Some(args[i + 1].clone());
i += 2;
}
"--dbfilename" => {
if i + 1 >= args.len() {
return Err("--dbfilename requires a value".to_string());
}
dbfilename = Some(args[i + 1].clone());
i += 2;
}
"--port" => {
if i + 1 >= args.len() {
return Err("--dbfilename requires a value".to_string());
}
port = Some(args[i + 1].clone());
i += 2;
}
_ => {
return Err(format!("Unknown argument: {}", args[i]));
}
}
}
Ok(Some(Config {
dir,
dbfilename,
port,
}))
}
}
|
