SimpleAggregateAdapter with FunctionState prototype

Differential Revision: D53556520
facebookincubator · Feb 8, 2024 · 4256ea3 · 4256ea3
1 parent 98c805a
commit 4256ea3
Show file tree

Hide file tree

Showing 7 changed files with 748 additions and 3 deletions.
diff --git a/velox/exec/Aggregate.h b/velox/exec/Aggregate.h
@@ -51,6 +51,10 @@ class Aggregate {
     return resultType_;
   }
 
+  virtual void initialize(
+      const TypePtr& resultType,
+      const std::vector<VectorPtr>& args) {}
+
   // Returns the fixed number of bytes the accumulator takes on a group
   // row. Variable width accumulators will reference the variable
   // width part of the state from the fixed part.

diff --git a/velox/exec/AggregateInfo.cpp b/velox/exec/AggregateInfo.cpp
@@ -107,6 +107,8 @@ std::vector<AggregateInfo> toAggregateInfo(
         aggResultType,
         operatorCtx.driverCtx()->queryConfig());
 
+    info.function->initialize(aggResultType, info.constantInputs);
+
     if (!isStreaming) {
       auto lambdas = extractLambdaInputs(aggregate);
       if (!lambdas.empty()) {