[HSTACK] - fix CoalescePartitionsExecNode proto

Author: aditanase

datafusion/proto/proto/datafusion.proto

@@ -1213,6 +1213,7 @@ message CoalesceBatchesExecNode {
 
 message CoalescePartitionsExecNode {
   PhysicalPlanNode input = 1;
+  optional uint32 fetch = 2;
 }
 
 message PhysicalHashRepartition {

datafusion/proto/src/generated/pbjson.rs

@@ -312,7 +312,8 @@ impl AsExecutionPlan for protobuf::PhysicalPlanNode {
             PhysicalPlanType::Merge(merge) => {
                 let input: Arc<dyn ExecutionPlan> =
                     into_physical_plan(&merge.input, registry, runtime, extension_codec)?;
-                Ok(Arc::new(CoalescePartitionsExec::new(input)))
+                Ok(CoalescePartitionsExec::new(input)
+                    .with_fetch(merge.fetch.map(|f| f as usize)).unwrap())
             }
             PhysicalPlanType::Repartition(repart) => {
                 let input: Arc<dyn ExecutionPlan> = into_physical_plan(
@@ -1736,6 +1737,7 @@ impl AsExecutionPlan for protobuf::PhysicalPlanNode {
                 physical_plan_type: Some(PhysicalPlanType::Merge(Box::new(
                     protobuf::CoalescePartitionsExecNode {
                         input: Some(Box::new(input)),
+                        fetch: exec.fetch().map(|n| n as u32),
                     },
                 ))),
             });