finalise compute

apache · Apr 15, 2024 · d9c5070 · d9c5070
1 parent 81202ac
commit d9c5070
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...ain/java/org/apache/druid/query/rowsandcols/semantic/DefaultFramedOnHeapAggregatable.java b/...ain/java/org/apache/druid/query/rowsandcols/semantic/DefaultFramedOnHeapAggregatable.java
@@ -459,7 +459,7 @@ private AppendableRowsAndColumns computeCumulativeAggregates(AggregatorFactory[]
  // In such a case the priming of results is not needed
  if (rowIdProvider.get() == numRows) {
  for (int i = 0; i < aggs.length; i++) {
- results[i][resultStorageIndex] = aggs[i].get();
+ results[i][resultStorageIndex] = aggFactories[i].finalizeComputation(aggs[i].get());
  }
  ++resultStorageIndex;
  }
@@ -555,7 +555,7 @@ private AppendableRowsAndColumns computeReverseCumulativeAggregates(AggregatorFa
  // In such a case the priming of results is not needed
  if (rowIdProvider.get() < 0) {
  for (int i = 0; i < aggs.length; i++) {
- results[i][resultStorageIndex] = aggs[i].get();
+ results[i][resultStorageIndex] = aggFactories[i].finalizeComputation(aggs[i].get());
  }
  --resultStorageIndex;
  }