Function sql_server

datafusion-examples/examples/flight/sql_server.rs:75–89

This example shows how to wrap DataFusion with `FlightSqlService` to support connecting to a standalone DataFusion-based server with a JDBC client, using the open source "JDBC Driver for Arrow Flight SQL". To install the JDBC driver in DBeaver for example, see these instructions: https://docs.dremio.com/software/client-applications/dbeaver/ When configuring the driver, specify property "UseEncryp

()

/// Based heavily on Ballista's implementation: https://github.com/apache/datafusion-ballista/blob/main/ballista/scheduler/src/flight_sql.rs
/// and the example in arrow-rs: https://github.com/apache/arrow-rs/blob/master/arrow-flight/examples/flight_sql_server.rs
pub async fn sql_server() -> Result<(), Box<dyn std::error::Error>> {
    env_logger::init();
    let addr = "0.0.0.0:50051".parse()?;
    let service = FlightSqlServiceImpl {
        contexts: Default::default(),
        statements: Default::default(),
        results: Default::default(),
    };
    info!("Listening on {addr:?}");
    let svc = FlightServiceServer::new(service);

    Server::builder().add_service(svc).serve(addr).await?;

    Ok(())
}

pub struct FlightSqlServiceImpl {
    contexts: Arc<DashMap<String, Arc<SessionContext>>>,