Able to get session context, but JDBC driver hung

arrow-flight/Cargo.toml

@@ -27,7 +27,7 @@ repository = "https://github.com/apache/arrow-rs"
 license = "Apache-2.0"
 
 [dependencies]
-arrow = { path = "../arrow", version = "19.0.0", default-features = false, features = ["ipc"] }
+arrow = { path = "../arrow", features = ["ipc"], optional = false, default-features = false }
 base64 = { version = "0.13", default-features = false }
 tonic = { version = "0.8", default-features = false, features = ["transport", "codegen", "prost"] }
 bytes = { version = "1", default-features = false }

arrow-flight/examples/flight_sql_server.rs

@@ -94,13 +94,15 @@ impl FlightSqlService for FlightSqlServiceImpl {
         &self,
         _query: CommandStatementQuery,
         _request: FlightDescriptor,
+        _auth: &Option<String>,
     ) -> Result<Response<FlightInfo>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
     async fn get_flight_info_prepared_statement(
         &self,
         _query: CommandPreparedStatementQuery,
         _request: FlightDescriptor,
+        _auth: &Option<String>
     ) -> Result<Response<FlightInfo>, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }
@@ -260,6 +262,7 @@ impl FlightSqlService for FlightSqlServiceImpl {
     async fn do_action_create_prepared_statement(
         &self,
         _query: ActionCreatePreparedStatementRequest,
+        auth: &Option<String>,
     ) -> Result<ActionCreatePreparedStatementResult, Status> {
         Err(Status::unimplemented("Not yet implemented"))
     }

arrow-flight/src/arrow.flight.protocol.rs

@@ -1,31 +1,31 @@
 // This file was automatically generated through the build.rs script, and should not be edited.
 
 ///
-///  The request that a client provides to a server on handshake.
+/// The request that a client provides to a server on handshake.
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct HandshakeRequest {
     ///
-    ///  A defined protocol version
+    /// A defined protocol version
     #[prost(uint64, tag="1")]
     pub protocol_version: u64,
     ///
-    ///  Arbitrary auth/handshake info.
+    /// Arbitrary auth/handshake info.
     #[prost(bytes="vec", tag="2")]
     pub payload: ::prost::alloc::vec::Vec<u8>,
 }
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct HandshakeResponse {
     ///
-    ///  A defined protocol version
+    /// A defined protocol version
     #[prost(uint64, tag="1")]
     pub protocol_version: u64,
     ///
-    ///  Arbitrary auth/handshake info.
+    /// Arbitrary auth/handshake info.
     #[prost(bytes="vec", tag="2")]
     pub payload: ::prost::alloc::vec::Vec<u8>,
 }
 ///
-///  A message for doing simple auth.
+/// A message for doing simple auth.
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct BasicAuth {
     #[prost(string, tag="2")]
@@ -37,8 +37,8 @@ pub struct BasicAuth {
 pub struct Empty {
 }
 ///
-///  Describes an available action, including both the name used for execution
-///  along with a short description of the purpose of the action.
+/// Describes an available action, including both the name used for execution
+/// along with a short description of the purpose of the action.
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct ActionType {
     #[prost(string, tag="1")]
@@ -47,15 +47,15 @@ pub struct ActionType {
     pub description: ::prost::alloc::string::String,
 }
 ///
-///  A service specific expression that can be used to return a limited set
-///  of available Arrow Flight streams.
+/// A service specific expression that can be used to return a limited set
+/// of available Arrow Flight streams.
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct Criteria {
     #[prost(bytes="vec", tag="1")]
     pub expression: ::prost::alloc::vec::Vec<u8>,
 }
 ///
-///  An opaque action specific for the service.
+/// An opaque action specific for the service.
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct Action {
     #[prost(string, tag="1")]
@@ -64,54 +64,54 @@ pub struct Action {
     pub body: ::prost::alloc::vec::Vec<u8>,
 }
 ///
-///  An opaque result returned after executing an action.
+/// An opaque result returned after executing an action.
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct Result {
     #[prost(bytes="vec", tag="1")]
     pub body: ::prost::alloc::vec::Vec<u8>,
 }
 ///
-///  Wrap the result of a getSchema call
+/// Wrap the result of a getSchema call
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct SchemaResult {
-    ///  schema of the dataset as described in Schema.fbs::Schema.
+    /// schema of the dataset as described in Schema.fbs::Schema.
     #[prost(bytes="vec", tag="1")]
     pub schema: ::prost::alloc::vec::Vec<u8>,
 }
 ///
-///  The name or tag for a Flight. May be used as a way to retrieve or generate
-///  a flight or be used to expose a set of previously defined flights.
+/// The name or tag for a Flight. May be used as a way to retrieve or generate
+/// a flight or be used to expose a set of previously defined flights.
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct FlightDescriptor {
     #[prost(enumeration="flight_descriptor::DescriptorType", tag="1")]
     pub r#type: i32,
     ///
-    ///  Opaque value used to express a command. Should only be defined when
-    ///  type = CMD.
+    /// Opaque value used to express a command. Should only be defined when
+    /// type = CMD.
     #[prost(bytes="vec", tag="2")]
     pub cmd: ::prost::alloc::vec::Vec<u8>,
     ///
-    ///  List of strings identifying a particular dataset. Should only be defined
-    ///  when type = PATH.
+    /// List of strings identifying a particular dataset. Should only be defined
+    /// when type = PATH.
     #[prost(string, repeated, tag="3")]
     pub path: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
 }
 /// Nested message and enum types in `FlightDescriptor`.
 pub mod flight_descriptor {
     ///
-    ///  Describes what type of descriptor is defined.
+    /// Describes what type of descriptor is defined.
     #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, PartialOrd, Ord, ::prost::Enumeration)]
     #[repr(i32)]
     pub enum DescriptorType {
-        ///  Protobuf pattern, not used.
+        /// Protobuf pattern, not used.
         Unknown = 0,
         ///
-        ///  A named path that identifies a dataset. A path is composed of a string
-        ///  or list of strings describing a particular dataset. This is conceptually
+        /// A named path that identifies a dataset. A path is composed of a string
+        /// or list of strings describing a particular dataset. This is conceptually
         ///   similar to a path inside a filesystem.
         Path = 1,
         ///
-        ///  An opaque command to generate a dataset.
+        /// An opaque command to generate a dataset.
         Cmd = 2,
     }
     impl DescriptorType {
@@ -129,86 +129,86 @@ pub mod flight_descriptor {
     }
 }
 ///
-///  The access coordinates for retrieval of a dataset. With a FlightInfo, a
-///  consumer is able to determine how to retrieve a dataset.
+/// The access coordinates for retrieval of a dataset. With a FlightInfo, a
+/// consumer is able to determine how to retrieve a dataset.
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct FlightInfo {
-    ///  schema of the dataset as described in Schema.fbs::Schema.
+    /// schema of the dataset as described in Schema.fbs::Schema.
     #[prost(bytes="vec", tag="1")]
     pub schema: ::prost::alloc::vec::Vec<u8>,
     ///
-    ///  The descriptor associated with this info.
+    /// The descriptor associated with this info.
     #[prost(message, optional, tag="2")]
     pub flight_descriptor: ::core::option::Option<FlightDescriptor>,
     ///
-    ///  A list of endpoints associated with the flight. To consume the whole
-    ///  flight, all endpoints must be consumed.
+    /// A list of endpoints associated with the flight. To consume the whole
+    /// flight, all endpoints must be consumed.
     #[prost(message, repeated, tag="3")]
     pub endpoint: ::prost::alloc::vec::Vec<FlightEndpoint>,
-    ///  Set these to -1 if unknown.
+    /// Set these to -1 if unknown.
     #[prost(int64, tag="4")]
     pub total_records: i64,
     #[prost(int64, tag="5")]
     pub total_bytes: i64,
 }
 ///
-///  A particular stream or split associated with a flight.
+/// A particular stream or split associated with a flight.
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct FlightEndpoint {
     ///
-    ///  Token used to retrieve this stream.
+    /// Token used to retrieve this stream.
     #[prost(message, optional, tag="1")]
     pub ticket: ::core::option::Option<Ticket>,
     ///
-    ///  A list of URIs where this ticket can be redeemed. If the list is
-    ///  empty, the expectation is that the ticket can only be redeemed on the
-    ///  current service where the ticket was generated.
+    /// A list of URIs where this ticket can be redeemed. If the list is
+    /// empty, the expectation is that the ticket can only be redeemed on the
+    /// current service where the ticket was generated.
     #[prost(message, repeated, tag="2")]
     pub location: ::prost::alloc::vec::Vec<Location>,
 }
 ///
-///  A location where a Flight service will accept retrieval of a particular
-///  stream given a ticket.
+/// A location where a Flight service will accept retrieval of a particular
+/// stream given a ticket.
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct Location {
     #[prost(string, tag="1")]
     pub uri: ::prost::alloc::string::String,
 }
 ///
-///  An opaque identifier that the service can use to retrieve a particular
-///  portion of a stream.
+/// An opaque identifier that the service can use to retrieve a particular
+/// portion of a stream.
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct Ticket {
     #[prost(bytes="vec", tag="1")]
     pub ticket: ::prost::alloc::vec::Vec<u8>,
 }
 ///
-///  A batch of Arrow data as part of a stream of batches.
+/// A batch of Arrow data as part of a stream of batches.
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct FlightData {
     ///
-    ///  The descriptor of the data. This is only relevant when a client is
-    ///  starting a new DoPut stream.
+    /// The descriptor of the data. This is only relevant when a client is
+    /// starting a new DoPut stream.
     #[prost(message, optional, tag="1")]
     pub flight_descriptor: ::core::option::Option<FlightDescriptor>,
     ///
-    ///  Header for message data as described in Message.fbs::Message.
+    /// Header for message data as described in Message.fbs::Message.
     #[prost(bytes="vec", tag="2")]
     pub data_header: ::prost::alloc::vec::Vec<u8>,
     ///
-    ///  Application-defined metadata.
+    /// Application-defined metadata.
     #[prost(bytes="vec", tag="3")]
     pub app_metadata: ::prost::alloc::vec::Vec<u8>,
     ///
-    ///  The actual batch of Arrow data. Preferably handled with minimal-copies
-    ///  coming last in the definition to help with sidecar patterns (it is
-    ///  expected that some implementations will fetch this field off the wire
-    ///  with specialized code to avoid extra memory copies).
+    /// The actual batch of Arrow data. Preferably handled with minimal-copies
+    /// coming last in the definition to help with sidecar patterns (it is
+    /// expected that some implementations will fetch this field off the wire
+    /// with specialized code to avoid extra memory copies).
     #[prost(bytes="vec", tag="1000")]
     pub data_body: ::prost::alloc::vec::Vec<u8>,
 }
 /// *
-///  The response message associated with the submission of a DoPut.
+/// The response message associated with the submission of a DoPut.
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct PutResult {
     #[prost(bytes="vec", tag="1")]