Pass metadata to impl

apache · Aug 4, 2022 · 4e28be7 · 4e28be7
1 parent e835853
commit 4e28be7
Show file tree

Hide file tree

Showing 2 changed files with 195 additions and 190 deletions.
diff --git a/arrow-flight/examples/flight_sql_server.rs b/arrow-flight/examples/flight_sql_server.rs
@@ -19,6 +19,7 @@ use arrow_flight::sql::{ActionCreatePreparedStatementResult, SqlInfo};
 use arrow_flight::{FlightData, HandshakeRequest, HandshakeResponse};
 use futures::Stream;
 use std::pin::Pin;
+use tonic::metadata::MetadataMap;
 use tonic::transport::Server;
 use tonic::{Request, Response, Status, Streaming};
 
@@ -94,178 +95,205 @@ impl FlightSqlService for FlightSqlServiceImpl {
         &self,
         _query: CommandStatementQuery,
         _request: FlightDescriptor,
+        _metadata: MetadataMap,
     ) -> Result<Response<FlightInfo>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn get_flight_info_prepared_statement(
         &self,
         _query: CommandPreparedStatementQuery,
         _request: FlightDescriptor,
+        _metadata: MetadataMap,
     ) -> Result<Response<FlightInfo>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn get_flight_info_catalogs(
         &self,
         _query: CommandGetCatalogs,
         _request: FlightDescriptor,
+        _metadata: MetadataMap,
     ) -> Result<Response<FlightInfo>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn get_flight_info_schemas(
         &self,
         _query: CommandGetDbSchemas,
         _request: FlightDescriptor,
+        _metadata: MetadataMap,
     ) -> Result<Response<FlightInfo>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn get_flight_info_tables(
         &self,
         _query: CommandGetTables,
         _request: FlightDescriptor,
+        _metadata: MetadataMap,
     ) -> Result<Response<FlightInfo>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn get_flight_info_table_types(
         &self,
         _query: CommandGetTableTypes,
         _request: FlightDescriptor,
+        _metadata: MetadataMap,
     ) -> Result<Response<FlightInfo>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn get_flight_info_sql_info(
         &self,
         _query: CommandGetSqlInfo,
         _request: FlightDescriptor,
+        _metadata: MetadataMap,
     ) -> Result<Response<FlightInfo>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn get_flight_info_primary_keys(
         &self,
         _query: CommandGetPrimaryKeys,
         _request: FlightDescriptor,
+        _metadata: MetadataMap,
     ) -> Result<Response<FlightInfo>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn get_flight_info_exported_keys(
         &self,
         _query: CommandGetExportedKeys,
         _request: FlightDescriptor,
+        _metadata: MetadataMap,
     ) -> Result<Response<FlightInfo>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn get_flight_info_imported_keys(
         &self,
         _query: CommandGetImportedKeys,
         _request: FlightDescriptor,
+        _metadata: MetadataMap,
     ) -> Result<Response<FlightInfo>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn get_flight_info_cross_reference(
         &self,
         _query: CommandGetCrossReference,
         _request: FlightDescriptor,
+        _metadata: MetadataMap,
     ) -> Result<Response<FlightInfo>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     // do_get
     async fn do_get_statement(
         &self,
         _ticket: TicketStatementQuery,
+        _metadata: MetadataMap,
     ) -> Result<Response<<Self as FlightService>::DoGetStream>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
 
     async fn do_get_prepared_statement(
         &self,
         _query: CommandPreparedStatementQuery,
+        _metadata: MetadataMap,
     ) -> Result<Response<<Self as FlightService>::DoGetStream>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn do_get_catalogs(
         &self,
         _query: CommandGetCatalogs,
+        _metadata: MetadataMap,
     ) -> Result<Response<<Self as FlightService>::DoGetStream>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn do_get_schemas(
         &self,
         _query: CommandGetDbSchemas,
+        _metadata: MetadataMap,
     ) -> Result<Response<<Self as FlightService>::DoGetStream>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn do_get_tables(
         &self,
         _query: CommandGetTables,
+        _metadata: MetadataMap,
     ) -> Result<Response<<Self as FlightService>::DoGetStream>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn do_get_table_types(
         &self,
         _query: CommandGetTableTypes,
+        _metadata: MetadataMap,
     ) -> Result<Response<<Self as FlightService>::DoGetStream>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn do_get_sql_info(
         &self,
         _query: CommandGetSqlInfo,
+        _metadata: MetadataMap,
     ) -> Result<Response<<Self as FlightService>::DoGetStream>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn do_get_primary_keys(
         &self,
         _query: CommandGetPrimaryKeys,
+        _metadata: MetadataMap,
     ) -> Result<Response<<Self as FlightService>::DoGetStream>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn do_get_exported_keys(
         &self,
         _query: CommandGetExportedKeys,
+        _metadata: MetadataMap,
     ) -> Result<Response<<Self as FlightService>::DoGetStream>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn do_get_imported_keys(
         &self,
         _query: CommandGetImportedKeys,
+        _metadata: MetadataMap,
     ) -> Result<Response<<Self as FlightService>::DoGetStream>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn do_get_cross_reference(
         &self,
         _query: CommandGetCrossReference,
+        _metadata: MetadataMap,
     ) -> Result<Response<<Self as FlightService>::DoGetStream>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     // do_put
     async fn do_put_statement_update(
         &self,
         _ticket: CommandStatementUpdate,
+        _metadata: MetadataMap,
     ) -> Result<i64, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn do_put_prepared_statement_query(
         &self,
         _query: CommandPreparedStatementQuery,
         _request: Streaming<FlightData>,
+        _metadata: MetadataMap,
     ) -> Result<Response<<Self as FlightService>::DoPutStream>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn do_put_prepared_statement_update(
         &self,
         _query: CommandPreparedStatementUpdate,
         _request: Streaming<FlightData>,
+        _metadata: MetadataMap,
     ) -> Result<i64, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     // do_action
     async fn do_action_create_prepared_statement(
         &self,
         _query: ActionCreatePreparedStatementRequest,
+        _metadata: MetadataMap,
     ) -> Result<ActionCreatePreparedStatementResult, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn do_action_close_prepared_statement(
         &self,
         _query: ActionClosePreparedStatementRequest,
+        _metadata: MetadataMap,
     ) {
         unimplemented!("Not yet implemented")
     }