Examples with FlinkLogicalTableAggregate org.apache.flink.table.planner.plan.nodes.logical.FlinkLogicalTableAggregate used on opensource projects

Example 1 with FlinkLogicalTableAggregate

use of org.apache.flink.table.planner.plan.nodes.logical.FlinkLogicalTableAggregate in project flink by apache.

the class StreamPhysicalPythonGroupTableAggregateRule method matches.

@Override
public boolean matches(RelOptRuleCall call) {
    FlinkLogicalTableAggregate agg = call.rel(0);
    List<AggregateCall> aggCalls = agg.getAggCallList();
    boolean existGeneralPythonFunction = aggCalls.stream().anyMatch(x -> PythonUtil.isPythonAggregate(x, PythonFunctionKind.GENERAL));
    boolean existPandasFunction = aggCalls.stream().anyMatch(x -> PythonUtil.isPythonAggregate(x, PythonFunctionKind.PANDAS));
    boolean existJavaUserDefinedFunction = aggCalls.stream().anyMatch(x -> !PythonUtil.isPythonAggregate(x, null) && !PythonUtil.isBuiltInAggregate(x));
    if (existPandasFunction || existGeneralPythonFunction) {
        if (existPandasFunction) {
            throw new TableException("Pandas UDAFs are not supported in streaming mode currently.");
        }
        if (existJavaUserDefinedFunction) {
            throw new TableException("Python UDAF and Java/Scala UDAF cannot be used together.");
        }
        return true;
    } else {
        return false;
    }
}

Example 2 with FlinkLogicalTableAggregate

use of org.apache.flink.table.planner.plan.nodes.logical.FlinkLogicalTableAggregate in project flink by apache.

the class RelTimeIndicatorConverter method visitTableAggregate.

private RelNode visitTableAggregate(FlinkLogicalTableAggregate tableAgg) {
    FlinkLogicalAggregate correspondingAgg = FlinkLogicalAggregate.create(tableAgg.getInput(), tableAgg.getGroupSet(), tableAgg.getGroupSets(), tableAgg.getAggCallList());
    FlinkLogicalAggregate convertedAgg = visitAggregate(correspondingAgg);
    return new FlinkLogicalTableAggregate(tableAgg.getCluster(), tableAgg.getTraitSet(), convertedAgg.getInput(), convertedAgg.getGroupSet(), convertedAgg.getGroupSets(), convertedAgg.getAggCallList());
}

Example 3 with FlinkLogicalTableAggregate

use of org.apache.flink.table.planner.plan.nodes.logical.FlinkLogicalTableAggregate in project flink by apache.

the class StreamPhysicalPythonGroupTableAggregateRule method convert.

@Override
public RelNode convert(RelNode rel) {
    FlinkLogicalTableAggregate agg = (FlinkLogicalTableAggregate) rel;
    FlinkRelDistribution requiredDistribution;
    if (agg.getGroupSet().cardinality() != 0) {
        requiredDistribution = FlinkRelDistribution.hash(agg.getGroupSet().asList(), true);
    } else {
        requiredDistribution = FlinkRelDistribution.SINGLETON();
    }
    RelTraitSet requiredTraitSet = rel.getCluster().getPlanner().emptyTraitSet().replace(requiredDistribution).replace(FlinkConventions.STREAM_PHYSICAL());
    RelTraitSet providedTraitSet = rel.getTraitSet().replace(FlinkConventions.STREAM_PHYSICAL());
    RelNode newInput = RelOptRule.convert(agg.getInput(), requiredTraitSet);
    return new StreamPhysicalPythonGroupTableAggregate(rel.getCluster(), providedTraitSet, newInput, rel.getRowType(), agg.getGroupSet().toArray(), JavaScalaConversionUtil.toScala(agg.getAggCallList()));
}