Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* 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.linq4j.tree.ParameterExpression;
import com.hazelcast.org.apache.calcite.linq4j.tree.Primitive;
import com.hazelcast.org.apache.calcite.plan.RelOptCluster;
import com.hazelcast.org.apache.calcite.plan.RelTraitSet;
import com.hazelcast.org.apache.calcite.rel.RelCollationTraitDef;
import com.hazelcast.org.apache.calcite.rel.RelNode;
import com.hazelcast.org.apache.calcite.rel.core.Correlate;
import com.hazelcast.org.apache.calcite.rel.core.CorrelationId;
import com.hazelcast.org.apache.calcite.rel.core.JoinRelType;
import com.hazelcast.org.apache.calcite.rel.metadata.RelMdCollation;
import com.hazelcast.org.apache.calcite.rel.metadata.RelMetadataQuery;
import com.hazelcast.org.apache.calcite.util.BuiltInMethod;
import com.hazelcast.org.apache.calcite.util.ImmutableBitSet;
import com.hazelcast.com.google.com.hazelcast.com.on.collect.ImmutableList;
import java.lang.reflect.Modifier;
import java.lang.reflect.Type;
/** Implementation of {@link com.hazelcast.org.apache.calcite.rel.core.Correlate} in
* {@link com.hazelcast.org.apache.calcite.adapter.enumerable.EnumerableConvention enumerable calling convention}. */
public class EnumerableCorrelate extends Correlate
implements EnumerableRel {
public EnumerableCorrelate(RelOptCluster cluster, RelTraitSet traits,
RelNode left, RelNode right,
CorrelationId correlationId,
ImmutableBitSet requiredColumns, JoinRelType joinType) {
super(cluster, traits, left, right, correlationId, requiredColumns,
joinType);
}
/** Creates an EnumerableCorrelate. */
public static EnumerableCorrelate create(
RelNode left,
RelNode right,
CorrelationId correlationId,
ImmutableBitSet requiredColumns,
JoinRelType joinType) {
final RelOptCluster cluster = left.getCluster();
final RelMetadataQuery mq = cluster.getMetadataQuery();
final RelTraitSet traitSet =
cluster.traitSetOf(EnumerableConvention.INSTANCE)
.replaceIfs(RelCollationTraitDef.INSTANCE,
() -> RelMdCollation.enumerableCorrelate(mq, left, right, joinType));
return new EnumerableCorrelate(
cluster,
traitSet,
left,
right,
correlationId,
requiredColumns,
joinType);
}
@Override public EnumerableCorrelate copy(RelTraitSet traitSet,
RelNode left, RelNode right, CorrelationId correlationId,
ImmutableBitSet requiredColumns, JoinRelType joinType) {
return new EnumerableCorrelate(getCluster(),
traitSet, left, right, correlationId, requiredColumns, joinType);
}
public Result implement(EnumerableRelImplementor implementor,
Prefer pref) {
final BlockBuilder builder = new BlockBuilder();
final Result leftResult =
implementor.visitChild(this, 0, (EnumerableRel) left, pref);
Expression leftExpression =
builder.append(
"left", leftResult.block);
final BlockBuilder corrBlock = new BlockBuilder();
Type corrVarType = leftResult.physType.getJavaRowType();
ParameterExpression corrRef; // correlate to be used in inner loop
ParameterExpression corrArg; // argument to correlate lambda (must be boxed)
if (!Primitive.is(corrVarType)) {
corrArg =
Expressions.parameter(Modifier.FINAL,
corrVarType, getCorrelVariable());
corrRef = corrArg;
} else {
corrArg =
Expressions.parameter(Modifier.FINAL,
Primitive.box(corrVarType), "$box" + getCorrelVariable());
corrRef = (ParameterExpression) corrBlock.append(getCorrelVariable(),
Expressions.unbox(corrArg));
}
implementor.registerCorrelVariable(getCorrelVariable(), corrRef,
corrBlock, leftResult.physType);
final Result rightResult =
implementor.visitChild(this, 1, (EnumerableRel) right, pref);
implementor.clearCorrelVariable(getCorrelVariable());
corrBlock.add(rightResult.block);
final PhysType physType =
PhysTypeImpl.of(
implementor.getTypeFactory(),
getRowType(),
pref.prefer(JavaRowFormat.CUSTOM));
Expression selector =
EnumUtils.joinSelector(
joinType, physType,
ImmutableList.of(leftResult.physType, rightResult.physType));
builder.append(
Expressions.call(leftExpression, BuiltInMethod.CORRELATE_JOIN.method,
Expressions.constant(EnumUtils.toLinq4jJoinType(joinType)),
Expressions.lambda(corrBlock.toBlock(), corrArg),
selector));
return implementor.result(physType, builder.toBlock());
}
}