diff options
-rw-r--r-- | src/main.rs | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/src/main.rs b/src/main.rs index 65b4430..be109d0 100644 --- a/src/main.rs +++ b/src/main.rs | |||
@@ -43,11 +43,17 @@ use log::{error}; | |||
43 | async fn main() { | 43 | async fn main() { |
44 | log4rs::init_file("log.conf.yml", log4rs::config::Deserializers::default()).unwrap(); | 44 | log4rs::init_file("log.conf.yml", log4rs::config::Deserializers::default()).unwrap(); |
45 | 45 | ||
46 | let configs = vec!["config.yaml"]; | 46 | let mut args: Vec<String> = std::env::args().collect(); |
47 | 47 | ||
48 | let combined_routes = configs.into_iter() | 48 | if args.len() == 1 { |
49 | // config.yaml is the default configuration file | ||
50 | args.push("config.yaml".to_string()); | ||
51 | } | ||
52 | |||
53 | let combined_routes = args.into_iter() | ||
54 | .skip(1) // Skip the program name | ||
49 | .filter_map(|filename| { | 55 | .filter_map(|filename| { |
50 | match Config::read(filename) { | 56 | match Config::read(&filename) { |
51 | Some(config) => Some(routes::network(Db::new(config))), | 57 | Some(config) => Some(routes::network(Db::new(config))), |
52 | None => None, | 58 | None => None, |
53 | } | 59 | } |