infrastore-cli 0.1.0

Command-line tool for loading and inspecting an infrastore store directly on disk
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

//! The `add` command: load one or more series from a descriptor JSON + CSV.

use std::path::Path;

use infrastore_core::Compression;

use crate::{color, descriptor, store_access};

pub fn run(
    store_path: &Path,
    descriptor_path: &Path,
    csv_override: Option<&Path>,
    compression: Option<Compression>,
) -> Result<(), String> {
    let descriptors = descriptor::load(descriptor_path)?;
    if csv_override.is_some() && descriptors.len() > 1 {
        return Err("--csv cannot be used with an array descriptor".to_string());
    }
    let base_dir = descriptor_path.parent();

    let mut requests = Vec::with_capacity(descriptors.len());
    for desc in &descriptors {
        requests.push(desc.to_add_request(base_dir, csv_override)?);
    }

    let mut store = store_access::open_writable_with(store_path, compression)?;
    let keys = store
        .add_time_series_bulk(requests)
        .map_err(|e| e.to_string())?;
    store.flush().map_err(|e| e.to_string())?;

    for k in &keys {
        println!(
            "added {} '{}' (owner {})",
            k.time_series_type().as_str(),
            k.name(),
            k.owner_id()
        );
    }
    println!(
        "{}",
        color::header(&format!(
            "Added {} time series to {}.",
            keys.len(),
            store_path.display()
        ))
    );
    Ok(())
}