add combine partial to protobuf.

ArroyoSystems · May 7, 2024 · b97a5b6 · b97a5b6
1 parent 019d66f
commit b97a5b6
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 3 deletions.
diff --git a/datafusion/proto/proto/datafusion.proto b/datafusion/proto/proto/datafusion.proto
@@ -1726,6 +1726,7 @@ enum AggregateMode {
   FINAL_PARTITIONED = 2;
   SINGLE = 3;
   SINGLE_PARTITIONED = 4;
+  COMBINE_PARTIAL = 5;
 }
 
 message PartiallySortedInputOrderMode {

diff --git a/datafusion/proto/src/generated/pbjson.rs b/datafusion/proto/src/generated/pbjson.rs
diff --git a/datafusion/proto/src/generated/prost.rs b/datafusion/proto/src/generated/prost.rs
diff --git a/datafusion/proto/src/physical_plan/mod.rs b/datafusion/proto/src/physical_plan/mod.rs
@@ -409,6 +409,9 @@ impl AsExecutionPlan for PhysicalPlanNode {
                     protobuf::AggregateMode::SinglePartitioned => {
                         AggregateMode::SinglePartitioned
                     }
+                    protobuf::AggregateMode::CombinePartial => {
+                        AggregateMode::CombinePartial
+                    }
                 };
 
                 let num_expr = hash_agg.group_expr.len();
@@ -1471,10 +1474,8 @@ impl AsExecutionPlan for PhysicalPlanNode {
                 AggregateMode::Single => protobuf::AggregateMode::Single,
                 AggregateMode::SinglePartitioned => {
                     protobuf::AggregateMode::SinglePartitioned
-                },
-                AggregateMode::CombinePartial => {
-                    unimplemented!()
                 }
+                AggregateMode::CombinePartial => protobuf::AggregateMode::CombinePartial,
             };
 
             let input_schema = exec.input_schema();