/*
* 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 com.facebook.presto.cassandra;
import com.facebook.presto.cassandra.util.CassandraCqlUtils;
import com.facebook.presto.spi.ConnectorColumnHandle;
import com.facebook.presto.spi.ColumnMetadata;
import com.facebook.presto.spi.type.Type;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.google.common.base.Function;
import com.google.common.base.Objects;
import com.google.common.base.Objects.ToStringHelper;
import javax.annotation.Nullable;
import java.util.List;
import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkNotNull;
public class CassandraColumnHandle
implements ConnectorColumnHandle
{
private final String connectorId;
private final String name;
private final int ordinalPosition;
private final CassandraType cassandraType;
private final List<CassandraType> typeArguments;
private final boolean partitionKey;
private final boolean clusteringKey;
@JsonCreator
public CassandraColumnHandle(
@JsonProperty("connectorId") String connectorId,
@JsonProperty("name") String name,
@JsonProperty("ordinalPosition") int ordinalPosition,
@JsonProperty("cassandraType") CassandraType cassandraType,
@Nullable @JsonProperty("typeArguments") List<CassandraType> typeArguments,
@JsonProperty("partitionKey") boolean partitionKey,
@JsonProperty("clusteringKey") boolean clusteringKey)
{
this.connectorId = checkNotNull(connectorId, "connectorId is null");
this.name = checkNotNull(name, "name is null");
checkArgument(ordinalPosition >= 0, "ordinalPosition is negative");
this.ordinalPosition = ordinalPosition;
this.cassandraType = checkNotNull(cassandraType, "cassandraType is null");
int typeArgsSize = cassandraType.getTypeArgumentSize();
if (typeArgsSize > 0) {
this.typeArguments = checkNotNull(typeArguments, "typeArguments is null");
checkArgument(typeArguments.size() == typeArgsSize, cassandraType
+ " must provide " + typeArgsSize + " type arguments");
}
else {
this.typeArguments = null;
}
this.partitionKey = partitionKey;
this.clusteringKey = clusteringKey;
}
@JsonProperty
public String getConnectorId()
{
return connectorId;
}
@JsonProperty
public String getName()
{
return name;
}
@JsonProperty
public int getOrdinalPosition()
{
return ordinalPosition;
}
@JsonProperty
public CassandraType getCassandraType()
{
return cassandraType;
}
@JsonProperty
public List<CassandraType> getTypeArguments()
{
return typeArguments;
}
@JsonProperty
public boolean isPartitionKey()
{
return partitionKey;
}
@JsonProperty
public boolean isClusteringKey()
{
return clusteringKey;
}
public ColumnMetadata getColumnMetadata()
{
return new ColumnMetadata(CassandraCqlUtils.cqlNameToSqlName(name), cassandraType.getNativeType(), ordinalPosition, partitionKey);
}
public Type getType()
{
return cassandraType.getNativeType();
}
@Override
public int hashCode()
{
return Objects.hashCode(
connectorId,
name,
ordinalPosition,
cassandraType,
typeArguments,
partitionKey,
clusteringKey);
}
@Override
public boolean equals(Object obj)
{
if (this == obj) {
return true;
}
if (obj == null || getClass() != obj.getClass()) {
return false;
}
CassandraColumnHandle other = (CassandraColumnHandle) obj;
return Objects.equal(this.connectorId, other.connectorId)
&& Objects.equal(this.name, other.name)
&& Objects.equal(this.ordinalPosition, other.ordinalPosition)
&& Objects.equal(this.cassandraType, other.cassandraType)
&& Objects.equal(this.typeArguments, other.typeArguments)
&& Objects.equal(this.partitionKey, other.partitionKey)
&& Objects.equal(this.clusteringKey, other.clusteringKey);
}
@Override
public String toString()
{
ToStringHelper helper = Objects.toStringHelper(this)
.add("connectorId", connectorId)
.add("name", name)
.add("ordinalPosition", ordinalPosition)
.add("cassandraType", cassandraType);
if (typeArguments != null && !typeArguments.isEmpty()) {
helper.add("typeArguments", typeArguments);
}
helper.add("partitionKey", partitionKey)
.add("clusteringKey", clusteringKey);
return helper.toString();
}
public static Function<ConnectorColumnHandle, CassandraColumnHandle> cassandraColumnHandle()
{
return new Function<ConnectorColumnHandle, CassandraColumnHandle>()
{
@Override
public CassandraColumnHandle apply(ConnectorColumnHandle columnHandle)
{
checkNotNull(columnHandle, "columnHandle is null");
checkArgument(columnHandle instanceof CassandraColumnHandle,
"columnHandle is not an instance of CassandraColumnHandle");
return (CassandraColumnHandle) columnHandle;
}
};
}
public static Function<CassandraColumnHandle, ColumnMetadata> columnMetadataGetter()
{
return new Function<CassandraColumnHandle, ColumnMetadata>()
{
@Override
public ColumnMetadata apply(CassandraColumnHandle input)
{
return input.getColumnMetadata();
}
};
}
public static Function<CassandraColumnHandle, Type> nativeTypeGetter()
{
return new Function<CassandraColumnHandle, Type>()
{
@Override
public Type apply(CassandraColumnHandle input)
{
return input.getType();
}
};
}
public static Function<CassandraColumnHandle, FullCassandraType> cassandraFullTypeGetter()
{
return new Function<CassandraColumnHandle, FullCassandraType>()
{
@Override
public FullCassandraType apply(CassandraColumnHandle input)
{
if (input.getCassandraType().getTypeArgumentSize() == 0) {
return input.getCassandraType();
}
else {
return new CassandraTypeWithTypeArguments(input.getCassandraType(), input.getTypeArguments());
}
}
};
}
}