razel 0.5.7

a command executor with caching for data processing pipelines
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
45
46
47
48
49

use crate::executors::ExecutionResult;
use crate::metadata::Tag;
use crate::Command;
use anyhow::Result;
use serde::Serialize;
use std::fs;
use std::path::PathBuf;

pub struct Profile {
    execution_times: Vec<ExecutionTimesItem>,
}

#[derive(Serialize)]
struct ExecutionTimesItem {
    name: String,
    #[serde(skip_serializing_if = "Vec::is_empty")]
    tags: Vec<Tag>,
    time: f32,
}

impl Profile {
    pub fn new() -> Self {
        Self {
            execution_times: vec![],
        }
    }

    pub fn collect(&mut self, command: &Command, execution_result: &ExecutionResult) {
        if let Some(duration) = &execution_result.exec_duration {
            self.execution_times.push(ExecutionTimesItem {
                name: command.name.clone(),
                tags: command.tags.clone(),
                time: duration.as_secs_f32(),
            })
        }
    }

    pub fn write_json(&self, path: &PathBuf) -> Result<()> {
        let vec = serde_json::to_vec(&self.execution_times).unwrap();
        fs::write(path, vec)?;
        Ok(())
    }
}

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