datum-sql 0.10.3

DataFusion and Arrow SQL front end for Datum streams
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

use std::sync::Arc;

use arrow::array::{Int64Array, StringArray};
use arrow::datatypes::{DataType, Field, Schema};
use arrow::record_batch::RecordBatch;
use datum::Source;
use datum_sql::DatumSqlContext;

#[tokio::main]
async fn main() -> datafusion::common::Result<()> {
    let schema = Arc::new(Schema::new(vec![
        Field::new("city", DataType::Utf8, false),
        Field::new("temp", DataType::Int64, false),
        Field::new("humidity", DataType::Int64, false),
    ]));
    let batch = RecordBatch::try_new(
        Arc::clone(&schema),
        vec![
            Arc::new(StringArray::from(vec!["sf", "nyc", "sea", "aus"])),
            Arc::new(Int64Array::from(vec![67, 74, 58, 91])),
            Arc::new(Int64Array::from(vec![72, 64, 80, 40])),
        ],
    )?;

    let context = DatumSqlContext::new();
    context.register_source("weather", schema, Source::from_iter([batch]))?;

    let batches = context
        .execute("SELECT city, temp FROM weather WHERE temp >= 70")
        .await?;
    for batch in batches {
        println!("{batch:?}");
    }

    Ok(())
}