yscv-cli 0.1.7

CLI for inference, benchmarking, and evaluation
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use std::fs;
use std::io::{BufWriter, Write};
use std::path::{Path, PathBuf};

use crate::error::AppError;
use crate::util::ensure_parent_dir;

#[derive(Debug)]
pub struct JsonlEventWriter {
    path: PathBuf,
    sink: BufWriter<fs::File>,
}

impl JsonlEventWriter {
    pub fn create(path: &Path) -> Result<Self, AppError> {
        ensure_parent_dir(path)?;
        let file = fs::File::create(path)?;
        Ok(Self {
            path: path.to_path_buf(),
            sink: BufWriter::new(file),
        })
    }

    pub fn write_record(&mut self, value: &serde_json::Value) -> Result<(), AppError> {
        serde_json::to_writer(&mut self.sink, value)?;
        self.sink.write_all(b"\n")?;
        Ok(())
    }

    pub fn flush(&mut self) -> Result<(), std::io::Error> {
        self.sink.flush()
    }

    pub fn path(&self) -> &Path {
        &self.path
    }
}