com.hazelcast.org.apache.calcite.adapter.enumerable.EnumerableCollect 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 compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.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.RelNode;
import com.hazelcast.org.apache.calcite.rel.core.Collect;
import com.hazelcast.org.apache.calcite.rel.type.RelDataType;
import com.hazelcast.org.apache.calcite.sql.type.SqlTypeName;
import com.hazelcast.org.apache.calcite.util.BuiltInMethod;
/** Implementation of {@link com.hazelcast.org.apache.calcite.rel.core.Collect} in
* {@link com.hazelcast.org.apache.calcite.adapter.enumerable.EnumerableConvention enumerable calling convention}. */
public class EnumerableCollect extends Collect implements EnumerableRel {
/**
* Creates an EnumerableCollect.
*
* Use {@link #create} unless you know what you're doing.
*
* @param cluster Cluster
* @param traitSet Trait set
* @param input Input relational expression
* @param rowType Row type
*/
public EnumerableCollect(RelOptCluster cluster, RelTraitSet traitSet,
RelNode input, RelDataType rowType) {
super(cluster, traitSet, input, rowType);
assert getConvention() instanceof EnumerableConvention;
assert getConvention() == input.getConvention();
}
@Deprecated // to be removed before 2.0
public EnumerableCollect(RelOptCluster cluster, RelTraitSet traitSet,
RelNode input, String fieldName) {
this(cluster, traitSet, input,
deriveRowType(cluster.getTypeFactory(), SqlTypeName.MULTISET, fieldName,
input.getRowType()));
}
/**
* Creates an EnumerableCollect.
*
* @param input Input relational expression
* @param rowType Row type
*/
public static Collect create(RelNode input, RelDataType rowType) {
final RelOptCluster cluster = input.getCluster();
final RelTraitSet traitSet =
cluster.traitSet().replace(EnumerableConvention.INSTANCE);
return new EnumerableCollect(cluster, traitSet, input, rowType);
}
@Override public EnumerableCollect copy(RelTraitSet traitSet,
RelNode newInput) {
return new EnumerableCollect(getCluster(), traitSet, newInput, rowType());
}
@Override public Result implement(EnumerableRelImplementor implementor, Prefer pref) {
final BlockBuilder builder = new BlockBuilder();
final EnumerableRel child = (EnumerableRel) getInput();
// REVIEW zabetak January 7, 2019: Even if we ask the implementor to provide a result
// where records are represented as arrays (Prefer.ARRAY) this may not be respected.
final Result result = implementor.visitChild(this, 0, child, Prefer.ARRAY);
final PhysType physType =
PhysTypeImpl.of(
implementor.getTypeFactory(),
getRowType(),
JavaRowFormat.LIST);
// final Enumerable child = <>;
// final Enumerable