rsigma_runtime/io/
file.rs

1use std::fs::{File, OpenOptions};
2use std::io::{BufWriter, Write};
3use std::path::Path;
4
5use rsigma_eval::ProcessResult;
6
7use crate::error::RuntimeError;
8
9/// Appends ProcessResult as NDJSON to a file with buffered writes.
10pub struct FileSink {
11    writer: BufWriter<File>,
12}
13
14impl FileSink {
15    /// Open (or create) the file at `path` for appending.
16    pub fn open(path: &Path) -> Result<Self, RuntimeError> {
17        let file = OpenOptions::new().create(true).append(true).open(path)?;
18        Ok(FileSink {
19            writer: BufWriter::new(file),
20        })
21    }
22
23    /// Serialize and append a ProcessResult to the file.
24    pub fn send(&mut self, result: &ProcessResult) -> Result<(), RuntimeError> {
25        if result.is_empty() {
26            return Ok(());
27        }
28
29        for m in result {
30            let json = serde_json::to_string(m)?;
31            writeln!(self.writer, "{json}")?;
32        }
33
34        self.writer.flush()?;
35        Ok(())
36    }
37
38    /// Write a pre-serialized JSON string directly to the file.
39    pub fn send_raw(&mut self, json: &str) -> Result<(), RuntimeError> {
40        writeln!(self.writer, "{json}")?;
41        self.writer.flush()?;
42        Ok(())
43    }
44}
rsigma_runtime/io/file.rs

rsigma_runtime/io/
file.rs
