Optimize perf teardown performance

crates/runner-shared/src/unwind_data.rs

@@ -3,6 +3,7 @@ use core::{
     hash::{Hash, Hasher},
 };
 use serde::{Deserialize, Serialize};
+use std::io::BufWriter;
 use std::{hash::DefaultHasher, ops::Range};
 
 pub const UNWIND_FILE_EXT: &str = "unwind_data";
@@ -41,9 +42,13 @@ impl UnwindData {
             log::warn!("{} {:x?}", self.path, self.avma_range);
         }
 
-        let mut writer = std::fs::File::create(path.as_ref())?;
         let compat = UnwindDataCompat::V2(self.clone());
-        bincode::serialize_into(&mut writer, &compat)?;
+        let file = std::fs::File::create(path.as_ref())?;
+        const BUFFER_SIZE: usize = 256 * 1024 /* 256 KB */;
+
+        let writer = BufWriter::with_capacity(BUFFER_SIZE, file);
+        bincode::serialize_into(writer, &compat)?;
+
         Ok(())
     }
 }

src/executor/shared/fifo.rs

@@ -147,7 +147,7 @@ impl RunnerFifo {
                 break;
             }
 
-            let result = tokio::time::timeout(Duration::from_secs(5), self.recv_cmd()).await;
+            let result = tokio::time::timeout(Duration::from_secs(1), self.recv_cmd()).await;
             let cmd = match result {
                 Ok(Ok(cmd)) => cmd,
                 Ok(Err(e)) => {

src/executor/wall_time/perf/debug_info.rs

@@ -101,7 +101,7 @@ impl ProcessDebugInfo {
                     modules.push(module_debug_info);
                 }
                 Err(error) => {
-                    debug!("Failed to load debug info for module {path:?}: {error}");
+                    trace!("Failed to load debug info for module {path:?}: {error}");
                 }
             }
         }

src/executor/wall_time/perf/parse_perf_file.rs

@@ -82,7 +82,7 @@ pub(super) fn parse_for_memmap2<P: AsRef<Path>>(perf_file_path: P) -> Result<Mem
             continue;
         }
 
-        debug!(
+        trace!(
             "Pid {}: {:016x}-{:016x} {:08x} {:?} (Prot {:?})",
             record.pid,
             record.address,
@@ -106,7 +106,7 @@ pub(super) fn parse_for_memmap2<P: AsRef<Path>>(perf_file_path: P) -> Result<Mem
                 end_addr,
                 record.page_offset,
             );
-        debug!("Added symbols mapping for module {record_path_string:?}");
+        trace!("Added symbols mapping for module {record_path_string:?}");
 
         match UnwindData::new(
             record_path_string.as_bytes(),
@@ -120,7 +120,7 @@ pub(super) fn parse_for_memmap2<P: AsRef<Path>>(perf_file_path: P) -> Result<Mem
                     .entry(record.pid)
                     .or_default()
                     .push(unwind_data);
-                debug!(
+                trace!(
                     "Added unwind data for {record_path_string} ({:x} - {:x})",
                     record.address, end_addr
                 );

src/executor/wall_time/perf/perf_map.rs

@@ -5,7 +5,7 @@ use object::{Object, ObjectSymbol, ObjectSymbolTable};
 use std::{
     collections::HashMap,
     fmt::Debug,
-    io::Write,
+    io::{BufWriter, Write},
     path::{Path, PathBuf},
 };
 
@@ -139,14 +139,16 @@ impl ModuleSymbols {
     }
 
     pub fn append_to_file<P: AsRef<Path>>(&self, path: P) -> anyhow::Result<()> {
-        let mut file = std::fs::OpenOptions::new()
+        let file = std::fs::OpenOptions::new()
             .create(true)
             .append(true)
             .open(path)?;
+        const BUFFER_SIZE: usize = 256 * 1024 /* 256 KB */;
 
+        let mut writer = BufWriter::with_capacity(BUFFER_SIZE, file);
         for symbol in &self.symbols {
             writeln!(
-                file,
+                writer,
                 "{:x} {:x} {}",
                 symbol.addr.wrapping_add(self.load_bias),
                 symbol.size,