Examples with CreateArrayExpression io.confluent.ksql.execution.expression.tree.CreateArrayExpression used on opensource projects

Example 11 with CreateArrayExpression

use of io.confluent.ksql.execution.expression.tree.CreateArrayExpression 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 12 with CreateArrayExpression

use of io.confluent.ksql.execution.expression.tree.CreateArrayExpression in project ksql by confluentinc.

the class CoercionUtilTest method shouldNotCoerceArrayOfIncompatibleLiterals.

@Test
public void shouldNotCoerceArrayOfIncompatibleLiterals() {
    // Given:
    final ImmutableList<Expression> expressions = ImmutableList.of(new CreateArrayExpression(ImmutableList.of(new IntegerLiteral(10))), new CreateArrayExpression(ImmutableList.of(new BooleanLiteral(false))));
    // When:
    final Exception e = assertThrows(KsqlException.class, () -> CoercionUtil.coerceUserList(expressions, typeManager));
    // Then:
    assertThat(e.getMessage(), startsWith("operator does not exist: ARRAY<INTEGER> = ARRAY<BOOLEAN> (ARRAY[false])"));
}

Example 13 with CreateArrayExpression

use of io.confluent.ksql.execution.expression.tree.CreateArrayExpression in project ksql by confluentinc.

the class CoercionUtilTest method shouldCoerceArrayOfCompatibleLiterals.

@Test
public void shouldCoerceArrayOfCompatibleLiterals() {
    // Given:
    final ImmutableList<Expression> expressions = ImmutableList.of(new CreateArrayExpression(ImmutableList.of(new IntegerLiteral(10), new IntegerLiteral(289476))), new CreateArrayExpression(ImmutableList.of(new StringLiteral("123456789000"), new StringLiteral("22"), new StringLiteral("\t -100 \t"))));
    // When:
    final Result result = CoercionUtil.coerceUserList(expressions, typeManager);
    // Then:
    assertThat(result.commonType(), is(Optional.of(SqlTypes.array(SqlTypes.BIGINT))));
    assertThat(result.expressions(), is(ImmutableList.of(cast(new CreateArrayExpression(ImmutableList.of(new IntegerLiteral(10), new IntegerLiteral(289476))), SqlTypes.array(SqlTypes.BIGINT)), cast(new CreateArrayExpression(ImmutableList.of(new StringLiteral("123456789000"), new StringLiteral("22"), new StringLiteral("\t -100 \t"))), SqlTypes.array(SqlTypes.BIGINT)))));
}