/*
* 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.mongodb;
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.type.AbstractVariableWidthType;
import com.facebook.presto.spi.type.SqlVarbinary;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;
import io.airlift.slice.Slice;
import org.bson.types.ObjectId;
import java.io.IOException;
import static com.facebook.presto.spi.type.TypeSignature.parseTypeSignature;
public class ObjectIdType
extends AbstractVariableWidthType
{
public static final ObjectIdType OBJECT_ID = new ObjectIdType();
@JsonCreator
public ObjectIdType()
{
super(parseTypeSignature("ObjectId"), Slice.class);
}
@Override
public boolean isComparable()
{
return true;
}
@Override
public boolean isOrderable()
{
return true;
}
@Override
public Object getObjectValue(ConnectorSession session, Block block, int position)
{
if (block.isNull(position)) {
return null;
}
// TODO: There's no way to represent string value of a custom type
return new SqlVarbinary(block.getSlice(position, 0, block.getSliceLength(position)).getBytes());
}
@Override
public boolean equalTo(Block leftBlock, int leftPosition, Block rightBlock, int rightPosition)
{
int leftLength = leftBlock.getSliceLength(leftPosition);
int rightLength = rightBlock.getSliceLength(rightPosition);
if (leftLength != rightLength) {
return false;
}
return leftBlock.equals(leftPosition, 0, rightBlock, rightPosition, 0, leftLength);
}
@Override
public long hash(Block block, int position)
{
return block.hash(position, 0, block.getSliceLength(position));
}
@Override
public int compareTo(Block leftBlock, int leftPosition, Block rightBlock, int rightPosition)
{
int leftLength = leftBlock.getSliceLength(leftPosition);
int rightLength = rightBlock.getSliceLength(rightPosition);
return leftBlock.compareTo(leftPosition, 0, leftLength, rightBlock, rightPosition, 0, rightLength);
}
@Override
public void appendTo(Block block, int position, BlockBuilder blockBuilder)
{
if (block.isNull(position)) {
blockBuilder.appendNull();
}
else {
block.writeBytesTo(position, 0, block.getSliceLength(position), blockBuilder);
blockBuilder.closeEntry();
}
}
@Override
public Slice getSlice(Block block, int position)
{
return block.getSlice(position, 0, block.getSliceLength(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();
}
public static class ObjectIdSerializer
extends JsonSerializer<ObjectId>
{
@Override
public void serialize(ObjectId objectId, JsonGenerator jsonGenerator, SerializerProvider serializerProvider)
throws IOException
{
jsonGenerator.writeString(objectId.toString());
}
}
}