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 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 io.trino.sql.planner;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableSet;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.Set;
import static com.google.common.base.MoreObjects.toStringHelper;
import static com.google.common.base.Preconditions.checkArgument;
import static java.util.Objects.requireNonNull;
public class PartitioningScheme
{
private final Partitioning partitioning;
private final List outputLayout;
private final Optional hashColumn;
private final boolean replicateNullsAndAny;
private final Optional bucketToPartition;
private final Optional partitionCount;
public PartitioningScheme(Partitioning partitioning, List outputLayout)
{
this(
partitioning,
outputLayout,
Optional.empty(),
false,
Optional.empty(),
Optional.empty());
}
public PartitioningScheme(Partitioning partitioning, List outputLayout, Optional hashColumn)
{
this(
partitioning,
outputLayout,
hashColumn,
false,
Optional.empty(),
Optional.empty());
}
@JsonCreator
public PartitioningScheme(
@JsonProperty("partitioning") Partitioning partitioning,
@JsonProperty("outputLayout") List outputLayout,
@JsonProperty("hashColumn") Optional hashColumn,
@JsonProperty("replicateNullsAndAny") boolean replicateNullsAndAny,
@JsonProperty("bucketToPartition") Optional bucketToPartition,
@JsonProperty("partitionCount") Optional partitionCount)
{
this.partitioning = requireNonNull(partitioning, "partitioning is null");
this.outputLayout = ImmutableList.copyOf(requireNonNull(outputLayout, "outputLayout is null"));
Set columns = partitioning.getColumns();
checkArgument(ImmutableSet.copyOf(outputLayout).containsAll(columns),
"Output layout (%s) don't include all partition columns (%s)", outputLayout, columns);
this.hashColumn = requireNonNull(hashColumn, "hashColumn is null");
hashColumn.ifPresent(column -> checkArgument(outputLayout.contains(column),
"Output layout (%s) don't include hash column (%s)", outputLayout, column));
checkArgument(!replicateNullsAndAny || columns.size() <= 1, "Must have at most one partitioning column when nullPartition is REPLICATE.");
this.replicateNullsAndAny = replicateNullsAndAny;
this.bucketToPartition = requireNonNull(bucketToPartition, "bucketToPartition is null");
this.partitionCount = requireNonNull(partitionCount, "partitionCount is null");
checkArgument(
partitionCount.isEmpty() || partitioning.getHandle().getConnectorHandle() instanceof SystemPartitioningHandle,
"Connector partitioning handle should be of type system partitioning when partitionCount is present");
}
@JsonProperty
public Partitioning getPartitioning()
{
return partitioning;
}
@JsonProperty
public List getOutputLayout()
{
return outputLayout;
}
@JsonProperty
public Optional getHashColumn()
{
return hashColumn;
}
@JsonProperty
public boolean isReplicateNullsAndAny()
{
return replicateNullsAndAny;
}
@JsonProperty
public Optional getBucketToPartition()
{
return bucketToPartition;
}
@JsonProperty
public Optional getPartitionCount()
{
return partitionCount;
}
public PartitioningScheme withBucketToPartition(Optional bucketToPartition)
{
return new PartitioningScheme(partitioning, outputLayout, hashColumn, replicateNullsAndAny, bucketToPartition, partitionCount);
}
public PartitioningScheme withPartitioningHandle(PartitioningHandle partitioningHandle)
{
Partitioning newPartitioning = partitioning.withAlternativePartitioningHandle(partitioningHandle);
return new PartitioningScheme(newPartitioning, outputLayout, hashColumn, replicateNullsAndAny, bucketToPartition, partitionCount);
}
public PartitioningScheme withPartitionCount(Optional partitionCount)
{
return new PartitioningScheme(partitioning, outputLayout, hashColumn, replicateNullsAndAny, bucketToPartition, partitionCount);
}
public PartitioningScheme translateOutputLayout(List newOutputLayout)
{
requireNonNull(newOutputLayout, "newOutputLayout is null");
checkArgument(newOutputLayout.size() == outputLayout.size());
Partitioning newPartitioning = partitioning.translate(symbol -> newOutputLayout.get(outputLayout.indexOf(symbol)));
Optional newHashSymbol = hashColumn
.map(outputLayout::indexOf)
.map(newOutputLayout::get);
return new PartitioningScheme(newPartitioning, newOutputLayout, newHashSymbol, replicateNullsAndAny, bucketToPartition, partitionCount);
}
@Override
public boolean equals(Object o)
{
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
PartitioningScheme that = (PartitioningScheme) o;
return Objects.equals(partitioning, that.partitioning) &&
Objects.equals(outputLayout, that.outputLayout) &&
replicateNullsAndAny == that.replicateNullsAndAny &&
Objects.equals(bucketToPartition, that.bucketToPartition) &&
Objects.equals(partitionCount, that.partitionCount);
}
@Override
public int hashCode()
{
return Objects.hash(partitioning, outputLayout, replicateNullsAndAny, bucketToPartition, partitionCount);
}
@Override
public String toString()
{
return toStringHelper(this)
.add("partitioning", partitioning)
.add("outputLayout", outputLayout)
.add("hashChannel", hashColumn)
.add("replicateNullsAndAny", replicateNullsAndAny)
.add("bucketToPartition", bucketToPartition)
.add("partitionCount", partitionCount)
.toString();
}
}
