Fix broken FlightSQL example (#3387)

* Fix flight sql example * Add into CI
apache · Dec 22, 2022 · af07998 · af07998
1 parent 98bba9c
commit af07998
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/.github/workflows/arrow_flight.yml b/.github/workflows/arrow_flight.yml
@@ -56,6 +56,9 @@ jobs:
       - name: Test --all-features
         run: |
           cargo test -p arrow-flight --all-features
+      - name: Test --examples
+        run: |
+          cargo test -p arrow-flight  --features=flight-sql-experimental --examples
       - name: Verify workspace clean
         run: git diff --exit-code
 

diff --git a/arrow-flight/examples/flight_sql_server.rs b/arrow-flight/examples/flight_sql_server.rs
@@ -482,8 +482,14 @@ impl ProstMessageExt for FetchResults {
 mod tests {
     use super::*;
     use futures::TryStreamExt;
+    use std::fs;
+    use tempfile::NamedTempFile;
+    use tokio::net::{UnixListener, UnixStream};
+    use tokio_stream::wrappers::UnixListenerStream;
+    use tonic::transport::Endpoint;
 
     use arrow::util::pretty::pretty_format_batches;
+    use arrow_flight::sql::client::FlightSqlServiceClient;
     use arrow_flight::utils::flight_data_to_batches;
     use tower::service_fn;