Examples with ExpressionTypeManager io.confluent.ksql.execution.util.ExpressionTypeManager used on opensource projects

Example 1 with ExpressionTypeManager

use of io.confluent.ksql.execution.util.ExpressionTypeManager in project ksql by confluentinc.

the class FilterTypeValidator method getExpressionReturnType.

private SqlType getExpressionReturnType(final Expression exp) {
    final ExpressionTypeManager expressionTypeManager = new ExpressionTypeManager(schema, functionRegistry);
    // Rewrite the expression with magic timestamps, so type checking can pass
    final Expression magicTimestampRewrite = new StatementRewriteForMagicPseudoTimestamp().rewrite(exp);
    try {
        return expressionTypeManager.getExpressionSqlType(magicTimestampRewrite);
    } catch (KsqlException e) {
        throw new KsqlStatementException("Error in " + filterType.name() + " expression: " + e.getMessage(), exp.toString());
    }
}

Example 2 with ExpressionTypeManager

use of io.confluent.ksql.execution.util.ExpressionTypeManager in project ksql by confluentinc.

the class StepSchemaResolver method handleStreamSelectKeyV1.

private LogicalSchema handleStreamSelectKeyV1(final LogicalSchema sourceSchema, final StreamSelectKeyV1 step) {
    final ExpressionTypeManager expressionTypeManager = new ExpressionTypeManager(sourceSchema, functionRegistry);
    final SqlType keyType = expressionTypeManager.getExpressionSqlType(step.getKeyExpression());
    return LogicalSchema.builder().keyColumn(SystemColumns.ROWKEY_NAME, keyType).valueColumns(sourceSchema.value()).build();
}

Example 3 with ExpressionTypeManager

use of io.confluent.ksql.execution.util.ExpressionTypeManager in project ksql by confluentinc.

the class TermCompiler method visitCreateArrayExpression.

@Override
public Term visitCreateArrayExpression(final CreateArrayExpression exp, final Context context) {
    final List<Expression> expressions = CoercionUtil.coerceUserList(exp.getValues(), expressionTypeManager, context.getLambdaSqlTypeMapping()).expressions();
    final List<Term> arrayTerms = expressions.stream().map(value -> process(value, context)).collect(ImmutableList.toImmutableList());
    final SqlType sqlType = expressionTypeManager.getExpressionSqlType(exp, context.getLambdaSqlTypeMapping());
    return new CreateArrayTerm(arrayTerms, sqlType);
}

Example 4 with ExpressionTypeManager

use of io.confluent.ksql.execution.util.ExpressionTypeManager in project ksql by confluentinc.

the class UdtfUtil method resolveTableFunction.

public static KsqlTableFunction resolveTableFunction(final FunctionRegistry functionRegistry, final FunctionCall functionCall, final LogicalSchema schema) {
    final ExpressionTypeManager expressionTypeManager = new ExpressionTypeManager(schema, functionRegistry);
    final List<Expression> functionArgs = functionCall.getArguments();
    final List<SqlArgument> argTypes = functionArgs.isEmpty() ? ImmutableList.of(SqlArgument.of(FunctionRegistry.DEFAULT_FUNCTION_ARG_SCHEMA)) : functionArgs.stream().map(expressionTypeManager::getExpressionSqlType).map(SqlArgument::of).collect(Collectors.toList());
    return functionRegistry.getTableFunction(functionCall.getName(), argTypes);
}

Example 5 with ExpressionTypeManager

use of io.confluent.ksql.execution.util.ExpressionTypeManager in project ksql by confluentinc.

the class InListEvaluator method preprocess.

/**
 * Preprocess the list of possible expression, ensuring compatible types, performing literal
 * coercion and removing null literals (which can never match).
 *
 * @param predicate The predicate to process
 * @param typeManager the type manager for the predicate
 * @param lambdaTypeMapping mapping of lambda variables to type
 * @return {@code predicate} after processing.
 */
public static InPredicate preprocess(final InPredicate predicate, final ExpressionTypeManager typeManager, final Map<String, SqlType> lambdaTypeMapping) {
    final List<Expression> nonNull = ImmutableList.<Expression>builder().add(predicate.getValue()).addAll(predicate.getValueList().getValues().stream().filter(e -> !(e instanceof NullLiteral)).collect(Collectors.toList())).build();
    final List<Expression> coerced = CoercionUtil.coerceUserList(nonNull, typeManager, lambdaTypeMapping).expressions();
    return new InPredicate(predicate.getLocation(), coerced.get(0), new InListExpression(predicate.getValueList().getLocation(), coerced.subList(1, coerced.size())));
}