Examples with TableAggregationFunction io.confluent.ksql.execution.function.TableAggregationFunction used on opensource projects

Example 1 with TableAggregationFunction

use of io.confluent.ksql.execution.function.TableAggregationFunction in project ksql by confluentinc.

the class SchemaKGroupedTable method aggregate.

@Override
public SchemaKTable<GenericKey> aggregate(final List<ColumnName> nonAggregateColumns, final List<FunctionCall> aggregations, final Optional<WindowExpression> windowExpression, final FormatInfo valueFormat, final Stacker contextStacker) {
    if (windowExpression.isPresent()) {
        throw new KsqlException("Windowing not supported for table aggregations.");
    }
    final List<String> unsupportedFunctionNames = aggregations.stream().map(call -> UdafUtil.resolveAggregateFunction(functionRegistry, call, schema, ksqlConfig)).filter(function -> !(function instanceof TableAggregationFunction)).map(KsqlAggregateFunction::name).map(FunctionName::text).distinct().collect(Collectors.toList());
    if (!unsupportedFunctionNames.isEmpty()) {
        final String postfix = unsupportedFunctionNames.size() == 1 ? "" : "s";
        throw new KsqlException("The aggregation function" + postfix + " " + GrammaticalJoiner.and().join(unsupportedFunctionNames) + " cannot be applied to a table source, only to a stream source.");
    }
    final TableAggregate step = ExecutionStepFactory.tableAggregate(contextStacker, sourceTableStep, InternalFormats.of(keyFormat, valueFormat), nonAggregateColumns, aggregations);
    return new SchemaKTable<>(step, resolveSchema(step), keyFormat, ksqlConfig, functionRegistry);
}

Example 2 with TableAggregationFunction

use of io.confluent.ksql.execution.function.TableAggregationFunction in project ksql by confluentinc.

the class KudafUndoAggregator method apply.

@SuppressWarnings({ "unchecked", "rawtypes" })
@Override
public GenericRow apply(final GenericKey k, final GenericRow rowValue, final GenericRow aggRowValue) {
    final GenericRow result = GenericRow.fromList(aggRowValue.values());
    for (int idx = 0; idx < nonAggColumnCount; idx++) {
        result.set(idx, rowValue.get(idx));
    }
    for (int idx = nonAggColumnCount; idx < columnCount; idx++) {
        final TableAggregationFunction function = aggregateFunctions.get(idx - nonAggColumnCount);
        final Object argument = rowValue.get(function.getArgIndexInValue());
        final Object previous = result.get(idx);
        result.set(idx, function.undo(argument, previous));
    }
    return result;
}