Examples with ExpressionResolver org.apache.flink.table.expressions.resolver.ExpressionResolver used on opensource projects

Example 1 with ExpressionResolver

use of org.apache.flink.table.expressions.resolver.ExpressionResolver in project flink by apache.

the class OperationTreeBuilder method filter.

public QueryOperation filter(Expression condition, QueryOperation child) {
    ExpressionResolver resolver = getResolver(child);
    ResolvedExpression resolvedExpression = resolveSingleExpression(condition, resolver);
    DataType conditionType = resolvedExpression.getOutputDataType();
    if (!conditionType.getLogicalType().is(BOOLEAN)) {
        throw new ValidationException("Filter operator requires a boolean expression as input," + " but $condition is of type " + conditionType);
    }
    return new FilterQueryOperation(resolvedExpression, child);
}

Example 2 with ExpressionResolver

use of org.apache.flink.table.expressions.resolver.ExpressionResolver in project flink by apache.

the class OperationTreeBuilder method join.

public QueryOperation join(QueryOperation left, QueryOperation right, JoinType joinType, Optional<Expression> condition, boolean correlated) {
    ExpressionResolver resolver = getResolver(left, right);
    Optional<ResolvedExpression> resolvedCondition = condition.map(expr -> resolveSingleExpression(expr, resolver));
    return joinOperationFactory.create(left, right, joinType, resolvedCondition.orElse(valueLiteral(true)), correlated);
}

Example 3 with ExpressionResolver

use of org.apache.flink.table.expressions.resolver.ExpressionResolver in project flink by apache.

the class OperationTreeBuilder method windowTableAggregate.

public QueryOperation windowTableAggregate(List<Expression> groupingExpressions, GroupWindow window, List<Expression> windowProperties, Expression tableAggFunction, QueryOperation child) {
    // Step1: add a default name to the call in the grouping expressions, e.g., groupBy(a % 5)
    // to
    // groupBy(a % 5 as TMP_0). We need a name for every column so that to perform alias for the
    // table aggregate function in Step4.
    List<Expression> newGroupingExpressions = addAliasToTheCallInAggregate(child.getResolvedSchema().getColumnNames(), groupingExpressions);
    // Step2: resolve expressions, including grouping, aggregates and window properties.
    ExpressionResolver resolver = getAggResolver(child, groupingExpressions);
    ResolvedGroupWindow resolvedWindow = aggregateOperationFactory.createResolvedWindow(window, resolver);
    ExpressionResolver resolverWithWindowReferences = getResolverBuilder(child).withLocalReferences(localRef(resolvedWindow.getAlias(), resolvedWindow.getTimeAttribute().getOutputDataType())).build();
    List<ResolvedExpression> convertedGroupings = resolverWithWindowReferences.resolve(newGroupingExpressions);
    List<ResolvedExpression> convertedAggregates = resolverWithWindowReferences.resolve(Collections.singletonList(tableAggFunction));
    List<ResolvedExpression> convertedProperties = resolverWithWindowReferences.resolve(windowProperties);
    Tuple2<ResolvedExpression, List<String>> resolvedFunctionAndAlias = aggregateOperationFactory.extractTableAggFunctionAndAliases(convertedAggregates.get(0));
    // Step3: create window table agg operation
    QueryOperation tableAggOperation = aggregateOperationFactory.createWindowAggregate(convertedGroupings, Collections.singletonList(resolvedFunctionAndAlias.f0), convertedProperties, resolvedWindow, child);
    // window attribute.
    return aliasBackwardFields(tableAggOperation, resolvedFunctionAndAlias.f1, groupingExpressions.size());
}

Example 4 with ExpressionResolver

use of org.apache.flink.table.expressions.resolver.ExpressionResolver in project flink by apache.

the class OperationTreeBuilder method windowAggregate.

public QueryOperation windowAggregate(List<Expression> groupingExpressions, GroupWindow window, List<Expression> windowProperties, List<Expression> aggregates, QueryOperation child) {
    ExpressionResolver resolver = getAggResolver(child, groupingExpressions);
    ResolvedGroupWindow resolvedWindow = aggregateOperationFactory.createResolvedWindow(window, resolver);
    ExpressionResolver resolverWithWindowReferences = getResolverBuilder(child).withLocalReferences(localRef(resolvedWindow.getAlias(), resolvedWindow.getTimeAttribute().getOutputDataType())).build();
    List<ResolvedExpression> convertedGroupings = resolverWithWindowReferences.resolve(groupingExpressions);
    List<ResolvedExpression> convertedAggregates = resolverWithWindowReferences.resolve(aggregates);
    List<ResolvedExpression> convertedProperties = resolverWithWindowReferences.resolve(windowProperties);
    return aggregateOperationFactory.createWindowAggregate(convertedGroupings, convertedAggregates, convertedProperties, resolvedWindow, child);
}

Example 5 with ExpressionResolver

use of org.apache.flink.table.expressions.resolver.ExpressionResolver in project flink by apache.

the class OperationTreeBuilder method dropColumns.

public QueryOperation dropColumns(List<Expression> fieldLists, QueryOperation child) {
    ExpressionResolver resolver = getResolver(child);
    List<String> inputFieldNames = child.getResolvedSchema().getColumnNames();
    List<Expression> finalFields = ColumnOperationUtils.dropFields(inputFieldNames, resolver.resolveExpanding(fieldLists));
    return project(finalFields, child, false);
}