avro-test.rs

use std::env;
  
use datafusion::arrow::util::pretty;

use datafusion::error::Result;
use datafusion::prelude::*;

/// This example demonstrates executing a simple query against an Arrow data source (Avro) and
/// fetching results
#[tokio::main]
async fn main() -> Result<()> {

    let args: Vec<String> = env::args().collect();
    let avro_file = &args[1];

    // create local execution context
    let mut ctx = ExecutionContext::new();

    // register avro file with the execution context
    let df = ctx.read_avro(avro_file, AvroReadOptions::default()).await?;

    let results = df.collect().await?;

    // print the results
    pretty::print_batches(&results)?;

    Ok(())
}