(
&self,
request: Request<FlightDescriptor>,
)
| 608 | } |
| 609 | |
| 610 | async fn get_flight_info( |
| 611 | &self, |
| 612 | request: Request<FlightDescriptor>, |
| 613 | ) -> Result<Response<FlightInfo>, Status> { |
| 614 | let message = Any::decode(&*request.get_ref().cmd).map_err(decode_error_to_status)?; |
| 615 | |
| 616 | match Command::try_from(message).map_err(arrow_error_to_status)? { |
| 617 | Command::CommandStatementQuery(token) => { |
| 618 | self.get_flight_info_statement(token, request).await |
| 619 | } |
| 620 | Command::CommandPreparedStatementQuery(handle) => { |
| 621 | self.get_flight_info_prepared_statement(handle, request) |
| 622 | .await |
| 623 | } |
| 624 | Command::CommandStatementSubstraitPlan(handle) => { |
| 625 | self.get_flight_info_substrait_plan(handle, request).await |
| 626 | } |
| 627 | Command::CommandGetCatalogs(token) => { |
| 628 | self.get_flight_info_catalogs(token, request).await |
| 629 | } |
| 630 | Command::CommandGetDbSchemas(token) => { |
| 631 | return self.get_flight_info_schemas(token, request).await; |
| 632 | } |
| 633 | Command::CommandGetTables(token) => self.get_flight_info_tables(token, request).await, |
| 634 | Command::CommandGetTableTypes(token) => { |
| 635 | self.get_flight_info_table_types(token, request).await |
| 636 | } |
| 637 | Command::CommandGetSqlInfo(token) => { |
| 638 | self.get_flight_info_sql_info(token, request).await |
| 639 | } |
| 640 | Command::CommandGetPrimaryKeys(token) => { |
| 641 | self.get_flight_info_primary_keys(token, request).await |
| 642 | } |
| 643 | Command::CommandGetExportedKeys(token) => { |
| 644 | self.get_flight_info_exported_keys(token, request).await |
| 645 | } |
| 646 | Command::CommandGetImportedKeys(token) => { |
| 647 | self.get_flight_info_imported_keys(token, request).await |
| 648 | } |
| 649 | Command::CommandGetCrossReference(token) => { |
| 650 | self.get_flight_info_cross_reference(token, request).await |
| 651 | } |
| 652 | Command::CommandGetXdbcTypeInfo(token) => { |
| 653 | self.get_flight_info_xdbc_type_info(token, request).await |
| 654 | } |
| 655 | cmd => self.get_flight_info_fallback(cmd, request).await, |
| 656 | } |
| 657 | } |
| 658 | |
| 659 | async fn poll_flight_info( |
| 660 | &self, |
no test coverage detected