serde_arrow 0.14.0

Convert sequences of Rust objects to Arrow arrays and back again
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

//! Common definitions for doc tests
use serde::{Serialize, Deserialize};

#[derive(Clone, Deserialize, Serialize)]
pub struct Record {
    pub a: Option<f32>,
    pub b: u64,
}

pub const fn example_records() -> &'static [Record] {
    &[Record { a: Some(1.0), b: 2}]
}

#[cfg(has_arrow)]
pub fn example_record_batch() -> crate::_impl::arrow::array::RecordBatch {
    use crate::schema::{SchemaLike, TracingOptions};

    let items = example_records();
    
    let fields = Vec::<crate::_impl::arrow::datatypes::FieldRef>::from_type::<Record>(TracingOptions::default()).unwrap();
    crate::to_record_batch(&fields, &items).unwrap()
}

#[cfg(has_arrow)]
pub fn example_arrow_arrays() -> (Vec<crate::_impl::arrow::datatypes::FieldRef>, Vec<crate::_impl::arrow::array::ArrayRef>) {
    use crate::schema::{SchemaLike, TracingOptions};

    let items = example_records();

    let fields = Vec::<crate::_impl::arrow::datatypes::FieldRef>::from_type::<Record>(TracingOptions::default()).unwrap();
    let arrays = crate::to_arrow(&fields, items).unwrap();

    (fields, arrays)
}

#[cfg(has_arrow2)]
pub fn example_arrow2_arrays() -> (Vec<crate::_impl::arrow2::datatypes::Field>, Vec<Box<dyn crate::_impl::arrow2::array::Array>>) {
    use crate::schema::{SchemaLike, TracingOptions};

    let items = example_records();

    let fields = Vec::<crate::_impl::arrow2::datatypes::Field>::from_type::<Record>(TracingOptions::default()).unwrap();
    let arrays = crate::to_arrow2(&fields, items).unwrap();

    (fields, arrays)
}