All Downloads are FREE. Search and download functionalities are using the official Maven repository.
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.
com.hazelcast.sql.impl.calcite.opt.OptUtils Maven / Gradle / Ivy
/*
* Copyright (c) 2008-2021, Hazelcast, Inc. All Rights Reserved.
*
* Licensed 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.sql.impl.calcite.opt;
import com.hazelcast.sql.impl.calcite.opt.distribution.DistributionTrait;
import com.hazelcast.sql.impl.calcite.opt.distribution.DistributionTraitDef;
import com.hazelcast.sql.impl.calcite.schema.HazelcastRelOptTable;
import com.hazelcast.sql.impl.calcite.schema.HazelcastTable;
import com.hazelcast.org.apache.calcite.plan.Convention;
import com.hazelcast.org.apache.calcite.plan.ConventionTraitDef;
import com.hazelcast.org.apache.calcite.plan.HazelcastRelOptCluster;
import com.hazelcast.org.apache.calcite.plan.RelOptRule;
import com.hazelcast.org.apache.calcite.plan.RelOptRuleOperand;
import com.hazelcast.org.apache.calcite.plan.RelTrait;
import com.hazelcast.org.apache.calcite.plan.RelTraitSet;
import com.hazelcast.org.apache.calcite.plan.volcano.HazelcastRelSubsetUtil;
import com.hazelcast.org.apache.calcite.prepare.RelOptTableImpl;
import com.hazelcast.org.apache.calcite.rel.RelNode;
import com.hazelcast.org.apache.calcite.rel.core.TableScan;
import com.hazelcast.org.apache.calcite.rel.logical.LogicalTableScan;
import com.hazelcast.org.apache.calcite.rel.type.RelDataTypeFactory;
import java.util.Collection;
import java.util.Collections;
import java.util.HashSet;
import java.util.IdentityHashMap;
import java.util.Set;
import static com.hazelcast.org.apache.calcite.plan.RelOptRule.convert;
/**
* Utility methods for rules.
*/
public final class OptUtils {
private OptUtils() {
// No-op.
}
/**
* Get operand matching a single node.
*
* @param cls Node class.
* @param convention Convention.
* @return Operand.
*/
public static RelOptRuleOperand single(Class cls, Convention convention) {
return RelOptRule.operand(cls, convention, RelOptRule.any());
}
/**
* Get operand matching a node with specific child node.
*
* @param cls Node class.
* @param childCls Child node class.
* @param convention Convention.
* @return Operand.
*/
public static RelOptRuleOperand parentChild(Class cls,
Class childCls, Convention convention) {
RelOptRuleOperand childOperand = RelOptRule.operand(childCls, RelOptRule.any());
return RelOptRule.operand(cls, convention, RelOptRule.some(childOperand));
}
/**
* Add a single trait to the trait set.
*
* @param traitSet Original trait set.
* @param trait Trait to add.
* @return Resulting trait set.
*/
public static RelTraitSet traitPlus(RelTraitSet traitSet, RelTrait trait) {
return traitSet.plus(trait).simplify();
}
/**
* Add two traits to the trait set.
*
* @param traitSet Original trait set.
* @param trait1 Trait to add.
* @param trait2 Trait to add.
* @return Resulting trait set.
*/
public static RelTraitSet traitPlus(RelTraitSet traitSet, RelTrait trait1, RelTrait trait2) {
return traitSet.plus(trait1).plus(trait2).simplify();
}
/**
* Convert the given trait set to logical convention.
*
* @param traitSet Original trait set.
* @return New trait set with logical convention.
*/
public static RelTraitSet toLogicalConvention(RelTraitSet traitSet) {
return traitPlus(traitSet, HazelcastConventions.LOGICAL);
}
/**
* Convert the given input into logical input.
*
* @param input Original input.
* @return Logical input.
*/
public static RelNode toLogicalInput(RelNode input) {
return convert(input, toLogicalConvention(input.getTraitSet()));
}
/**
* Convert the given trait set to physical convention.
*
* @param traitSet Original trait set.
* @return New trait set with physical convention and provided distribution.
*/
public static RelTraitSet toPhysicalConvention(RelTraitSet traitSet) {
return traitPlus(traitSet, HazelcastConventions.PHYSICAL);
}
/**
* Convert the given trait set to physical convention.
*
* @param traitSet Original trait set.
* @param distribution Distribution.
* @return New trait set with physical convention and provided distribution.
*/
public static RelTraitSet toPhysicalConvention(RelTraitSet traitSet, DistributionTrait distribution) {
return traitPlus(traitSet, HazelcastConventions.PHYSICAL, distribution);
}
/**
* Convert the given input into physical input.
*
* @param input Original input.
* @return Logical input.
*/
public static RelNode toPhysicalInput(RelNode input) {
return convert(input, toPhysicalConvention(input.getTraitSet()));
}
/**
* Convert the given input into physical input.
*
* @param input Original input.
* @param distribution Distribution.
* @return Logical input.
*/
public static RelNode toPhysicalInput(RelNode input, DistributionTrait distribution) {
return convert(input, toPhysicalConvention(input.getTraitSet(), distribution));
}
/**
* @param rel Node.
* @return {@code true} if the given node is physical node.
*/
public static boolean isPhysical(RelNode rel) {
return rel.getTraitSet().getTrait(ConventionTraitDef.INSTANCE).equals(HazelcastConventions.PHYSICAL);
}
/**
* Get possible physical rels from the given subset. Every returned input is guaranteed to have a unique trait set.
*
* @param input Subset.
* @return Physical rels.
*/
public static Collection getPhysicalRelsFromSubset(RelNode input) {
Set traitSets = new HashSet<>();
Set res = Collections.newSetFromMap(new IdentityHashMap<>());
for (RelNode rel : HazelcastRelSubsetUtil.getSubsets(input)) {
if (!isPhysical(rel)) {
continue;
}
if (traitSets.add(rel.getTraitSet())) {
res.add(rel);
}
}
return res;
}
public static boolean isHazelcastTable(TableScan scan) {
HazelcastTable table = scan.getTable().unwrap(HazelcastTable.class);
return table != null;
}
public static HazelcastTable getHazelcastTable(TableScan scan) {
HazelcastTable table = scan.getTable().unwrap(HazelcastTable.class);
assert table != null;
return table;
}
public static HazelcastRelOptCluster getCluster(RelNode rel) {
assert rel.getCluster() instanceof HazelcastRelOptCluster;
return (HazelcastRelOptCluster) rel.getCluster();
}
public static DistributionTraitDef getDistributionDef(RelNode rel) {
return getCluster(rel).getDistributionTraitDef();
}
public static DistributionTrait getDistribution(RelNode rel) {
return rel.getTraitSet().getTrait(getDistributionDef(rel));
}
public static HazelcastRelOptTable createRelTable(
HazelcastRelOptTable originalRelTable,
HazelcastTable newHazelcastTable,
RelDataTypeFactory typeFactory
) {
RelOptTableImpl newTable = RelOptTableImpl.create(
originalRelTable.getRelOptSchema(),
newHazelcastTable.getRowType(typeFactory),
originalRelTable.getDelegate().getQualifiedName(),
newHazelcastTable,
null
);
return new HazelcastRelOptTable(newTable);
}
public static LogicalTableScan createLogicalScanWithNewTable(
TableScan originalScan,
HazelcastTable newHazelcastTable
) {
HazelcastRelOptTable originalRelTable = (HazelcastRelOptTable) originalScan.getTable();
HazelcastRelOptTable newTable = createRelTable(
originalRelTable,
newHazelcastTable,
originalScan.getCluster().getTypeFactory()
);
return LogicalTableScan.create(
originalScan.getCluster(),
newTable,
originalScan.getHints()
);
}
}
