redshift 1.0.3

Rust library for parsing redshift files generated by the UNLOAD command.
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

#[cfg(test)]
mod test {

    extern crate redshift;

    use std::str;

    struct TestItem<'a> {
        pub A: &'a str,
        pub B: &'a str,
        pub C: &'a str,
    }

    #[test]
    fn basic_writer_test() {

        // Arrange
        let test_column_definitions = vec![
            redshift::writer::ColumnDefinition::<TestItem> {
                name:  "Acolumn",
                extract_column: Box::new(move |i: &TestItem| i.A.to_string().clone()),
            },
            redshift::writer::ColumnDefinition::<TestItem> {

                name: "Bcolumn",
                extract_column: Box::new(move |i: &TestItem| i.B.to_string().clone()),
            },
            redshift::writer::ColumnDefinition::<TestItem> {
                name: "Ccolumn",
                extract_column: Box::new(move |i: &TestItem| i.C.to_string().clone()),
            },
        ];
        let items = vec![
            TestItem { A: "a1", B: "b1", C: "c1" },
            TestItem { A: "a2", B: "b2", C: "c2" },
            TestItem { A: "a3", B: "b3", C: "c3" },
        ];

        // Act
        let mut byte_vec: Vec<u8> = Vec::new();
        let mut writer = redshift::writer::Writer::new();
        let res = writer.write_columns(&mut byte_vec, test_column_definitions, items);

        // Assert
        assert!(res.is_ok());
        let result_str = str::from_utf8(&byte_vec).unwrap();
        assert_eq!("\"a1\"|\"b1\"|\"c1\"\n\"a2\"|\"b2\"|\"c2\"\n\"a3\"|\"b3\"|\"c3\"\n", result_str);
    }

    #[test]
    fn basic_writer_with_header() {

        // Arrange
        let test_column_definitions = vec![
            redshift::writer::ColumnDefinition::<TestItem> {
                name:  "Acolumn",
                extract_column: Box::new(move |i: &TestItem| i.A.to_string().clone()),
            },
            redshift::writer::ColumnDefinition::<TestItem> {

                name: "Bcolumn",
                extract_column: Box::new(move |i: &TestItem| i.B.to_string().clone()),
            },
            redshift::writer::ColumnDefinition::<TestItem> {
                name: "Ccolumn",
                extract_column: Box::new(move |i: &TestItem| i.C.to_string().clone()),
            },
        ];
        let items = vec![
            TestItem { A: "a1", B: "b1", C: "c1" },
            TestItem { A: "a2", B: "b2", C: "c2" },
            TestItem { A: "a3", B: "b3", C: "c3" },
        ];

        // Act
        let mut byte_vec: Vec<u8> = Vec::new();
        let mut writer = redshift::writer::Writer::new();
        let res = writer.with_header().write_columns(&mut byte_vec, test_column_definitions, items);

        // Assert
        assert!(res.is_ok());
        let result_str = str::from_utf8(&byte_vec).unwrap();
        assert_eq!("\"Acolumn\"|\"Bcolumn\"|\"Ccolumn\"\n\"a1\"|\"b1\"|\"c1\"\n\"a2\"|\"b2\"|\"c2\"\n\"a3\"|\"b3\"|\"c3\"\n", result_str);
    }
}