ARROW-8308: [Rust] Implement DoExchange on examples
Closes #6800 from nevi-me/ARROW-8308
Authored-by: Neville Dipale <nevilledips@gmail.com>
Signed-off-by: Benjamin Kietzman <bengilgit@gmail.com>
diff --git a/rust/arrow-flight/examples/server.rs b/rust/arrow-flight/examples/server.rs
index b71ad78..6651e39 100644
--- a/rust/arrow-flight/examples/server.rs
+++ b/rust/arrow-flight/examples/server.rs
@@ -46,6 +46,8 @@
>;
type ListActionsStream =
Pin<Box<dyn Stream<Item = Result<ActionType, Status>> + Send + Sync + 'static>>;
+ type DoExchangeStream =
+ Pin<Box<dyn Stream<Item = Result<FlightData, Status>> + Send + Sync + 'static>>;
async fn handshake(
&self,
@@ -102,6 +104,13 @@
) -> Result<Response<Self::ListActionsStream>, Status> {
Err(Status::unimplemented("Not yet implemented"))
}
+
+ async fn do_exchange(
+ &self,
+ _request: Request<Streaming<FlightData>>,
+ ) -> Result<Response<Self::DoExchangeStream>, Status> {
+ Err(Status::unimplemented("Not yet implemented"))
+ }
}
#[tokio::main]
diff --git a/rust/datafusion/examples/flight_server.rs b/rust/datafusion/examples/flight_server.rs
index 30afa5d..b4592ba 100644
--- a/rust/datafusion/examples/flight_server.rs
+++ b/rust/datafusion/examples/flight_server.rs
@@ -54,6 +54,8 @@
>;
type ListActionsStream =
Pin<Box<dyn Stream<Item = Result<ActionType, Status>> + Send + Sync + 'static>>;
+ type DoExchangeStream =
+ Pin<Box<dyn Stream<Item = Result<FlightData, Status>> + Send + Sync + 'static>>;
async fn get_schema(
&self,
@@ -163,6 +165,13 @@
) -> Result<Response<Self::ListActionsStream>, Status> {
Err(Status::unimplemented("Not yet implemented"))
}
+
+ async fn do_exchange(
+ &self,
+ _request: Request<Streaming<FlightData>>,
+ ) -> Result<Response<Self::DoExchangeStream>, Status> {
+ Err(Status::unimplemented("Not yet implemented"))
+ }
}
fn to_tonic_err(e: &datafusion::error::ExecutionError) -> Status {