Fix

raviqqe · raviqqe · Apr 28, 2024 · Apr 28, 2024 · Apr 28, 2024 · Apr 28, 2024
commit be782b8229a43dc989476e453d34684ded1c3e8c
diff --git a/profiler/src/lib.rs b/profiler/src/lib.rs
@@ -9,6 +9,7 @@ mod flamegraph;
 mod read;
 mod record;
 mod stack_profiler;
+mod write;
 
 pub use collapse::collapse_stacks;
 pub use duration::calculate_durations;
@@ -19,6 +20,7 @@ pub use record::{
     DurationRecord, ProcedureOperation, ProcedureRecord, Record, Stack, StackedRecord,
 };
 pub use stack_profiler::StackProfiler;
+pub use write::write_records;
 
 const COLUMN_SEPARATOR: char = '\t';
 const FRAME_SEPARATOR: char = ';';

diff --git a/profiler/src/write.rs b/profiler/src/write.rs
@@ -0,0 +1,59 @@
+use crate::Record;
+use std::io::{self, Write};
+
+pub fn write_records(
+    records: impl IntoIterator<Item = impl Record>,
+    mut writer: impl Write,
+) -> Result<(), io::Error> {
+    for record in records {
+        writeln!(writer, "{}", record)?;
+    }
+
+    Ok(())
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::{ProcedureOperation, ProcedureRecord, Stack};
+    use indoc::indoc;
+    use pretty_assertions::assert_eq;
+
+    #[test]
+    fn write() {
+        let mut buffer = vec![];
+
+        write_records(
+            [
+                ProcedureRecord::new(
+                    ProcedureOperation::Call,
+                    Stack::new(vec![Some("baz".into())]),
+                    0,
+                ),
+                ProcedureRecord::new(
+                    ProcedureOperation::Return,
+                    Stack::new(vec![Some("foo".into()), Some("bar".into())]),
+                    42,
+                ),
+                ProcedureRecord::new(
+                    ProcedureOperation::ReturnCall,
+                    Stack::new(vec![None; 3]),
+                    2045,
+                ),
+            ],
+            &mut buffer,
+        )
+        .unwrap();
+
+        assert_eq!(
+            buffer,
+            indoc!(
+                b"
+                call\tbaz\t0
+                return\tfoo;bar\t42
+                return_call\t;;\t2045
+                "
+            )
+        );
+    }
+}