com.hazelcast.org.apache.calcite.adapter.enumerable.EnumerableRepeatUnion 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.function.Experimental;
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.RelNode;
import com.hazelcast.org.apache.calcite.rel.core.RepeatUnion;
import com.hazelcast.org.apache.calcite.util.BuiltInMethod;
import com.hazelcast.org.apache.calcite.util.Util;
import java.util.List;
/**
* Implementation of {@link RepeatUnion} in
* {@link EnumerableConvention enumerable calling convention}.
*
* NOTE: The current API is experimental and subject to change without
* notice.
*/
@Experimental
public class EnumerableRepeatUnion extends RepeatUnion implements EnumerableRel {
/**
* Creates an EnumerableRepeatUnion.
*/
EnumerableRepeatUnion(RelOptCluster cluster, RelTraitSet traitSet,
RelNode seed, RelNode iterative, boolean all, int iterationLimit) {
super(cluster, traitSet, seed, iterative, all, iterationLimit);
}
@Override public EnumerableRepeatUnion copy(RelTraitSet traitSet, List inputs) {
assert inputs.size() == 2;
return new EnumerableRepeatUnion(getCluster(), traitSet,
inputs.get(0), inputs.get(1), all, iterationLimit);
}
@Override public Result implement(EnumerableRelImplementor implementor, Prefer pref) {
// return repeatUnion(, , iterationLimit, all, );
BlockBuilder builder = new BlockBuilder();
RelNode seed = getSeedRel();
RelNode iteration = getIterativeRel();
Result seedResult = implementor.visitChild(this, 0, (EnumerableRel) seed, pref);
Result iterationResult = implementor.visitChild(this, 1, (EnumerableRel) iteration, pref);
Expression seedExp = builder.append("seed", seedResult.block);
Expression iterativeExp = builder.append("iteration", iterationResult.block);
PhysType physType = PhysTypeImpl.of(
implementor.getTypeFactory(),
getRowType(),
pref.prefer(seedResult.format));
Expression unionExp = Expressions.call(
BuiltInMethod.REPEAT_UNION.method,
seedExp,
iterativeExp,
Expressions.constant(iterationLimit, int.class),
Expressions.constant(all, boolean.class),
Util.first(physType.com.hazelcast.com.arer(), Expressions.call(BuiltInMethod.IDENTITY_COMPARER.method)));
builder.add(unionExp);
return implementor.result(physType, builder.toBlock());
}
}