org.apache.flink.table.runtime.operators.aggregate

Class MiniBatchGroupAggFunction

java.lang.Object

org.apache.flink.table.runtime.operators.bundle.MapBundleFunction<RowData,List<RowData>,RowData,RowData>

org.apache.flink.table.runtime.operators.aggregate.MiniBatchGroupAggFunction

All Implemented Interfaces:

Serializable, Function

public class MiniBatchGroupAggFunction
extends MapBundleFunction<RowData,List<RowData>,RowData,RowData>

Aggregate Function used for the groupby (without window) aggregate in miniBatch mode.

This function buffers input row in heap HashMap, and aggregates them when minibatch invoked.

See Also:

Serialized Form

Field Summary

Fields inherited from class org.apache.flink.table.runtime.operators.bundle.MapBundleFunction

ctx

Constructor Summary

Constructors

Constructor and Description
MiniBatchGroupAggFunction(GeneratedAggsHandleFunction genAggsHandler, GeneratedRecordEqualiser genRecordEqualiser, LogicalType[] accTypes, RowType inputType, int indexOfCountStar, boolean generateUpdateBefore, long stateRetentionTime) Creates a MiniBatchGroupAggFunction.

Method Summary

Modifier and Type	Method and Description
List<RowData>	addInput(List<RowData> value, RowData input) Adds the given input to the given value, returning the new bundle value.
void	close()
void	finishBundle(Map<RowData,List<RowData>> buffer, Collector<RowData> out) Called when a bundle is finished.
void	open(ExecutionContext ctx)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

MiniBatchGroupAggFunction

public MiniBatchGroupAggFunction(GeneratedAggsHandleFunction genAggsHandler,
                                 GeneratedRecordEqualiser genRecordEqualiser,
                                 LogicalType[] accTypes,
                                 RowType inputType,
                                 int indexOfCountStar,
                                 boolean generateUpdateBefore,
                                 long stateRetentionTime)

Creates a MiniBatchGroupAggFunction.

Parameters:

genAggsHandler - The code generated function used to handle aggregates.

genRecordEqualiser - The code generated equaliser used to equal RowData.

accTypes - The accumulator types.

inputType - The input row type.

indexOfCountStar - The index of COUNT(*) in the aggregates. -1 when the input doesn't contain COUNT(*), i.e. doesn't contain retraction messages. We make sure there is a COUNT(*) if input stream contains retraction.

generateUpdateBefore - Whether this operator will generate UPDATE_BEFORE messages.

stateRetentionTime - state idle retention time which unit is MILLISECONDS.

Method Detail

open

public void open(ExecutionContext ctx)
          throws Exception

Overrides:

open in class MapBundleFunction<RowData,List<RowData>,RowData,RowData>

Throws:

Exception

addInput

public List<RowData> addInput(@Nullable
                              List<RowData> value,
                              RowData input)
                       throws Exception

Description copied from class: MapBundleFunction

Adds the given input to the given value, returning the new bundle value.

Specified by:

addInput in class MapBundleFunction<RowData,List<RowData>,RowData,RowData>

Parameters:

value - the existing bundle value, maybe null

input - the given input, not null

Throws:

Exception

finishBundle

public void finishBundle(Map<RowData,List<RowData>> buffer,
                         Collector<RowData> out)
                  throws Exception

Description copied from class: MapBundleFunction

Called when a bundle is finished. Transform a bundle to zero, one, or more output elements.

Specified by:

finishBundle in class MapBundleFunction<RowData,List<RowData>,RowData,RowData>

Throws:

Exception

close

public void close()
           throws Exception

Overrides:

close in class MapBundleFunction<RowData,List<RowData>,RowData,RowData>

Throws:

Exception