tsfile-writer 0.1.0

TsFile Writer in Rust
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

use std::fs::File;
use std::io::Write;
use std::process::Command;
use crate::{PositionedWrite, Schema, TsFileWriter, WriteWrapper};

const PATH_TO_TSFILE_TOOL: &str = "/Users/julian/Downloads/apache-iotdb-0.13.0-all-bin/tools/tsfileToolSet/print-tsfile-sketch.sh";

pub fn validate_output(filename: &str, expected_structure: &str) {
    let output = Command::new(PATH_TO_TSFILE_TOOL)
        .arg(filename)
        .output()
        .expect("Failed to execute command");

    let structure = String::from_utf8(output.stdout).unwrap();
    let real = structure.lines().filter(|line| !line.contains("[main]") && !line.contains("TsFile path") && !line.contains("file path")).map(|line| line.trim_end()).collect::<Vec<&str>>().join("\n");
    assert_eq!(expected_structure, real)
}

pub fn write_ts_file<F: FnOnce(&mut TsFileWriter<WriteWrapper<File>>) -> ()>(filename: &str, schema: Schema, test_code: F) {
    let mut writer = TsFileWriter::new(filename, schema, Default::default());

    // Execute the test
    test_code(&mut writer);

    // writer.flush();
    writer.close();
}