Examples with RelSubset org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.plan.volcano.RelSubset used on opensource projects

Example 16 with RelSubset

use of org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.plan.volcano.RelSubset in project drill by apache.

the class SubsetTransformer method go.

public boolean go(T n, RelNode candidateSet) throws E {
    if (!(candidateSet instanceof RelSubset)) {
        return false;
    }
    boolean transform = false;
    Set<RelNode> transformedRels = Sets.newIdentityHashSet();
    Set<RelTraitSet> traitSets = Sets.newHashSet();
    // 1, get all the target traitsets from candidateSet's rel list,
    for (RelNode rel : ((RelSubset) candidateSet).getRelList()) {
        if (isPhysical(rel)) {
            final RelTraitSet relTraitSet = rel.getTraitSet();
            if (!traitSets.contains(relTraitSet)) {
                traitSets.add(relTraitSet);
                logger.trace("{}.convertChild get traitSet {}", this.getClass().getSimpleName(), relTraitSet);
            }
        }
    }
    // 2, convert the candidateSet to targeted taitSets
    if (traitSets.size() == 0 && forceConvert()) {
        RelNode out = convertChild(n, null);
        if (out != null) {
            call.transformTo(out);
            return true;
        }
        return false;
    }
    for (RelTraitSet traitSet : traitSets) {
        RelNode newRel = RelOptRule.convert(candidateSet, traitSet.simplify());
        if (transformedRels.contains(newRel)) {
            continue;
        }
        transformedRels.add(newRel);
        logger.trace("{}.convertChild to convert NODE {} ,AND {}", this.getClass().getSimpleName(), n, newRel);
        RelNode out = convertChild(n, newRel);
        // RelNode out = convertChild(n, rel);
        if (out != null) {
            call.transformTo(out);
            transform = true;
        }
    }
    return transform;
}

Example 17 with RelSubset

use of org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.plan.volcano.RelSubset in project drill by apache.

the class RuntimeFilterVisitor method findLeftScanPrel.

/**
 * Find a join condition's left input source scan Prel. If we can't find a target scan Prel then this
 * RuntimeFilter can not pushed down to a probe side scan Prel.
 *
 * @param fieldName   left join condition field Name
 * @param leftRelNode left RelNode of a BiRel or the SingleRel
 * @return a left scan Prel which contains the left join condition name or null
 */
private ScanPrel findLeftScanPrel(String fieldName, RelNode leftRelNode) {
    if (leftRelNode instanceof ScanPrel) {
        RelDataType scanRowType = leftRelNode.getRowType();
        RelDataTypeField field = scanRowType.getField(fieldName, true, true);
        if (field != null) {
            // found
            return (ScanPrel) leftRelNode;
        } else {
            return null;
        }
    } else if (leftRelNode instanceof RelSubset) {
        RelNode bestNode = ((RelSubset) leftRelNode).getBest();
        if (bestNode != null) {
            return findLeftScanPrel(fieldName, bestNode);
        } else {
            return null;
        }
    } else {
        List<RelNode> relNodes = leftRelNode.getInputs();
        RelNode leftNode = relNodes.get(0);
        return findLeftScanPrel(fieldName, leftNode);
    }
}