typed-arrow 0.7.0

Compile-time Arrow schemas for 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
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

use arrow_array::Array;
use typed_arrow::prelude::*;

#[derive(Record)]
pub struct PersonO {
    pub id: i64,
    pub name: Option<String>,
}

#[test]
fn build_from_option_rows_flat() {
    let rows: Vec<Option<PersonO>> = vec![
        Some(PersonO {
            id: 1,
            name: Some("a".into()),
        }),
        None,
        Some(PersonO { id: 3, name: None }),
    ];
    let mut b = <PersonO as BuildRows>::new_builders(rows.len());
    b.append_option_rows(rows);
    let arrays = b.finish();

    assert_eq!(arrays.id.len(), 3);
    assert!(arrays.id.is_null(1));
    assert_eq!(arrays.id.value(0), 1);
}

#[derive(Record)]
pub struct AddressO {
    pub city: String,
}

#[derive(Record)]
pub struct PersonNO {
    pub id: i64,
    pub address: Option<AddressO>,
}

#[test]
fn build_from_option_rows_nested() {
    let rows: Vec<Option<PersonNO>> = vec![
        Some(PersonNO {
            id: 1,
            address: Some(AddressO { city: "NYC".into() }),
        }),
        None,
        Some(PersonNO {
            id: 3,
            address: None,
        }),
    ];
    let mut b = <PersonNO as BuildRows>::new_builders(rows.len());
    b.append_option_rows(rows);
    let arrays = b.finish();
    assert_eq!(arrays.id.len(), 3);
    assert!(arrays.id.is_null(1));
    assert!(arrays.address.is_null(1));
}