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

use arrow_array::{Array, cast::as_string_array};
use typed_arrow::prelude::*;

#[derive(Record)]
pub struct AddressN {
    pub city: String,
    pub zip: Option<i32>,
}

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

#[test]
fn build_rows_with_nested_struct() {
    let rows = vec![
        PersonN {
            id: 1,
            address: Some(AddressN {
                city: "NYC".into(),
                zip: None,
            }),
        },
        PersonN {
            id: 2,
            address: None,
        },
        PersonN {
            id: 3,
            address: Some(AddressN {
                city: "SF".into(),
                zip: Some(94111),
            }),
        },
    ];

    let mut b = <PersonN as BuildRows>::new_builders(rows.len());
    b.append_rows(rows);
    let arrays = b.finish();

    // Validate nested struct array
    let addr = arrays.address;
    assert_eq!(addr.len(), 3);
    assert!(addr.is_null(1));
    let city = as_string_array(addr.column(0));
    assert_eq!(city.value(0), "NYC");
    assert_eq!(city.value(2), "SF");
}