Skip to content

Commit

Permalink
Put feature cfg on flight imports
Browse files Browse the repository at this point in the history
  • Loading branch information
gruuya committed Dec 8, 2023
1 parent cfec21e commit 60e42d9
Show file tree
Hide file tree
Showing 4 changed files with 4 additions and 3 deletions.
2 changes: 1 addition & 1 deletion src/frontend/flight/sql.rs
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,7 @@ impl FlightSqlService for SeafowlFlightHandler {

let flight_info = FlightInfo::new()
.try_with_schema(&schema)
.expect("encoding schema")
.map_err(|e| Status::internal(e.to_string()))?
.with_endpoint(endpoint)
.with_descriptor(request.into_inner());

Expand Down
1 change: 1 addition & 0 deletions src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@ use clap::Parser;
use futures::{future::join_all, Future, FutureExt};

use pretty_env_logger::env_logger;
#[cfg(feature = "frontend-arrow-flight")]
use seafowl::frontend::flight::run_flight_server;
use seafowl::{
cli,
Expand Down
3 changes: 1 addition & 2 deletions tests/flight/client.rs
Original file line number Diff line number Diff line change
Expand Up @@ -35,9 +35,8 @@ async fn test_basic_queries() -> Result<()> {
.clone()
.expect("expected ticket");

// Retrieve the corresponding Flight stream with do_get
// Retrieve the corresponding Flight stream and collect into batches
let flight_stream = client.do_get(ticket).await.expect("error fetching data");

let results: Vec<RecordBatch> = flight_stream.try_collect().await?;

let expected = [
Expand Down
1 change: 1 addition & 0 deletions tests/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
// https://endler.dev/2020/rust-compile-times/#combine-all-integration-tests-in-a-single-binary

mod cli;
#[cfg(feature = "frontend-arrow-flight")]
mod flight;
mod http;
mod statements;

0 comments on commit 60e42d9

Please sign in to comment.