Bläddra i källkod

util: Support conditional pretty-writing of JSON files.

parazyd 3 år sedan
förälder
incheckning
1004b78ece

+ 1 - 1
bin/darkirc/src/main.rs

@@ -112,7 +112,7 @@ async fn realmain(settings: Args, executor: Arc<smol::Executor<'_>>) -> Result<(
 
 
         if settings.output.is_some() {
         if settings.output.is_some() {
             let datastore = expand_path(&settings.output.unwrap())?;
             let datastore = expand_path(&settings.output.unwrap())?;
-            save_json_file(&datastore, &kp)?;
+            save_json_file(&datastore, &kp, false)?;
         } else {
         } else {
             println!("Generated keypair:\n{}", kp);
             println!("Generated keypair:\n{}", kp);
         }
         }

+ 1 - 1
bin/tau/taud/src/month_tasks.rs

@@ -98,7 +98,7 @@ impl MonthTasks {
 
 
     pub fn save(&self, dataset_path: &Path) -> TaudResult<()> {
     pub fn save(&self, dataset_path: &Path) -> TaudResult<()> {
         debug!(target: "tau", "MonthTasks::save()");
         debug!(target: "tau", "MonthTasks::save()");
-        save_json_file::<Self>(&Self::get_path(&self.created_at, dataset_path), self)
+        save_json_file::<Self>(&Self::get_path(&self.created_at, dataset_path), self, true)
             .map_err(TaudError::Darkfi)
             .map_err(TaudError::Darkfi)
     }
     }
 
 

+ 1 - 1
bin/tau/taud/src/task_info.rs

@@ -153,7 +153,7 @@ impl TaskInfo {
 
 
     pub fn save(&self, dataset_path: &Path) -> TaudResult<()> {
     pub fn save(&self, dataset_path: &Path) -> TaudResult<()> {
         debug!(target: "tau", "TaskInfo::save()");
         debug!(target: "tau", "TaskInfo::save()");
-        save_json_file::<Self>(&Self::get_path(&self.ref_id, dataset_path), self)
+        save_json_file::<Self>(&Self::get_path(&self.ref_id, dataset_path), self, true)
             .map_err(TaudError::Darkfi)?;
             .map_err(TaudError::Darkfi)?;
 
 
         if self.get_state() == "stop" {
         if self.get_state() == "stop" {

+ 1 - 1
src/event_graph/model.rs

@@ -102,7 +102,7 @@ where
         let tree = self.event_map.clone();
         let tree = self.event_map.clone();
         let ser_tree = serialize(&tree);
         let ser_tree = serialize(&tree);
 
 
-        save_json_file(&path, &ser_tree)?;
+        save_json_file(&path, &ser_tree, false)?;
 
 
         info!("Tree is saved to disk");
         info!("Tree is saved to disk");
 
 

+ 8 - 2
src/util/file.rs

@@ -48,8 +48,14 @@ pub fn load_json_file<T: DeserializeOwned>(path: &Path) -> Result<T> {
     Ok(value)
     Ok(value)
 }
 }
 
 
-pub fn save_json_file<T: Serialize>(path: &Path, value: &T) -> Result<()> {
+pub fn save_json_file<T: Serialize>(path: &Path, value: &T, pretty: bool) -> Result<()> {
     let file = File::create(path)?;
     let file = File::create(path)?;
-    serde_json::to_writer_pretty(file, value)?;
+
+    if pretty {
+        serde_json::to_writer_pretty(file, value)?;
+    } else {
+        serde_json::to_writer(file, value)?;
+    }
+
     Ok(())
     Ok(())
 }
 }