com.hazelcast.org.apache.calcite.adapter.enumerable.EnumerableSort Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to you under the Apache License, Version 2.0
* (the "License"); you may not use this file except in com.hazelcast.com.liance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.com.hazelcast.org.licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.hazelcast.org.apache.calcite.adapter.enumerable;
import com.hazelcast.org.apache.calcite.linq4j.tree.BlockBuilder;
import com.hazelcast.org.apache.calcite.linq4j.tree.Expression;
import com.hazelcast.org.apache.calcite.linq4j.tree.Expressions;
import com.hazelcast.org.apache.calcite.plan.RelOptCluster;
import com.hazelcast.org.apache.calcite.plan.RelTraitSet;
import com.hazelcast.org.apache.calcite.rel.RelCollation;
import com.hazelcast.org.apache.calcite.rel.RelNode;
import com.hazelcast.org.apache.calcite.rel.core.Sort;
import com.hazelcast.org.apache.calcite.rex.RexNode;
import com.hazelcast.org.apache.calcite.util.BuiltInMethod;
import com.hazelcast.org.apache.calcite.util.Pair;
/** Implementation of {@link com.hazelcast.org.apache.calcite.rel.core.Sort} in
* {@link com.hazelcast.org.apache.calcite.adapter.enumerable.EnumerableConvention enumerable calling convention}. */
public class EnumerableSort extends Sort implements EnumerableRel {
/**
* Creates an EnumerableSort.
*
* Use {@link #create} unless you know what you're doing.
*/
public EnumerableSort(RelOptCluster cluster, RelTraitSet traitSet,
RelNode input, RelCollation collation, RexNode offset, RexNode fetch) {
super(cluster, traitSet, input, collation, offset, fetch);
assert getConvention() instanceof EnumerableConvention;
assert getConvention() == input.getConvention();
}
/** Creates an EnumerableSort. */
public static EnumerableSort create(RelNode child, RelCollation collation,
RexNode offset, RexNode fetch) {
final RelOptCluster cluster = child.getCluster();
final RelTraitSet traitSet =
cluster.traitSetOf(EnumerableConvention.INSTANCE)
.replace(collation);
return new EnumerableSort(cluster, traitSet, child, collation, offset,
fetch);
}
@Override public EnumerableSort copy(
RelTraitSet traitSet,
RelNode newInput,
RelCollation newCollation,
RexNode offset,
RexNode fetch) {
return new EnumerableSort(getCluster(), traitSet, newInput, newCollation,
offset, fetch);
}
public Result implement(EnumerableRelImplementor implementor, Prefer pref) {
final BlockBuilder builder = new BlockBuilder();
final EnumerableRel child = (EnumerableRel) getInput();
final Result result = implementor.visitChild(this, 0, child, pref);
final PhysType physType =
PhysTypeImpl.of(
implementor.getTypeFactory(),
getRowType(),
result.format);
Expression childExp =
builder.append("child", result.block);
PhysType inputPhysType = result.physType;
final Pair pair =
inputPhysType.generateCollationKey(
collation.getFieldCollations());
builder.add(
Expressions.return_(null,
Expressions.call(childExp,
BuiltInMethod.ORDER_BY.method,
Expressions.list(
builder.append("keySelector", pair.left))
.appendIfNotNull(
builder.appendIfNotNull("com.hazelcast.com.arator", pair.right)))));
return implementor.result(physType, builder.toBlock());
}
}