return correlate.copy(correlate.getTraitSet(), Arrays.asList(left, right)); Correlate newCorrelate = correlate.copy(correlate.getTraitSet(), leftInputs.get(correlate.getCorrelationId()), right, updatedCorrelationIds.get(correlate.getCorrelationId()),
@Override public RelNode visit(LogicalCorrelate correlate) { final RelNode left = correlate.getLeft().accept(this); final RelNode right = correlate.getRight().accept(this); return new LogicalCorrelate( cluster, copyOf(correlate.getTraitSet()), left, right, correlate.getCorrelationId(), correlate.getRequiredColumns(), correlate.getJoinType() ); }
public RelNode convert(RelNode rel) { final LogicalCorrelate c = (LogicalCorrelate) rel; final RelTraitSet traitSet = c.getTraitSet().replace(EnumerableConvention.INSTANCE); return new EnumerableCorrelate( rel.getCluster(), traitSet, convert(c.getLeft(), c.getLeft().getTraitSet() .replace(EnumerableConvention.INSTANCE)), convert(c.getRight(), c.getRight().getTraitSet() .replace(EnumerableConvention.INSTANCE)), c.getCorrelationId(), c.getRequiredColumns(), c.getJoinType()); } }
public RelNode convert(RelNode rel) { final LogicalCorrelate c = (LogicalCorrelate) rel; final RelTraitSet traitSet = c.getTraitSet().replace(EnumerableConvention.INSTANCE); return new EnumerableCorrelate( rel.getCluster(), traitSet, convert(c.getLeft(), c.getLeft().getTraitSet() .replace(EnumerableConvention.INSTANCE)), convert(c.getRight(), c.getRight().getTraitSet() .replace(EnumerableConvention.INSTANCE)), c.getCorrelationId(), c.getRequiredColumns(), c.getJoinType()); } }
@Override public void onMatch(RelOptRuleCall call) { final LogicalCorrelate correlate = call.rel(0); final RelNode left = correlate.getLeft(); final RelNode right = correlate.getRight(); final RelNode convertedLeft = convert(left, left.getTraitSet().plus(DrillRel.DRILL_LOGICAL).simplify()); final RelNode convertedRight = convert(right, right.getTraitSet().plus(DrillRel.DRILL_LOGICAL).simplify()); final RelTraitSet traits = correlate.getTraitSet().plus(DrillRel.DRILL_LOGICAL); DrillLateralJoinRel lateralJoinRel = new DrillLateralJoinRel(correlate.getCluster(), traits, convertedLeft, convertedRight, false, correlate.getCorrelationId(), correlate.getRequiredColumns(), correlate.getJoinType()); call.transformTo(lateralJoinRel); } }