com.hazelcast.org.apache.calcite.rel.metadata.RelMdDistribution 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.rel.metadata;
import com.hazelcast.org.apache.calcite.plan.RelOptTable;
import com.hazelcast.org.apache.calcite.rel.BiRel;
import com.hazelcast.org.apache.calcite.rel.RelDistribution;
import com.hazelcast.org.apache.calcite.rel.RelDistributions;
import com.hazelcast.org.apache.calcite.rel.RelNode;
import com.hazelcast.org.apache.calcite.rel.SingleRel;
import com.hazelcast.org.apache.calcite.rel.core.Exchange;
import com.hazelcast.org.apache.calcite.rel.core.Filter;
import com.hazelcast.org.apache.calcite.rel.core.Project;
import com.hazelcast.org.apache.calcite.rel.core.SetOp;
import com.hazelcast.org.apache.calcite.rel.core.Snapshot;
import com.hazelcast.org.apache.calcite.rel.core.Sort;
import com.hazelcast.org.apache.calcite.rel.core.TableModify;
import com.hazelcast.org.apache.calcite.rel.core.TableScan;
import com.hazelcast.org.apache.calcite.rel.core.Values;
import com.hazelcast.org.apache.calcite.rel.type.RelDataType;
import com.hazelcast.org.apache.calcite.rex.RexLiteral;
import com.hazelcast.org.apache.calcite.rex.RexNode;
import com.hazelcast.org.apache.calcite.rex.RexProgram;
import com.hazelcast.org.apache.calcite.util.mapping.Mappings;
import com.hazelcast.com.google.common.collect.ImmutableList;
import com.hazelcast.org.checkerframework.checker.nullness.qual.Nullable;
import java.util.List;
/**
* RelMdCollation supplies a default implementation of
* {@link RelMetadataQuery#distribution}
* for the standard logical algebra.
*/
public class RelMdDistribution
implements MetadataHandler {
public static final RelMetadataProvider SOURCE =
ReflectiveRelMetadataProvider.reflectiveSource(
new RelMdDistribution(), BuiltInMetadata.Distribution.Handler.class);
//~ Constructors -----------------------------------------------------------
private RelMdDistribution() {}
//~ Methods ----------------------------------------------------------------
@Override public MetadataDef getDef() {
return BuiltInMetadata.Distribution.DEF;
}
/** Fallback method to deduce distribution for any relational expression not
* handled by a more specific method.
*
* @param rel Relational expression
* @return Relational expression's distribution
*/
public RelDistribution distribution(RelNode rel, RelMetadataQuery mq) {
return RelDistributions.SINGLETON;
}
public RelDistribution distribution(SingleRel rel, RelMetadataQuery mq) {
return mq.distribution(rel.getInput());
}
public RelDistribution distribution(BiRel rel, RelMetadataQuery mq) {
return mq.distribution(rel.getLeft());
}
public RelDistribution distribution(SetOp rel, RelMetadataQuery mq) {
return mq.distribution(rel.getInputs().get(0));
}
public RelDistribution distribution(TableModify rel, RelMetadataQuery mq) {
return mq.distribution(rel.getInput());
}
public @Nullable RelDistribution distribution(TableScan scan, RelMetadataQuery mq) {
return table(scan.getTable());
}
public RelDistribution distribution(Project project, RelMetadataQuery mq) {
return project(mq, project.getInput(), project.getProjects());
}
public RelDistribution distribution(Values values, RelMetadataQuery mq) {
return values(values.getRowType(), values.getTuples());
}
public RelDistribution distribution(Exchange exchange, RelMetadataQuery mq) {
return exchange(exchange.distribution);
}
// Helper methods
/** Helper method to determine a
* {@link TableScan}'s distribution. */
public static @Nullable RelDistribution table(RelOptTable table) {
return table.getDistribution();
}
/** Helper method to determine a
* {@link Snapshot}'s distribution. */
public static RelDistribution snapshot(RelMetadataQuery mq, RelNode input) {
return mq.distribution(input);
}
/** Helper method to determine a
* {@link Sort}'s distribution. */
public static RelDistribution sort(RelMetadataQuery mq, RelNode input) {
return mq.distribution(input);
}
/** Helper method to determine a
* {@link Filter}'s distribution. */
public static RelDistribution filter(RelMetadataQuery mq, RelNode input) {
return mq.distribution(input);
}
/** Helper method to determine a
* limit's distribution. */
public static RelDistribution limit(RelMetadataQuery mq, RelNode input) {
return mq.distribution(input);
}
/** Helper method to determine a
* {@link com.hazelcast.org.apache.calcite.rel.core.Calc}'s distribution. */
public static RelDistribution calc(RelMetadataQuery mq, RelNode input,
RexProgram program) {
assert program.getCondition() != null || !program.getProjectList().isEmpty();
final RelDistribution inputDistribution = mq.distribution(input);
if (!program.getProjectList().isEmpty()) {
final Mappings.TargetMapping mapping = program.getPartialMapping(
input.getRowType().getFieldCount());
return inputDistribution.apply(mapping);
}
return inputDistribution;
}
/** Helper method to determine a {@link Project}'s distribution. */
public static RelDistribution project(RelMetadataQuery mq, RelNode input,
List projects) {
final RelDistribution inputDistribution = mq.distribution(input);
final Mappings.TargetMapping mapping =
Project.getPartialMapping(input.getRowType().getFieldCount(),
projects);
return inputDistribution.apply(mapping);
}
/** Helper method to determine a
* {@link Values}'s distribution. */
public static RelDistribution values(RelDataType rowType,
ImmutableList> tuples) {
return RelDistributions.BROADCAST_DISTRIBUTED;
}
/** Helper method to determine an
* {@link Exchange}'s
* or {@link com.hazelcast.org.apache.calcite.rel.core.SortExchange}'s distribution. */
public static RelDistribution exchange(RelDistribution distribution) {
return distribution;
}
}