ARROW-8308: [Rust] Implement DoExchange on examples

nevi-me · bkietz · commit e381a72703c1 · 2020-04-01T18:12:17.000-04:00
Closes apache#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
@@ -46,6 +46,8 @@ impl FlightService for FlightServiceImpl {
     >;
     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 @@ impl FlightService for FlightServiceImpl {
     ) -> 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
@@ -54,6 +54,8 @@ impl FlightService for FlightServiceImpl {
     >;
     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 @@ impl FlightService for FlightServiceImpl {
     ) -> 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 {
