From ddca67d60302db99f4af81d0e6d7e9ad33de6846 Mon Sep 17 00:00:00 2001
From: griffi-gh <prasol258@gmail.com>
Date: Sun, 1 Sep 2024 19:56:34 +0200
Subject: [PATCH] exit after writing workloads

---
 kubi/src/lib.rs | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/kubi/src/lib.rs b/kubi/src/lib.rs
index 654c4ab..b3b6fb5 100644
--- a/kubi/src/lib.rs
+++ b/kubi/src/lib.rs
@@ -229,11 +229,13 @@ pub fn kubi_main(
   world.add_workload(after_render);
 
   //Save _visualizer.json
-  #[cfg(feature = "generate_visualizer_data")]
-  std::fs::write(
-    "_visualizer.json",
-    serde_json::to_string(&world.workloads_info()).unwrap(),
-  ).unwrap();
+  #[cfg(feature = "generate_visualizer_data")] {
+    std::fs::write(
+      "_visualizer.json",
+      serde_json::to_string(&world.workloads_info()).unwrap(),
+    ).unwrap();
+    std::process::exit(0);
+  }
 
   //Run pre-startup procedure
   world.run_workload(pre_startup).unwrap();