com.facebook.presto.jdbc.internal.common.block.SingleRowBlockWriter Maven / Gradle / Ivy
/*
* 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.jdbc.internal.common.block;
import com.facebook.presto.jdbc.internal.io.airlift.slice.Slice;
import com.facebook.presto.jdbc.internal.io.airlift.slice.SliceInput;
import com.facebook.presto.jdbc.internal.jol.info.ClassLayout;
import java.util.OptionalInt;
import java.util.function.ObjLongConsumer;
import static java.lang.String.format;
public class SingleRowBlockWriter
extends AbstractSingleRowBlock
implements BlockBuilder
{
private static final int INSTANCE_SIZE = ClassLayout.parseClass(SingleRowBlockWriter.class).instanceSize();
private final BlockBuilder[] fieldBlockBuilders;
private boolean isEntryOpen;
private int currentFieldIndexToWrite;
private boolean fieldBlockBuilderReturned;
SingleRowBlockWriter(int rowIndex, BlockBuilder[] fieldBlockBuilders)
{
super(rowIndex);
this.fieldBlockBuilders = fieldBlockBuilders;
}
/**
* Obtains the field {@code BlockBuilder}.
*
* This method is used to perform random write to {@code SingleRowBlockWriter}.
* Each field {@code BlockBuilder} must be written EXACTLY once.
*
* Field {@code BlockBuilder} can only be obtained before any sequential write has done.
* Once obtained, sequential write is no longer allowed.
*/
public BlockBuilder getFieldBlockBuilder(int fieldIndex)
{
if (currentFieldIndexToWrite != 0) {
throw new IllegalStateException("field block builder can only be obtained before any sequential write has done");
}
fieldBlockBuilderReturned = true;
return fieldBlockBuilders[fieldIndex];
}
@Override
protected Block getRawFieldBlock(int fieldIndex)
{
return fieldBlockBuilders[fieldIndex];
}
@Override
public long getSizeInBytes()
{
long size = 0;
if (rowIndex == 0) {
for (BlockBuilder blockBuilder : fieldBlockBuilders) {
size += blockBuilder.getSizeInBytes();
}
return size;
}
int endIndex = currentFieldIndexToWrite + (isEntryOpen ? 1 : 0);
for (int i = 0; i < endIndex; i++) {
BlockBuilder builder = fieldBlockBuilders[i];
size += (builder.getSizeInBytes() - builder.getRegionSizeInBytes(0, rowIndex));
}
return size;
}
@Override
public OptionalInt fixedSizeInBytesPerPosition()
{
return OptionalInt.empty();
}
@Override
public long getRetainedSizeInBytes()
{
long size = INSTANCE_SIZE;
for (BlockBuilder fieldBlockBuilder : fieldBlockBuilders) {
size += fieldBlockBuilder.getRetainedSizeInBytes();
}
return size;
}
@Override
public void retainedBytesForEachPart(ObjLongConsumer