benchmark: buffer write in rust providers

cdump · Aug 27, 2024 · 805c78d · 805c78d
1 parent c650665
commit 805c78d
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 14 deletions.
diff --git a/benchmark/providers/evm-hound-rs/src/main.rs b/benchmark/providers/evm-hound-rs/src/main.rs
@@ -1,5 +1,5 @@
 use std::collections::HashMap;
-use std::io::Write;
+use std::io::{BufWriter, Write};
 use std::{env, fs};
 
 #[derive(Debug, serde::Deserialize)]
@@ -43,9 +43,10 @@ fn main() -> std::io::Result<()> {
         ret.insert(fname, string_selectors);
     }
 
-    let mut file = fs::File::create(outfile)?;
-    let _ = serde_json::to_writer(&mut file, &ret);
-    file.flush()?;
+    let file = fs::File::create(outfile)?;
+    let mut bw = BufWriter::new(file);
+    let _ = serde_json::to_writer(&mut bw, &ret);
+    bw.flush()?;
 
     Ok(())
 }
diff --git a/benchmark/providers/evmole-rs/src/main.rs b/benchmark/providers/evmole-rs/src/main.rs
@@ -1,5 +1,5 @@
 use std::collections::HashMap;
-use std::io::Write;
+use std::io::{BufWriter, Write};
 use std::fs;
 
 use clap::Parser;
@@ -92,13 +92,14 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
         }
     }
 
-    let mut file = fs::File::create(cfg.output_file)?;
+    let file = fs::File::create(cfg.output_file)?;
+    let mut bw = BufWriter::new(file);
     if cfg.mode == "arguments" {
-        let _ = serde_json::to_writer(&mut file, &ret_arguments);
+        let _ = serde_json::to_writer(&mut bw, &ret_arguments);
     } else {
-        let _ = serde_json::to_writer(&mut file, &ret_selectors);
+        let _ = serde_json::to_writer(&mut bw, &ret_selectors);
     }
-    file.flush()?;
+    bw.flush()?;
 
     Ok(())
 }
diff --git a/benchmark/providers/heimdall-rs/src/main.rs b/benchmark/providers/heimdall-rs/src/main.rs
@@ -1,5 +1,5 @@
 use std::collections::HashMap;
-use std::io::Write;
+use std::io::{BufWriter, Write};
 use std::{env, fs};
 
 use heimdall_core::heimdall_decompiler::DecompilerArgsBuilder;
@@ -98,13 +98,14 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
         }
     }
 
-    let mut file = fs::File::create(outfile)?;
+    let file = fs::File::create(outfile)?;
+    let mut bw = BufWriter::new(file);
     if mode == "arguments" {
-        let _ = serde_json::to_writer(&mut file, &ret_arguments);
+        let _ = serde_json::to_writer(&mut bw, &ret_arguments);
     } else {
-        let _ = serde_json::to_writer(&mut file, &ret_selectors);
+        let _ = serde_json::to_writer(&mut bw, &ret_selectors);
     }
-    file.flush()?;
+    bw.flush()?;
 
     Ok(())
 }