/*
* 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.kafka;
import com.facebook.presto.spi.ColumnMetadata;
import com.facebook.presto.spi.ConnectorColumnHandle;
import com.facebook.presto.spi.type.Type;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.google.common.base.Objects;
import com.google.common.primitives.Ints;
import static com.google.common.base.Preconditions.checkNotNull;
/**
* Kafka specific connector column handle.
*/
public final class KafkaColumnHandle
implements ConnectorColumnHandle, Comparable<KafkaColumnHandle>
{
private final String connectorId;
private final int ordinalPosition;
/**
* Column Name
*/
private final String name;
/**
* Column type
*/
private final Type type;
/**
* Mapping hint for the decoder. Can be null.
*/
private final String mapping;
/**
* Data format to use (selects the decoder). Can be null.
*/
private final String dataFormat;
/**
* Additional format hint for the selected decoder. Selects a decoder subtype (e.g. which timestamp decoder).
*/
private final String formatHint;
/**
* True if the key decoder should be used, false if the message decoder should be used.
*/
private final boolean keyDecoder;
/**
* True if the column should be hidden.
*/
private final boolean hidden;
/**
* True if the column is internal to the connector and not defined by a topic definition.
*/
private final boolean internal;
@JsonCreator
public KafkaColumnHandle(
@JsonProperty("connectorId") String connectorId,
@JsonProperty("ordinalPosition") int ordinalPosition,
@JsonProperty("name") String name,
@JsonProperty("type") Type type,
@JsonProperty("mapping") String mapping,
@JsonProperty("dataFormat") String dataFormat,
@JsonProperty("formatHint") String formatHint,
@JsonProperty("keyDecoder") boolean keyDecoder,
@JsonProperty("hidden") boolean hidden,
@JsonProperty("internal") boolean internal)
{
this.connectorId = checkNotNull(connectorId, "connectorId is null");
this.ordinalPosition = ordinalPosition;
this.name = checkNotNull(name, "name is null");
this.type = checkNotNull(type, "type is null");
this.mapping = mapping;
this.dataFormat = dataFormat;
this.formatHint = formatHint;
this.keyDecoder = keyDecoder;
this.hidden = hidden;
this.internal = internal;
}
@JsonProperty
public String getConnectorId()
{
return connectorId;
}
@JsonProperty
public int getOrdinalPosition()
{
return ordinalPosition;
}
@JsonProperty
public String getName()
{
return name;
}
@JsonProperty
public Type getType()
{
return type;
}
@JsonProperty
public String getMapping()
{
return mapping;
}
@JsonProperty
public String getDataFormat()
{
return dataFormat;
}
@JsonProperty
public String getFormatHint()
{
return formatHint;
}
@JsonProperty
public boolean isKeyDecoder()
{
return keyDecoder;
}
@JsonProperty
public boolean isHidden()
{
return hidden;
}
@JsonProperty
public boolean isInternal()
{
return internal;
}
ColumnMetadata getColumnMetadata()
{
return new ColumnMetadata(name, type, ordinalPosition, false, null, hidden);
}
@Override
public int hashCode()
{
return Objects.hashCode(connectorId, ordinalPosition, name, type, mapping, dataFormat, formatHint, keyDecoder, hidden, internal);
}
@Override
public boolean equals(Object obj)
{
if (this == obj) {
return true;
}
if (obj == null || getClass() != obj.getClass()) {
return false;
}
KafkaColumnHandle other = (KafkaColumnHandle) obj;
return Objects.equal(this.connectorId, other.connectorId) &&
Objects.equal(this.ordinalPosition, other.ordinalPosition) &&
Objects.equal(this.name, other.name) &&
Objects.equal(this.type, other.type) &&
Objects.equal(this.mapping, other.mapping) &&
Objects.equal(this.dataFormat, other.dataFormat) &&
Objects.equal(this.formatHint, other.formatHint) &&
Objects.equal(this.keyDecoder, other.keyDecoder) &&
Objects.equal(this.hidden, other.hidden) &&
Objects.equal(this.internal, other.internal);
}
@Override
public int compareTo(KafkaColumnHandle otherHandle)
{
return Ints.compare(this.getOrdinalPosition(), otherHandle.getOrdinalPosition());
}
@Override
public String toString()
{
return Objects.toStringHelper(this)
.add("connectorId", connectorId)
.add("ordinalPosition", ordinalPosition)
.add("name", name)
.add("type", type)
.add("mapping", mapping)
.add("dataFormat", dataFormat)
.add("formatHint", formatHint)
.add("keyDecoder", keyDecoder)
.add("hidden", hidden)
.add("internal", internal)
.toString();
}
}