Skip to content

AggregateFunction -- Aggregate Function Expressions

AggregateFunction is the <> for Catalyst expressions that represent aggregate functions.

AggregateFunction is used wrapped inside a AggregateExpression (using <> method) when:

[source, scala]

import org.apache.spark.sql.functions.collect_list scala> val fn = collect_list("gid") fn: org.apache.spark.sql.Column = collect_list(gid)

import org.apache.spark.sql.catalyst.expressions.aggregate.AggregateExpression scala> val aggFn = fn.expr.asInstanceOf[AggregateExpression].aggregateFunction aggFn: org.apache.spark.sql.catalyst.expressions.aggregate.AggregateFunction = collect_list('gid, 0, 0)

scala> println(aggFn.numberedTreeString) 00 collect_list('gid, 0, 0) 01 +- 'gid

NOTE: Aggregate functions are not foldable, i.e. FIXME

[[top-level-expressions]] .AggregateFunction Top-Level Catalyst Expressions [cols="1,1,2",options="header",width="100%"] |=== | Name | Behaviour | Examples


=== [[contract]] AggregateFunction Contract

[source, scala]

abstract class AggregateFunction extends Expression { def aggBufferSchema: StructType def aggBufferAttributes: Seq[AttributeReference] def inputAggBufferAttributes: Seq[AttributeReference] def defaultResult: Option[Literal] = None }

.AggregateFunction Contract [cols="1,2",options="header",width="100%"] |=== | Method | Description

| [[aggBufferSchema]] aggBufferSchema | Schema of an aggregation buffer to hold partial aggregate results.

Used mostly in expressions/[ScalaUDAF] and[AggregationIterator]

| [[aggBufferAttributes]] aggBufferAttributes a| <> of an aggregation buffer to hold partial aggregate results.

Used in:

  • AggregateExpression for references
  • Expression-based aggregate's bufferSchema in[DeclarativeAggregate]
  • ...
[[inputAggBufferAttributes]] inputAggBufferAttributes

| [[defaultResult]] defaultResult | Defaults to None.


=== [[toAggregateExpression]] Creating AggregateExpression for AggregateFunction -- toAggregateExpression Method

toAggregateExpression(): AggregateExpression  // <1>
toAggregateExpression(isDistinct: Boolean): AggregateExpression
<1> Calls the other toAggregateExpression with isDistinct disabled (i.e. false)

toAggregateExpression creates a AggregateExpression for the current AggregateFunction with Complete aggregate mode.

toAggregateExpression is used in:

  • functions object's withAggregateFunction block to create a[Column] with AggregateExpression for a AggregateFunction

Last update: 2021-02-18