Run cargo fmt

Author: gabotechs

src/flight_service/mod.rs

@@ -1,11 +1,11 @@
 mod do_get;
 mod do_put;
 mod service;
-mod stream_partitioner_registry;
 mod session_builder;
+mod stream_partitioner_registry;
 
 pub(crate) use do_get::DoGet;
 pub(crate) use do_put::DoPut;
 
 pub use service::ArrowFlightEndpoint;
-pub use session_builder::SessionBuilder;
+pub use session_builder::SessionBuilder;

src/stage_delegation/context.rs

@@ -21,5 +21,5 @@ pub struct ActorContext {
     #[prost(uint64, tag = "1")]
     pub caller_actor_idx: u64,
     #[prost(uint64, tag = "2")]
-    pub actor_idx: u64
-}
+    pub actor_idx: u64,
+}

src/stage_delegation/delegation.rs

@@ -238,13 +238,9 @@ mod tests {
         let id1 = stage_id1.clone();
         let id2 = stage_id2.clone();
         let wait_task1 =
-            tokio::spawn(
-                async move { delegation_clone1.wait_for_delegate_info(id1, 0).await },
-            );
+            tokio::spawn(async move { delegation_clone1.wait_for_delegate_info(id1, 0).await });
         let wait_task2 =
-            tokio::spawn(
-                async move { delegation_clone2.wait_for_delegate_info(id2, 0).await },
-            );
+            tokio::spawn(async move { delegation_clone2.wait_for_delegate_info(id2, 0).await });
 
         // Give wait tasks a moment to start
         tokio::time::sleep(Duration::from_millis(10)).await;

src/stage_delegation/mod.rs

@@ -1,5 +1,5 @@
 mod context;
 mod delegation;
 
-pub use context::{StageContext, ActorContext};
+pub use context::{ActorContext, StageContext};
 pub use delegation::StageDelegation;

src/test_utils/mod.rs

@@ -2,4 +2,4 @@
 mod mock_exec;
 
 #[cfg(test)]
-pub use mock_exec::MockExec;
+pub use mock_exec::MockExec;

tests/common/mod.rs

@@ -1,4 +1,4 @@
+pub mod insta;
 pub mod localhost;
 pub mod parquet;
-pub mod insta;
 pub mod plan;

tests/highly_distributed_query.rs

@@ -14,9 +14,12 @@ mod tests {
 
     #[tokio::test]
     async fn highly_distributed_query() -> Result<(), Box<dyn Error>> {
-        let (ctx, _guard) = start_localhost_context([
-            50050, 50051, 50053, 50054, 50055, 50056, 50057, 50058, 50059,
-        ], NoopSessionBuilder)
+        let (ctx, _guard) = start_localhost_context(
+            [
+                50050, 50051, 50053, 50054, 50055, 50056, 50057, 50058, 50059,
+            ],
+            NoopSessionBuilder,
+        )
         .await;
         register_parquet_tables(&ctx).await?;
 
@@ -62,7 +65,10 @@ mod tests {
 
         assert_eq!(
             batches.iter().map(|v| v.num_rows()).sum::<usize>(),
-            batches_distributed.iter().map(|v| v.num_rows()).sum::<usize>(),
+            batches_distributed
+                .iter()
+                .map(|v| v.num_rows())
+                .sum::<usize>(),
         );
 
         Ok(())