emotibit-data 0.1.1

Emotibit DataParser Library
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
38
39
40
41
42
43
44

//! Writer types and functions
use crate::types::Csv;
use anyhow::Result;

/// Use `WriterBuilder` to build this struct.
pub struct Writer {
    writer: csv::Writer<std::fs::File>,
}

impl Writer {
    /// Writes a `DataPacket` to a file
    pub fn write<T: Csv>(&mut self, datapacket: &T) -> Result<()> {
        for item in datapacket.csv() {
            self.writer.write_record(&item)?;
        }
        self.writer.flush()?;
        Ok(())
    }
}

/// Builder struct for `Writer`
pub struct WriterBuilder {
    builder: csv::WriterBuilder,
}

impl Default for WriterBuilder {
    fn default() -> Self {
        Self::new()
    }
}

impl WriterBuilder {
    pub fn new() -> Self {
        WriterBuilder {
            builder: csv::WriterBuilder::new(),
        }
    }
    /// Creates `Writer` with a flie path
    pub fn from_path(mut self, path: &str) -> Result<Writer> {
        Ok(Writer {
            writer: self.builder.flexible(true).from_path(path)?,
        })
    }
}