Search in sources :

Example 56 with RexCall

use of org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.rex.RexCall in project drill by apache.

the class DrillRelOptUtil method isProjectFlatten.

@SuppressWarnings("deprecation")
public static boolean isProjectFlatten(RelNode project) {
    assert project instanceof Project : "Rel is NOT an instance of project!";
    for (RexNode rex : project.getChildExps()) {
        if (rex instanceof RexCall) {
            RexCall function = (RexCall) rex;
            String functionName = function.getOperator().getName();
            if (functionName.equalsIgnoreCase("flatten")) {
                return true;
            }
        }
    }
    return false;
}
Also used : RexCall(org.apache.calcite.rex.RexCall) Project(org.apache.calcite.rel.core.Project) RexNode(org.apache.calcite.rex.RexNode)

Example 57 with RexCall

use of org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.rex.RexCall in project drill by apache.

the class DrillRelMdSelectivity method getScanSelectivityInternal.

private double getScanSelectivityInternal(TableMetadata tableMetadata, RexNode predicate, List<SchemaPath> fieldNames, RexBuilder rexBuilder) {
    double sel = 1.0;
    if ((predicate == null) || predicate.isAlwaysTrue()) {
        return sel;
    }
    List<RexNode> conjuncts1 = RelOptUtil.conjunctions(predicate);
    // a Set that holds range predicates that are combined based on whether they are defined on the same column
    Set<RexNode> combinedRangePredicates = new HashSet<>();
    // pre-process the conjuncts such that predicates on the same column are grouped together
    List<RexNode> conjuncts2 = preprocessRangePredicates(conjuncts1, fieldNames, rexBuilder, combinedRangePredicates);
    for (RexNode pred : conjuncts2) {
        double orSel = 0;
        for (RexNode orPred : RelOptUtil.disjunctions(pred)) {
            if (isMultiColumnPredicate(orPred) && !combinedRangePredicates.contains(orPred)) {
                Set uniqueRefs = new HashSet<>();
                uniqueRefs.add(DrillRelOptUtil.findAllRexInputRefs(orPred));
                // If equality predicate involving single column - selectivity is 1.0
                if (uniqueRefs.size() == 1) {
                    try {
                        RexVisitor<Void> visitor = new RexVisitorImpl<Void>(true) {

                            @Override
                            public Void visitCall(RexCall call) {
                                if (call.getKind() != SqlKind.EQUALS) {
                                    throw new Util.FoundOne(call);
                                }
                                return super.visitCall(call);
                            }
                        };
                        pred.accept(visitor);
                        orSel += 1.0;
                    } catch (Util.FoundOne e) {
                        // CALCITE guess
                        orSel += RelMdUtil.guessSelectivity(orPred);
                    }
                }
            } else if (orPred.isA(SqlKind.EQUALS)) {
                orSel += computeEqualsSelectivity(tableMetadata, orPred, fieldNames);
            } else if (orPred.isA(RANGE_PREDICATE) || combinedRangePredicates.contains(orPred)) {
                orSel += computeRangeSelectivity(tableMetadata, orPred, fieldNames);
            } else if (orPred.isA(SqlKind.NOT_EQUALS)) {
                orSel += 1.0 - computeEqualsSelectivity(tableMetadata, orPred, fieldNames);
            } else if (orPred.isA(SqlKind.LIKE)) {
                // LIKE selectivity is 5% more than a similar equality predicate, capped at CALCITE guess
                orSel += Math.min(computeEqualsSelectivity(tableMetadata, orPred, fieldNames) + LIKE_PREDICATE_SELECTIVITY, guessSelectivity(orPred));
            } else if (orPred.isA(SqlKind.NOT)) {
                if (orPred instanceof RexCall) {
                    // LIKE selectivity is 5% more than a similar equality predicate, capped at CALCITE guess
                    RexNode childOp = ((RexCall) orPred).getOperands().get(0);
                    if (childOp.isA(SqlKind.LIKE)) {
                        orSel += 1.0 - Math.min(computeEqualsSelectivity(tableMetadata, childOp, fieldNames) + LIKE_PREDICATE_SELECTIVITY, guessSelectivity(childOp));
                    } else {
                        orSel += 1.0 - guessSelectivity(orPred);
                    }
                }
            } else if (orPred.isA(SqlKind.IS_NULL)) {
                orSel += 1.0 - computeIsNotNullSelectivity(tableMetadata, orPred, fieldNames);
            } else if (orPred.isA(SqlKind.IS_NOT_NULL)) {
                orSel += computeIsNotNullSelectivity(tableMetadata, orPred, fieldNames);
            } else {
                // Use the CALCITE guess.
                orSel += guessSelectivity(orPred);
            }
        }
        sel *= orSel;
    }
    // Cap selectivity if it exceeds 1.0
    return (sel > 1.0) ? 1.0 : sel;
}
Also used : HashSet(java.util.HashSet) EnumSet(java.util.EnumSet) Set(java.util.Set) RelOptUtil(org.apache.calcite.plan.RelOptUtil) RelMdUtil(org.apache.calcite.rel.metadata.RelMdUtil) RexUtil(org.apache.calcite.rex.RexUtil) DrillRelOptUtil(org.apache.drill.exec.planner.common.DrillRelOptUtil) PrelUtil(org.apache.drill.exec.planner.physical.PrelUtil) Util(org.apache.calcite.util.Util) RexCall(org.apache.calcite.rex.RexCall) RexVisitorImpl(org.apache.calcite.rex.RexVisitorImpl) RexNode(org.apache.calcite.rex.RexNode) HashSet(java.util.HashSet)

Example 58 with RexCall

use of org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.rex.RexCall in project drill by apache.

the class DrillRelMdSelectivity method findRexInputRef.

private static RexInputRef findRexInputRef(final RexNode node) {
    try {
        RexVisitor<Void> visitor = new RexVisitorImpl<Void>(true) {

            @Override
            public Void visitCall(RexCall call) {
                for (RexNode child : call.getOperands()) {
                    child.accept(this);
                }
                return super.visitCall(call);
            }

            @Override
            public Void visitInputRef(RexInputRef inputRef) {
                throw new Util.FoundOne(inputRef);
            }
        };
        node.accept(visitor);
        return null;
    } catch (Util.FoundOne e) {
        Util.swallow(e, null);
        return (RexInputRef) e.getNode();
    }
}
Also used : RexCall(org.apache.calcite.rex.RexCall) RexInputRef(org.apache.calcite.rex.RexInputRef) RelOptUtil(org.apache.calcite.plan.RelOptUtil) RelMdUtil(org.apache.calcite.rel.metadata.RelMdUtil) RexUtil(org.apache.calcite.rex.RexUtil) DrillRelOptUtil(org.apache.drill.exec.planner.common.DrillRelOptUtil) PrelUtil(org.apache.drill.exec.planner.physical.PrelUtil) Util(org.apache.calcite.util.Util) RexVisitorImpl(org.apache.calcite.rex.RexVisitorImpl) RexNode(org.apache.calcite.rex.RexNode)

Example 59 with RexCall

use of org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.rex.RexCall in project drill by apache.

the class RexToMongoTranslator method visitCall.

@Override
public BsonValue visitCall(RexCall call) {
    String name = isItem(call);
    if (name != null) {
        return new BsonString("'$" + name + "'");
    }
    List<BsonValue> strings = call.operands.stream().map(operand -> operand.accept(this)).collect(Collectors.toList());
    if (call.getKind() == SqlKind.CAST) {
        return strings.get(0);
    }
    String stdOperator = MONGO_OPERATORS.get(call.getOperator());
    if (stdOperator != null) {
        return new BsonDocument(stdOperator, new BsonArray(strings));
    }
    if (call.getOperator() == SqlStdOperatorTable.ITEM) {
        RexNode op1 = call.operands.get(1);
        if (op1 instanceof RexLiteral) {
            if (op1.getType().getSqlTypeName() == SqlTypeName.INTEGER) {
                return new BsonDocument("$arrayElemAt", new BsonArray(Arrays.asList(strings.get(0), new BsonInt32(((RexLiteral) op1).getValueAs(Integer.class)))));
            } else if (op1.getType().getSqlTypeName() == SqlTypeName.CHAR) {
                return new BsonString(strings.get(0).asString().getValue() + "." + ((RexLiteral) op1).getValueAs(String.class));
            }
        }
    }
    if (call.getOperator() == SqlStdOperatorTable.CASE) {
        // case(a, b, c)  -> $cond:[a, b, c]
        // case(a, b, c, d) -> $cond:[a, b, $cond:[c, d, null]]
        // case(a, b, c, d, e) -> $cond:[a, b, $cond:[c, d, e]]
        BsonDocument result = new BsonDocument();
        BsonArray args = new BsonArray();
        result.put("$cond", args);
        for (int i = 0; i < strings.size(); i += 2) {
            args.add(strings.get(i));
            args.add(strings.get(i + 1));
            if (i == strings.size() - 3) {
                args.add(strings.get(i + 2));
                break;
            }
            if (i == strings.size() - 2) {
                args.add(BsonNull.VALUE);
                break;
            }
            BsonArray innerArgs = new BsonArray();
            BsonDocument innerDocument = new BsonDocument();
            innerDocument.put("$cond", innerArgs);
            args.add(innerDocument);
            args = innerArgs;
        }
        return result;
    }
    throw new IllegalArgumentException("Translation of " + call + " is not supported by MongoProject");
}
Also used : RexToLixTranslator(org.apache.calcite.adapter.enumerable.RexToLixTranslator) JavaTypeFactory(org.apache.calcite.adapter.java.JavaTypeFactory) Arrays(java.util.Arrays) SqlKind(org.apache.calcite.sql.SqlKind) SqlTypeName(org.apache.calcite.sql.type.SqlTypeName) BsonNull(org.bson.BsonNull) RexVisitorImpl(org.apache.calcite.rex.RexVisitorImpl) RexLiteral(org.apache.calcite.rex.RexLiteral) RexImpTable(org.apache.calcite.adapter.enumerable.RexImpTable) BsonString(org.bson.BsonString) Collectors(java.util.stream.Collectors) BsonDocument(org.bson.BsonDocument) BsonValue(org.bson.BsonValue) RexInputRef(org.apache.calcite.rex.RexInputRef) MongoOp(org.apache.drill.exec.store.mongo.common.MongoOp) List(java.util.List) ImmutableMap(org.apache.drill.shaded.guava.com.google.common.collect.ImmutableMap) SqlStdOperatorTable(org.apache.calcite.sql.fun.SqlStdOperatorTable) RexNode(org.apache.calcite.rex.RexNode) BsonArray(org.bson.BsonArray) Map(java.util.Map) BsonInt32(org.bson.BsonInt32) SqlOperator(org.apache.calcite.sql.SqlOperator) RexCall(org.apache.calcite.rex.RexCall) RexLiteral(org.apache.calcite.rex.RexLiteral) BsonInt32(org.bson.BsonInt32) BsonDocument(org.bson.BsonDocument) BsonString(org.bson.BsonString) BsonArray(org.bson.BsonArray) BsonString(org.bson.BsonString) BsonValue(org.bson.BsonValue) RexNode(org.apache.calcite.rex.RexNode)

Example 60 with RexCall

use of org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.rex.RexCall in project drill by apache.

the class DrillMergeProjectRule method simplifyCast.

public static List<RexNode> simplifyCast(List<RexNode> projectExprs) {
    final List<RexNode> list = new ArrayList<>();
    for (RexNode rex : projectExprs) {
        if (rex.getKind() == SqlKind.CAST) {
            RexNode operand = ((RexCall) rex).getOperands().get(0);
            while (operand.getKind() == SqlKind.CAST && operand.getType().equals(rex.getType())) {
                rex = operand;
                operand = ((RexCall) rex).getOperands().get(0);
            }
        }
        list.add(rex);
    }
    return list;
}
Also used : RexCall(org.apache.calcite.rex.RexCall) ArrayList(java.util.ArrayList) RexNode(org.apache.calcite.rex.RexNode)

Aggregations

RexCall (org.apache.calcite.rex.RexCall)213 RexNode (org.apache.calcite.rex.RexNode)172 RexInputRef (org.apache.calcite.rex.RexInputRef)61 ArrayList (java.util.ArrayList)60 RexLiteral (org.apache.calcite.rex.RexLiteral)44 Nullable (javax.annotation.Nullable)35 RelNode (org.apache.calcite.rel.RelNode)26 RelDataType (org.apache.calcite.rel.type.RelDataType)24 SqlOperator (org.apache.calcite.sql.SqlOperator)23 List (java.util.List)22 RexBuilder (org.apache.calcite.rex.RexBuilder)22 DruidExpression (org.apache.druid.sql.calcite.expression.DruidExpression)19 SqlKind (org.apache.calcite.sql.SqlKind)14 ImmutableBitSet (org.apache.calcite.util.ImmutableBitSet)14 RelOptUtil (org.apache.calcite.plan.RelOptUtil)11 PostAggregator (org.apache.druid.query.aggregation.PostAggregator)11 RexCall (org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.rex.RexCall)10 RexNode (org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.rex.RexNode)10 RexTableInputRef (org.apache.calcite.rex.RexTableInputRef)10 RelDataTypeField (org.apache.calcite.rel.type.RelDataTypeField)9