Assert that output directory exists

ecrs-org · Oct 9, 2023 · cff5813 · cff5813
1 parent 55fa7f7
commit cff5813
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/examples/jssp/main.rs b/examples/jssp/main.rs
@@ -58,7 +58,8 @@ fn run_with_ecrs(instance: JsspInstance, _args: Args) {
 
 fn run() {
     let args = cli::parse_args();
-
+
+    util::assert_dir_exists(args.output_dir.as_ref());
     let event_map = util::create_event_map(args.output_dir.as_ref());
     if let Err(err) = logging::init_logging(&event_map) {
         panic!("Logger initialization failed with error: {err}");

diff --git a/examples/jssp/util.rs b/examples/jssp/util.rs
@@ -26,3 +26,14 @@ pub fn create_event_map(base_dir: &Path) -> HashMap<String, PathBuf> {
         ("iterinfo".to_owned(), base_dir.join("event_iterinfo.csv"))
     ])
 }
+
+pub fn assert_dir_exists(dir: &Path) {
+    if dir.is_dir() {
+        return;
+    }
+
+    match std::fs::create_dir_all(dir) {
+        Ok(()) => return,
+        Err(err) => panic!("Failed to create outuput directory with error {err}"),
+    };
+}