Increase max_decoding_message_size

Author: gabotechs

src/execution_plans/network_coalesce.rs

@@ -287,6 +287,7 @@ impl ExecutionPlan for NetworkCoalesceExec {
         let stream = async move {
             let channel = channel_resolver.get_channel_for_url(&url).await?;
             let stream = FlightServiceClient::new(channel)
+                .max_decoding_message_size(2 * 1024 * 1024 * 1024)
                 .do_get(ticket)
                 .await
                 .map_err(map_status_to_datafusion_error)?

src/execution_plans/network_shuffle.rs

@@ -324,6 +324,7 @@ impl ExecutionPlan for NetworkShuffleExec {
 
                 let channel = channel_resolver.get_channel_for_url(&url).await?;
                 let stream = FlightServiceClient::new(channel)
+                    .max_decoding_message_size(2 * 1024 * 1024 * 1024)
                     .do_get(ticket)
                     .await
                     .map_err(map_status_to_datafusion_error)?