Examples with DrillLimitRelBase org.apache.drill.exec.planner.common.DrillLimitRelBase used on opensource projects

Example 1 with DrillLimitRelBase

use of org.apache.drill.exec.planner.common.DrillLimitRelBase in project drill by apache.

the class JdbcLimitRule method convert.

@Override
public RelNode convert(RelNode rel) {
    DrillLimitRelBase limit = (DrillLimitRelBase) rel;
    if (limit.getOffset() == null || !limit.getTraitSet().contains(RelCollations.EMPTY) || !(convention.getPlugin().getDialect() instanceof MssqlSqlDialect)) {
        return super.convert(limit);
    } else {
        // MS SQL doesn't support either OFFSET or FETCH without ORDER BY.
        // But for the case of FETCH without OFFSET, Calcite generates TOP N
        // instead of FETCH, and it is supported by MS SQL.
        // So do splitting the limit with both OFFSET and FETCH but without ORDER BY
        int offset = Math.max(0, RexLiteral.intValue(limit.getOffset()));
        int fetch = Math.max(0, RexLiteral.intValue(limit.getFetch()));
        // child Limit uses conservative approach: use offset 0 and fetch = parent limit offset + parent limit fetch.
        RexNode childFetch = limit.getCluster().getRexBuilder().makeExactLiteral(BigDecimal.valueOf(offset + fetch));
        RelNode jdbcSort = new DrillJdbcSort(limit.getCluster(), limit.getTraitSet().plus(RelCollations.EMPTY).replace(this.out).simplify(), convert(limit.getInput(), limit.getInput().getTraitSet().replace(this.out).simplify()), RelCollations.EMPTY, null, childFetch);
        return limit.copy(limit.getTraitSet(), Collections.singletonList(jdbcSort), true);
    }
}

Example 2 with DrillLimitRelBase

use of org.apache.drill.exec.planner.common.DrillLimitRelBase in project drill by apache.

the class DrillCassandraLimitRule method convert.

public RelNode convert(RelNode relNode) {
    DrillLimitRelBase limit = (DrillLimitRelBase) relNode;
    final RelTraitSet traitSet = limit.getTraitSet().replace(CassandraRel.CONVENTION);
    return new CassandraLimit(limit.getCluster(), traitSet, convert(limit.getInput(), CassandraRel.CONVENTION), limit.getOffset(), limit.getFetch());
}

Example 3 with DrillLimitRelBase

use of org.apache.drill.exec.planner.common.DrillLimitRelBase in project drill by apache.

the class PluginConverterRule method matches.

@Override
public boolean matches(RelOptRuleCall call) {
    RelNode rel = call.rel(0);
    boolean canImplement = false;
    // cannot use visitor pattern here, since RelShuttle supports only logical rel implementations
    if (rel instanceof Aggregate) {
        canImplement = pluginImplementor.canImplement(((Aggregate) rel));
    } else if (rel instanceof Filter) {
        canImplement = pluginImplementor.canImplement(((Filter) rel));
    } else if (rel instanceof DrillLimitRelBase) {
        canImplement = pluginImplementor.canImplement(((DrillLimitRelBase) rel));
    } else if (rel instanceof Project) {
        canImplement = pluginImplementor.canImplement(((Project) rel));
    } else if (rel instanceof Sort) {
        canImplement = pluginImplementor.canImplement(((Sort) rel));
    } else if (rel instanceof Union) {
        canImplement = pluginImplementor.canImplement(((Union) rel));
    } else if (rel instanceof Join) {
        canImplement = pluginImplementor.canImplement(((Join) rel));
    }
    return canImplement && super.matches(call);
}

Example 4 with DrillLimitRelBase

use of org.apache.drill.exec.planner.common.DrillLimitRelBase in project drill by apache.

the class PluginLimitRule method convert.

@Override
public RelNode convert(RelNode rel) {
    DrillLimitRelBase limit = (DrillLimitRelBase) rel;
    PluginLimitRel pluginLimitRel = getPluginLimitRel(limit);
    if (getPluginImplementor().artificialLimit()) {
        // preserve original limit
        return limit.copy(limit.getTraitSet(), Collections.singletonList(pluginLimitRel));
    } else {
        return pluginLimitRel;
    }
}