Examples with Cast io.prestosql.sql.tree.Cast used on opensource projects

Example 36 with Cast

use of io.prestosql.sql.tree.Cast in project hetu-core by openlookeng.

the class TestEffectivePredicateExtractor method testValues.

@Test
public void testValues() {
    TypeProvider types = TypeProvider.copyOf(ImmutableMap.<Symbol, Type>builder().put(A, BIGINT).put(B, BIGINT).build());
    // one column
    assertEquals(effectivePredicateExtractor.extract(SESSION, new ValuesNode(newId(), ImmutableList.of(A), ImmutableList.of(ImmutableList.of(bigintLiteralRowExpression(1)), ImmutableList.of(bigintLiteralRowExpression(2)))), types, typeAnalyzer), new InPredicate(AE, new InListExpression(ImmutableList.of(bigintLiteral(1), bigintLiteral(2)))));
    // one column with null
    assertEquals(effectivePredicateExtractor.extract(SESSION, new ValuesNode(newId(), ImmutableList.of(A), ImmutableList.of(ImmutableList.of(bigintLiteralRowExpression(1)), ImmutableList.of(bigintLiteralRowExpression(2)), ImmutableList.of(castToRowExpression(new Cast(new NullLiteral(), BIGINT.toString()))))), types, typeAnalyzer), or(new InPredicate(AE, new InListExpression(ImmutableList.of(bigintLiteral(1), bigintLiteral(2)))), new IsNullPredicate(AE)));
    // all nulls
    assertEquals(effectivePredicateExtractor.extract(SESSION, new ValuesNode(newId(), ImmutableList.of(A), ImmutableList.of(ImmutableList.of(castToRowExpression(new Cast(new NullLiteral(), BIGINT.toString()))))), types, typeAnalyzer), new IsNullPredicate(AE));
    // many rows
    List<List<RowExpression>> rows = IntStream.range(0, 500).mapToObj(TestEffectivePredicateExtractor::bigintLiteralRowExpression).map(ImmutableList::of).collect(toImmutableList());
    assertEquals(effectivePredicateExtractor.extract(SESSION, new ValuesNode(newId(), ImmutableList.of(A), rows), types, typeAnalyzer), new BetweenPredicate(AE, bigintLiteral(0), bigintLiteral(499)));
    // multiple columns
    assertEquals(effectivePredicateExtractor.extract(SESSION, new ValuesNode(newId(), ImmutableList.of(A, B), ImmutableList.of(ImmutableList.of(bigintLiteralRowExpression(1), bigintLiteralRowExpression(100)), ImmutableList.of(bigintLiteralRowExpression(2), bigintLiteralRowExpression(200)))), types, typeAnalyzer), and(new InPredicate(AE, new InListExpression(ImmutableList.of(bigintLiteral(1), bigintLiteral(2)))), new InPredicate(BE, new InListExpression(ImmutableList.of(bigintLiteral(100), bigintLiteral(200))))));
    // multiple columns with null
    assertEquals(effectivePredicateExtractor.extract(SESSION, new ValuesNode(newId(), ImmutableList.of(A, B), ImmutableList.of(ImmutableList.of(bigintLiteralRowExpression(1), castToRowExpression(new Cast(new NullLiteral(), BIGINT.toString()))), ImmutableList.of(castToRowExpression(new Cast(new NullLiteral(), BIGINT.toString())), bigintLiteralRowExpression(200)))), types, typeAnalyzer), and(or(new ComparisonExpression(EQUAL, AE, bigintLiteral(1)), new IsNullPredicate(AE)), or(new ComparisonExpression(EQUAL, BE, bigintLiteral(200)), new IsNullPredicate(BE))));
    // non-deterministic
    assertEquals(effectivePredicateExtractor.extract(SESSION, new ValuesNode(newId(), ImmutableList.of(A, B), ImmutableList.of(ImmutableList.of(bigintLiteralRowExpression(1), castToRowExpression(new FunctionCall(QualifiedName.of("rand"), ImmutableList.of()))))), types, typeAnalyzer), new ComparisonExpression(EQUAL, AE, bigintLiteral(1)));
    // non-constant
    assertEquals(effectivePredicateExtractor.extract(SESSION, new ValuesNode(newId(), ImmutableList.of(A), ImmutableList.of(ImmutableList.of(bigintLiteralRowExpression(1)), ImmutableList.of(castToRowExpression(BE)))), types, typeAnalyzer), TRUE_LITERAL);
}

Example 37 with Cast

use of io.prestosql.sql.tree.Cast in project hetu-core by openlookeng.

the class TestTypeValidator method testValidTypeOnlyCoercion.

@Test
public void testValidTypeOnlyCoercion() {
    Expression expression = new Cast(toSymbolReference(columnB), StandardTypes.BIGINT);
    Assignments assignments = Assignments.builder().put(planSymbolAllocator.newSymbol(expression, BIGINT), castToRowExpression(expression)).put(planSymbolAllocator.newSymbol(toSymbolReference(columnE), VARCHAR), // implicit coercion from varchar(3) to varchar
    castToRowExpression(toSymbolReference(columnE))).build();
    PlanNode node = new ProjectNode(newId(), baseTableScan, assignments);
    assertTypesValid(node);
}

Example 38 with Cast

use of io.prestosql.sql.tree.Cast in project hetu-core by openlookeng.

the class TestTypeValidator method testValidProject.

@Test
public void testValidProject() {
    Expression expression1 = new Cast(toSymbolReference(columnB), StandardTypes.BIGINT);
    Expression expression2 = new Cast(toSymbolReference(columnC), StandardTypes.BIGINT);
    Assignments assignments = Assignments.builder().put(planSymbolAllocator.newSymbol(expression1, BIGINT), castToRowExpression(expression1)).put(planSymbolAllocator.newSymbol(expression2, BIGINT), castToRowExpression(expression2)).build();
    PlanNode node = new ProjectNode(newId(), baseTableScan, assignments);
    assertTypesValid(node);
}

Example 39 with Cast

use of io.prestosql.sql.tree.Cast in project boostkit-bigdata by kunpengcompute.

the class OmniRowExpressionUtil method generateOmniExpr.

/**
 * Generates a RowExpression to ensure like query compatibility with omniruntime
 *
 * @param staticExpr Expression from openLookeng
 * @param translatedExpr RowExpression from staticFilter conversion
 * @return new Optional<RowExpression> with new, regex-syntax arguments for like
 * queries
 */
public static Optional<RowExpression> generateOmniExpr(Expression staticExpr, RowExpression translatedExpr) {
    // parse the expr
    if (staticExpr instanceof SearchedCaseExpression && translatedExpr instanceof SpecialForm) {
        return getWhenExpr((SearchedCaseExpression) staticExpr, (SpecialForm) translatedExpr);
    }
    if (translatedExpr instanceof SpecialForm) {
        SpecialForm specialExpr = (SpecialForm) translatedExpr;
        List<RowExpression> newArguments = new ArrayList<RowExpression>();
        for (int i = 0; i < specialExpr.getArguments().size(); i++) {
            RowExpression nestedExpr = specialExpr.getArguments().get(i);
            if (nestedExpr instanceof SpecialForm || nestedExpr instanceof CallExpression) {
                newArguments.add(generateOmniExpr((Expression) staticExpr.getChildren().get(i), nestedExpr).get());
            } else {
                newArguments.add(specialExpr.getArguments().get(i));
            }
        }
        Optional<RowExpression> newOmniFilter = Optional.of(new SpecialForm(specialExpr.getForm(), specialExpr.getType(), newArguments));
        return newOmniFilter;
    } else if (translatedExpr instanceof CallExpression) {
        CallExpression callExpr = (CallExpression) translatedExpr;
        List<RowExpression> newArguments = new ArrayList<RowExpression>();
        for (int i = 0; i < callExpr.getArguments().size(); i++) {
            RowExpression nestedExpr = callExpr.getArguments().get(i);
            if (nestedExpr instanceof SpecialForm || nestedExpr instanceof CallExpression) {
                newArguments.add(generateOmniExpr((Expression) staticExpr.getChildren().get(i), nestedExpr).get());
            } else {
                newArguments.add(callExpr.getArguments().get(i));
            }
        }
        Optional<RowExpression> newOmniFilter = Optional.of(new CallExpression(callExpr.getDisplayName(), callExpr.getFunctionHandle(), callExpr.getType(), newArguments));
        if ("LIKE".equals(((CallExpression) newOmniFilter.get()).getDisplayName().toUpperCase(Locale.ROOT))) {
            String sqlString = "";
            if (staticExpr.getChildren().get(1) instanceof Cast) {
                sqlString = ((StringLiteral) ((Cast) staticExpr.getChildren().get(1)).getExpression()).getValue();
            } else {
                sqlString = ((StringLiteral) staticExpr.getChildren().get(1)).getValue();
            }
            return generateLikeExpr(sqlString, newOmniFilter);
        }
        return newOmniFilter;
    }
    return Optional.of(translatedExpr);
}