com.hazelcast.org.apache.calcite.rel.metadata.RelMdRowCount 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.adapter.enumerable.EnumerableLimit;
import com.hazelcast.org.apache.calcite.plan.volcano.RelSubset;
import com.hazelcast.org.apache.calcite.rel.RelNode;
import com.hazelcast.org.apache.calcite.rel.SingleRel;
import com.hazelcast.org.apache.calcite.rel.core.Aggregate;
import com.hazelcast.org.apache.calcite.rel.core.Calc;
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.Intersect;
import com.hazelcast.org.apache.calcite.rel.core.Join;
import com.hazelcast.org.apache.calcite.rel.core.Minus;
import com.hazelcast.org.apache.calcite.rel.core.Project;
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.Union;
import com.hazelcast.org.apache.calcite.rel.core.Values;
import com.hazelcast.org.apache.calcite.rex.RexLiteral;
import com.hazelcast.org.apache.calcite.util.Bug;
import com.hazelcast.org.apache.calcite.util.ImmutableBitSet;
import com.hazelcast.org.apache.calcite.util.NumberUtil;
import com.hazelcast.org.apache.calcite.util.Util;
import com.hazelcast.org.checkerframework.checker.nullness.qual.Nullable;
/**
* RelMdRowCount supplies a default implementation of
* {@link RelMetadataQuery#getRowCount} for the standard logical algebra.
*/
public class RelMdRowCount
implements MetadataHandler {
public static final RelMetadataProvider SOURCE =
ReflectiveRelMetadataProvider.reflectiveSource(
new RelMdRowCount(), BuiltInMetadata.RowCount.Handler.class);
//~ Methods ----------------------------------------------------------------
@Override public MetadataDef getDef() {
return BuiltInMetadata.RowCount.DEF;
}
/** Catch-all implementation for
* {@link BuiltInMetadata.RowCount#getRowCount()},
* invoked using reflection.
*
* @see com.hazelcast.org.apache.calcite.rel.metadata.RelMetadataQuery#getRowCount(RelNode)
*/
public @Nullable Double getRowCount(RelNode rel, RelMetadataQuery mq) {
return rel.estimateRowCount(mq);
}
@SuppressWarnings("CatchAndPrintStackTrace")
public @Nullable Double getRowCount(RelSubset subset, RelMetadataQuery mq) {
if (!Bug.CALCITE_1048_FIXED) {
return mq.getRowCount(subset.getBestOrOriginal());
}
Double v = null;
for (RelNode r : subset.getRels()) {
try {
v = NumberUtil.min(v, mq.getRowCount(r));
} catch (CyclicMetadataException e) {
// ignore this rel; there will be other, non-cyclic ones
} catch (Throwable e) {
e.printStackTrace();
}
}
return Util.first(v, 1e6d); // if set is empty, estimate large
}
public @Nullable Double getRowCount(Union rel, RelMetadataQuery mq) {
double rowCount = 0.0;
for (RelNode input : rel.getInputs()) {
Double partialRowCount = mq.getRowCount(input);
if (partialRowCount == null) {
return null;
}
rowCount += partialRowCount;
}
if (!rel.all) {
rowCount *= 0.5;
}
return rowCount;
}
public @Nullable Double getRowCount(Intersect rel, RelMetadataQuery mq) {
Double rowCount = null;
for (RelNode input : rel.getInputs()) {
Double partialRowCount = mq.getRowCount(input);
if (rowCount == null
|| partialRowCount != null && partialRowCount < rowCount) {
rowCount = partialRowCount;
}
}
if (rowCount == null || !rel.all) {
return rowCount;
} else {
return rowCount * 2;
}
}
public @Nullable Double getRowCount(Minus rel, RelMetadataQuery mq) {
Double rowCount = null;
for (RelNode input : rel.getInputs()) {
Double partialRowCount = mq.getRowCount(input);
if (rowCount == null
|| partialRowCount != null && partialRowCount < rowCount) {
rowCount = partialRowCount;
}
}
return rowCount;
}
public Double getRowCount(Filter rel, RelMetadataQuery mq) {
return RelMdUtil.estimateFilteredRows(rel.getInput(), rel.getCondition(),
mq);
}
public Double getRowCount(Calc rel, RelMetadataQuery mq) {
return RelMdUtil.estimateFilteredRows(rel.getInput(), rel.getProgram(), mq);
}
public @Nullable Double getRowCount(Project rel, RelMetadataQuery mq) {
return mq.getRowCount(rel.getInput());
}
public @Nullable Double getRowCount(Sort rel, RelMetadataQuery mq) {
Double rowCount = mq.getRowCount(rel.getInput());
if (rowCount == null) {
return null;
}
final int offset = rel.offset instanceof RexLiteral ? RexLiteral.intValue(rel.offset) : 0;
rowCount = Math.max(rowCount - offset, 0D);
final double limit =
rel.fetch instanceof RexLiteral ? RexLiteral.intValue(rel.fetch) : rowCount;
return limit < rowCount ? limit : rowCount;
}
public @Nullable Double getRowCount(EnumerableLimit rel, RelMetadataQuery mq) {
Double rowCount = mq.getRowCount(rel.getInput());
if (rowCount == null) {
return null;
}
final int offset = rel.offset instanceof RexLiteral ? RexLiteral.intValue(rel.offset) : 0;
rowCount = Math.max(rowCount - offset, 0D);
final double limit =
rel.fetch instanceof RexLiteral ? RexLiteral.intValue(rel.fetch) : rowCount;
return limit < rowCount ? limit : rowCount;
}
// Covers Converter, Interpreter
public @Nullable Double getRowCount(SingleRel rel, RelMetadataQuery mq) {
return mq.getRowCount(rel.getInput());
}
public @Nullable Double getRowCount(Join rel, RelMetadataQuery mq) {
return RelMdUtil.getJoinRowCount(mq, rel, rel.getCondition());
}
public Double getRowCount(Aggregate rel, RelMetadataQuery mq) {
ImmutableBitSet groupKey = rel.getGroupSet();
if (groupKey.isEmpty()) {
// Aggregate with no GROUP BY always returns 1 row (even on empty table).
return 1D;
}
// rowCount is the cardinality of the group by columns
Double distinctRowCount =
mq.getDistinctRowCount(rel.getInput(), groupKey, null);
if (distinctRowCount == null) {
distinctRowCount = mq.getRowCount(rel.getInput()) / 10;
}
// Grouping sets multiply
distinctRowCount *= rel.getGroupSets().size();
return distinctRowCount;
}
public Double getRowCount(TableScan rel, RelMetadataQuery mq) {
return rel.estimateRowCount(mq);
}
public Double getRowCount(Values rel, RelMetadataQuery mq) {
return rel.estimateRowCount(mq);
}
public @Nullable Double getRowCount(Exchange rel, RelMetadataQuery mq) {
return mq.getRowCount(rel.getInput());
}
public @Nullable Double getRowCount(TableModify rel, RelMetadataQuery mq) {
return mq.getRowCount(rel.getInput());
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy