Fix examples and conflicts

Author: gabotechs

examples/localhost_worker.rs

@@ -33,7 +33,7 @@ async fn main() -> Result<(), Box<dyn Error>> {
         cached: DashMap::new(),
     };
 
-    let endpoint = ArrowFlightEndpoint::new(move |ctx: DistributedSessionBuilderContext| {
+    let endpoint = ArrowFlightEndpoint::try_new(move |ctx: DistributedSessionBuilderContext| {
         let local_host_resolver = localhost_resolver.clone();
         async move {
             Ok(SessionStateBuilder::new()
@@ -42,7 +42,7 @@ async fn main() -> Result<(), Box<dyn Error>> {
                 .with_default_features()
                 .build())
         }
-    });
+    })?;
 
     Server::builder()
         .add_service(FlightServiceServer::new(endpoint))

src/flight_service/do_get.rs

@@ -181,7 +181,7 @@ mod tests {
 
         // Create ArrowFlightEndpoint with DefaultSessionBuilder
         let endpoint =
-            ArrowFlightEndpoint::new(DefaultSessionBuilder).expect("Failed to create endpoint");
+            ArrowFlightEndpoint::try_new(DefaultSessionBuilder).expect("Failed to create endpoint");
 
         // Create 3 tasks with 3 partitions each.
         let num_tasks = 3;
@@ -210,7 +210,7 @@ mod tests {
             tasks,
         };
 
-        let task_keys = vec![
+        let task_keys = [
             StageKey {
                 query_id: query_id_uuid.to_string(),
                 stage_id,

src/flight_service/service.rs

@@ -36,7 +36,7 @@ pub struct ArrowFlightEndpoint {
 }
 
 impl ArrowFlightEndpoint {
-    pub fn new(
+    pub fn try_new(
         session_builder: impl DistributedSessionBuilder + Send + Sync + 'static,
     ) -> Result<Self, DataFusionError> {
         let ttl_map = TTLMap::try_new(TTLMapConfig::default())?;

src/test_utils/localhost.rs

@@ -109,7 +109,7 @@ pub async fn spawn_flight_service(
     session_builder: impl DistributedSessionBuilder + Send + Sync + 'static,
     incoming: TcpListener,
 ) -> Result<(), Box<dyn Error + Send + Sync>> {
-    let endpoint = ArrowFlightEndpoint::new(session_builder)?;
+    let endpoint = ArrowFlightEndpoint::try_new(session_builder)?;
 
     let incoming = tokio_stream::wrappers::TcpListenerStream::new(incoming);