fix issue with variable assignment

milenkovicm · Feb 18, 2025 · 09510df · 09510df
1 parent ca7d13b
commit 09510df
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 10 deletions.
diff --git a/ballista/core/src/client.rs b/ballista/core/src/client.rs
@@ -61,11 +61,7 @@ const IO_RETRY_WAIT_TIME_MS: u64 = 3000;
 impl BallistaClient {
     /// Create a new BallistaClient to connect to the executor listening on the specified
     /// host and port
-    pub async fn try_new(
-        host: &str,
-        port: u16,
-        grpc_client_max_message_size: usize,
-    ) -> Result<Self> {
+    pub async fn try_new(host: &str, port: u16, max_message_size: usize) -> Result<Self> {
         let addr = format!("http://{host}:{port}");
         debug!("BallistaClient connecting to {}", addr);
         let connection =
@@ -77,8 +73,8 @@ impl BallistaClient {
                 ))
                 })?;
         let flight_client = FlightServiceClient::new(connection)
-            .max_decoding_message_size(grpc_client_max_message_size)
-            .max_encoding_message_size(grpc_client_max_message_size);
+            .max_decoding_message_size(max_message_size)
+            .max_encoding_message_size(max_message_size);
 
         debug!("BallistaClient connected OK: {:?}", flight_client);
 

diff --git a/ballista/core/src/execution_plans/shuffle_reader.rs b/ballista/core/src/execution_plans/shuffle_reader.rs
@@ -149,9 +149,8 @@ impl ExecutionPlan for ShuffleReaderExec {
 
         let config = context.session_config();
 
-        let max_request_num = config.ballista_grpc_client_max_message_size();
-        let max_message_size =
-            config.ballista_shuffle_reader_maximum_in_flight_requests();
+        let max_request_num = config.ballista_shuffle_reader_maximum_in_flight_requests();
+        let max_message_size = config.ballista_grpc_client_max_message_size();
 
         log::debug!(
             "ShuffleReaderExec::execute({}) max_request_num: {}, max_message_size: {}",