add combine partial to protobuf.

ArroyoSystems · Jul 23, 2024 · bb69311 · bb69311
1 parent 88613ad
commit bb69311
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
@@ -1091,6 +1091,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
@@ -388,6 +388,9 @@ impl AsExecutionPlan for protobuf::PhysicalPlanNode {
                     protobuf::AggregateMode::SinglePartitioned => {
                         AggregateMode::SinglePartitioned
                     }
+                    protobuf::AggregateMode::CombinePartial => {
+                        AggregateMode::CombinePartial
+                    }
                 };
 
                 let num_expr = hash_agg.group_expr.len();
@@ -1469,10 +1472,8 @@ impl AsExecutionPlan for protobuf::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();