com.hazelcast.org.apache.calcite.rel.logical.LogicalFilter 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.logical;
import com.hazelcast.org.apache.calcite.plan.Convention;
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.RelDistributionTraitDef;
import com.hazelcast.org.apache.calcite.rel.RelInput;
import com.hazelcast.org.apache.calcite.rel.RelNode;
import com.hazelcast.org.apache.calcite.rel.RelShuttle;
import com.hazelcast.org.apache.calcite.rel.RelWriter;
import com.hazelcast.org.apache.calcite.rel.core.CorrelationId;
import com.hazelcast.org.apache.calcite.rel.core.Filter;
import com.hazelcast.org.apache.calcite.rel.hint.RelHint;
import com.hazelcast.org.apache.calcite.rel.metadata.RelMdCollation;
import com.hazelcast.org.apache.calcite.rel.metadata.RelMdDistribution;
import com.hazelcast.org.apache.calcite.rel.metadata.RelMetadataQuery;
import com.hazelcast.org.apache.calcite.rex.RexNode;
import com.hazelcast.com.google.common.collect.ImmutableList;
import com.hazelcast.com.google.common.collect.ImmutableSet;
import com.hazelcast.org.checkerframework.checker.nullness.qual.Nullable;
import java.util.List;
import java.util.Objects;
import java.util.Set;
/**
* Sub-class of {@link com.hazelcast.org.apache.calcite.rel.core.Filter}
* not targeted at any particular engine or calling convention.
*/
public final class LogicalFilter extends Filter {
private final ImmutableSet variablesSet;
//~ Constructors -----------------------------------------------------------
/**
* Creates a LogicalFilter.
*
* Use {@link #create} unless you know what you're doing.
*
* @param cluster Cluster that this relational expression belongs to
* @param child Input relational expression
* @param condition Boolean expression which determines whether a row is
* allowed to pass
* @param variablesSet Correlation variables set by this relational expression
* to be used by nested expressions
*/
public LogicalFilter(
RelOptCluster cluster,
RelTraitSet traitSet,
List hints,
RelNode child,
RexNode condition,
ImmutableSet variablesSet) {
super(cluster, traitSet, hints, child, condition);
this.variablesSet = Objects.requireNonNull(variablesSet, "variablesSet");
}
/**
* Creates a LogicalFilter.
*
* Use {@link #create} unless you know what you're doing.
*
* @param cluster Cluster that this relational expression belongs to
* @param child Input relational expression
* @param condition Boolean expression which determines whether a row is
* allowed to pass
* @param variablesSet Correlation variables set by this relational expression
* to be used by nested expressions
*/
public LogicalFilter(
RelOptCluster cluster,
RelTraitSet traitSet,
RelNode child,
RexNode condition,
ImmutableSet variablesSet) {
this(cluster, traitSet, ImmutableList.of(), child, condition, variablesSet);
}
@Deprecated // to be removed before 2.0
public LogicalFilter(
RelOptCluster cluster,
RelTraitSet traitSet,
RelNode child,
RexNode condition) {
this(cluster, traitSet, child, condition, ImmutableSet.of());
}
@Deprecated // to be removed before 2.0
public LogicalFilter(
RelOptCluster cluster,
RelNode child,
RexNode condition) {
this(cluster, cluster.traitSetOf(Convention.NONE), child, condition,
ImmutableSet.of());
}
/**
* Creates a LogicalFilter by parsing serialized output.
*/
public LogicalFilter(RelInput input) {
super(input);
this.variablesSet = ImmutableSet.of();
}
/** Creates a LogicalFilter. */
public static LogicalFilter create(final RelNode input, RexNode condition) {
return create(input, condition, ImmutableSet.of());
}
/** Creates a LogicalFilter. */
public static LogicalFilter create(final RelNode input, RexNode condition,
ImmutableSet variablesSet) {
final RelOptCluster cluster = input.getCluster();
final RelMetadataQuery mq = cluster.getMetadataQuery();
final RelTraitSet traitSet = cluster.traitSetOf(Convention.NONE)
.replaceIfs(RelCollationTraitDef.INSTANCE,
() -> RelMdCollation.filter(mq, input))
.replaceIf(RelDistributionTraitDef.INSTANCE,
() -> RelMdDistribution.filter(mq, input));
return new LogicalFilter(cluster, traitSet, input, condition, variablesSet);
}
//~ Methods ----------------------------------------------------------------
@Override public Set getVariablesSet() {
return variablesSet;
}
@Override public LogicalFilter copy(RelTraitSet traitSet, RelNode input,
RexNode condition) {
assert traitSet.containsIfApplicable(Convention.NONE);
return new LogicalFilter(getCluster(), traitSet, hints, input, condition,
variablesSet);
}
@Override public RelNode accept(RelShuttle shuttle) {
return shuttle.visit(this);
}
@Override public RelWriter explainTerms(RelWriter pw) {
return super.explainTerms(pw)
.itemIf("variablesSet", variablesSet, !variablesSet.isEmpty());
}
@Override public boolean deepEquals(@Nullable Object obj) {
return deepEquals0(obj)
&& variablesSet.equals(((LogicalFilter) obj).variablesSet);
}
@Override public int deepHashCode() {
return Objects.hash(deepHashCode0(), variablesSet);
}
@Override public RelNode withHints(List hintList) {
return new LogicalFilter(getCluster(), traitSet, hintList, input, condition, variablesSet);
}
}