bigquery-client 0.2.0-alpha

An idiomatic library for Google BigQuery 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
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

use uuid::Uuid;
use serde::{Serialize, Deserialize};

#[derive(Serialize, Deserialize, Debug)]
#[serde(rename_all = "camelCase")]
pub(super) struct InsertAll {
    kind: String,
    skip_invalid_rows: bool,
    ignore_unknown_values: bool,
    template_suffix: Option<String>,
    rows: Vec<Row>,
    trace_id: Uuid,
}

#[derive(Serialize, Debug, Deserialize)]
#[serde(rename_all = "camelCase")]
pub(super) struct Row {
    insert_id: Uuid,
    json: serde_json::Value
}

impl Row {
    pub fn new_option(entity: &impl Serialize) -> Option<Row> {
        match serde_json::to_value(entity) {
            Ok(json) => {
                let row = Row {
                    insert_id: Uuid::new_v4(),
                    json
                };

                Some(row)
            },
            _ => None
        }
    }
}

impl InsertAll {
    pub fn new(entities: &Vec<&impl Serialize>) -> InsertAll {
        let rows = 
            entities.iter().filter_map(|entity|{
                Row::new_option(entity)
            }).collect();

        InsertAll {
            kind: "bigquery#tableDataInsertAllRequest".to_owned(),
            skip_invalid_rows: true,
            ignore_unknown_values: true,
            template_suffix: None,
            rows: rows,
            trace_id: Uuid::new_v4(),
        }
    }
}