From 09510df9c055673ca98b6635df4b49719503cd7c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Marko=20Milenkovi=C4=87?= Date: Tue, 18 Feb 2025 22:11:09 +0000 Subject: [PATCH] fix issue with variable assignment --- ballista/core/src/client.rs | 10 +++------- ballista/core/src/execution_plans/shuffle_reader.rs | 5 ++--- 2 files changed, 5 insertions(+), 10 deletions(-) diff --git a/ballista/core/src/client.rs b/ballista/core/src/client.rs index 06d862764..3c76dc3dd 100644 --- 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 { + pub async fn try_new(host: &str, port: u16, max_message_size: usize) -> Result { 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 index 2ad389283..371e4946a 100644 --- 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: {}",