/*
* 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.ml.type;
import com.facebook.presto.spi.ConnectorSession;
import com.facebook.presto.spi.block.Block;
import com.facebook.presto.spi.block.BlockBuilder;
import com.facebook.presto.spi.block.BlockBuilderStatus;
import com.facebook.presto.spi.block.VariableWidthBlockBuilder;
import com.facebook.presto.spi.type.Type;
import com.facebook.presto.spi.type.VariableWidthType;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.google.common.collect.ImmutableList;
import io.airlift.slice.Slice;
import java.util.List;
// Layout is <size>:<model>, where
// size: is an int describing the length of the model bytes
// model: is the serialized model
public class ModelType
implements VariableWidthType
{
public static final ModelType MODEL = new ModelType();
@JsonCreator
public ModelType()
{
}
@Override
public String getName()
{
return "Model";
}
@Override
public boolean isComparable()
{
return false;
}
@Override
public boolean isOrderable()
{
return false;
}
@Override
public Class<?> getJavaType()
{
return Slice.class;
}
@Override
public List<Type> getTypeParameters()
{
return ImmutableList.of();
}
@Override
public boolean equalTo(Block leftBlock, int leftPosition, Block rightBlock, int rightPosition)
{
throw new UnsupportedOperationException(String.format("%s type is not comparable", getName()));
}
@Override
public int hash(Block block, int position)
{
throw new UnsupportedOperationException(String.format("%s type is not comparable", getName()));
}
@Override
public int compareTo(Block leftBlock, int leftPosition, Block rightBlock, int rightPosition)
{
throw new UnsupportedOperationException(String.format("%s type is not ordered", getName()));
}
@Override
public void appendTo(Block block, int position, BlockBuilder blockBuilder)
{
if (block.isNull(position)) {
blockBuilder.appendNull();
}
else {
block.writeBytesTo(position, 0, block.getLength(position), blockBuilder);
blockBuilder.closeEntry();
}
}
@Override
public boolean getBoolean(Block block, int position)
{
throw new UnsupportedOperationException();
}
@Override
public void writeBoolean(BlockBuilder sliceOutput, boolean value)
{
throw new UnsupportedOperationException();
}
@Override
public long getLong(Block block, int position)
{
throw new UnsupportedOperationException();
}
@Override
public void writeLong(BlockBuilder sliceOutput, long value)
{
throw new UnsupportedOperationException();
}
@Override
public double getDouble(Block block, int position)
{
throw new UnsupportedOperationException();
}
@Override
public void writeDouble(BlockBuilder sliceOutput, double value)
{
throw new UnsupportedOperationException();
}
@Override
public Slice getSlice(Block block, int position)
{
return block.getSlice(position, 0, block.getLength(position));
}
@Override
public void writeSlice(BlockBuilder blockBuilder, Slice value)
{
writeSlice(blockBuilder, value, 0, value.length());
}
@Override
public void writeSlice(BlockBuilder blockBuilder, Slice value, int offset, int length)
{
blockBuilder.writeBytes(value, offset, length).closeEntry();
}
@Override
public Object getObjectValue(ConnectorSession session, Block block, int position)
{
if (block.isNull(position)) {
return null;
}
return String.format("<%s>", getName());
}
@Override
public BlockBuilder createBlockBuilder(BlockBuilderStatus blockBuilderStatus)
{
return new VariableWidthBlockBuilder(blockBuilderStatus);
}
@Override
public boolean equals(Object o)
{
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
return true;
}
@Override
public int hashCode()
{
return getClass().hashCode();
}
@Override
public String toString()
{
return getName();
}
}