opensky 0.2.1

Rust client for OpenSky Network Trino database
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

//! Basic query example for opensky.
//!
//! Run with: cargo run --example basic_query
//!
//! Make sure you have credentials configured in ~/.config/opensky/settings.conf

use opensky::{QueryParams, Trino};
use std::env;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Get parameters from command line or use defaults
    let args: Vec<String> = env::args().collect();

    let icao24 = args.get(1).map(|s| s.as_str()).unwrap_or("485a32");
    let start = args
        .get(2)
        .map(|s| s.as_str())
        .unwrap_or("2024-11-08 10:00:00");
    let stop = args
        .get(3)
        .map(|s| s.as_str())
        .unwrap_or("2024-11-08 12:00:00");

    println!("OpenSky-RS Basic Query Example");
    println!("================================");
    println!("ICAO24: {}", icao24);
    println!("Start:  {}", start);
    println!("Stop:   {}", stop);
    println!();

    // Create Trino client
    println!("Connecting to OpenSky Trino...");
    let mut trino = Trino::new().await?;

    // Build query parameters
    let params = QueryParams::new()
        .icao24(icao24)
        .time_range(start, stop)
        .limit(1000);

    // Show the generated SQL
    println!("\nGenerated query preview:");
    println!("{}", opensky::build_query_preview(&params));
    println!();

    // Execute query with progress
    println!("Executing query...");
    let data = trino
        .history_with_progress(params, |status| {
            println!(
                "  State: {} | Progress: {:.1}% | Rows: {}",
                status.state, status.progress, status.row_count
            );
        })
        .await?;

    println!("\nQuery complete!");
    println!("Rows returned: {}", data.len());
    println!("Columns: {:?}", data.columns());

    // Show first few rows
    if !data.is_empty() {
        println!("\nFirst 5 rows:");
        let df = data.dataframe();
        println!("{}", df.head(Some(5)));

        // Export to CSV
        let filename = format!("flight_{}.csv", icao24);
        data.to_csv(&filename)?;
        println!("\nExported to: {}", filename);
    }

    Ok(())
}