io.trino.plugin.cassandra.CassandraSplit Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of trino-cassandra Show documentation
Show all versions of trino-cassandra Show documentation
Trino - Cassandra connector
The newest version!
/*
* 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.plugin.cassandra;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import io.trino.spi.HostAddress;
import io.trino.spi.connector.ConnectorSplit;
import java.util.List;
import java.util.Map;
import static com.google.common.base.MoreObjects.toStringHelper;
import static io.airlift.slice.SizeOf.estimatedSizeOf;
import static io.airlift.slice.SizeOf.instanceSize;
import static java.util.Objects.requireNonNull;
import static java.util.stream.Collectors.joining;
public class CassandraSplit
implements ConnectorSplit
{
private static final int INSTANCE_SIZE = instanceSize(CassandraSplit.class);
private final String partitionId;
private final List addresses;
private final String splitCondition;
@JsonCreator
public CassandraSplit(
@JsonProperty("partitionId") String partitionId,
@JsonProperty("splitCondition") String splitCondition,
@JsonProperty("addresses") List addresses)
{
requireNonNull(partitionId, "partitionId is null");
requireNonNull(addresses, "addresses is null");
this.partitionId = partitionId;
this.addresses = ImmutableList.copyOf(addresses);
this.splitCondition = splitCondition;
}
@JsonProperty
public String getSplitCondition()
{
return splitCondition;
}
@JsonProperty
public String getPartitionId()
{
return partitionId;
}
@JsonProperty
@Override
public List getAddresses()
{
return addresses;
}
@Override
public Map getSplitInfo()
{
return ImmutableMap.builder()
.put("hosts", addresses.stream().map(HostAddress::toString).collect(joining(",")))
.put("partitionId", partitionId)
.buildOrThrow();
}
@Override
public long getRetainedSizeInBytes()
{
return INSTANCE_SIZE
+ estimatedSizeOf(partitionId)
+ estimatedSizeOf(addresses, HostAddress::getRetainedSizeInBytes)
+ estimatedSizeOf(splitCondition);
}
@Override
public String toString()
{
return toStringHelper(this)
.addValue(partitionId)
.toString();
}
public String getWhereClause()
{
if (partitionId.equals(CassandraPartition.UNPARTITIONED_ID)) {
if (splitCondition != null) {
return " WHERE " + splitCondition;
}
return "";
}
if (splitCondition != null) {
return " WHERE " + partitionId + " AND " + splitCondition;
}
return " WHERE " + partitionId;
}
}