com.facebook.presto.jdbc.internal.spi.PageBuilder 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.spi;
import com.facebook.presto.jdbc.internal.spi.block.Block;
import com.facebook.presto.jdbc.internal.spi.block.BlockBuilder;
import com.facebook.presto.jdbc.internal.spi.block.BlockBuilderStatus;
import com.facebook.presto.jdbc.internal.spi.type.Type;
import java.util.ArrayList;
import java.util.List;
import static com.facebook.presto.jdbc.internal.spi.block.BlockBuilderStatus.DEFAULT_MAX_BLOCK_SIZE_IN_BYTES;
import static com.facebook.presto.jdbc.internal.spi.block.BlockBuilderStatus.DEFAULT_MAX_PAGE_SIZE_IN_BYTES;
import static java.util.Collections.unmodifiableList;
import static java.util.Objects.requireNonNull;
public class PageBuilder
{
private final BlockBuilder[] blockBuilders;
private final List types;
private BlockBuilderStatus blockBuilderStatus;
private int declaredPositions;
public PageBuilder(List types)
{
this.types = unmodifiableList(new ArrayList<>(requireNonNull(types, "types is null")));
int maxBlockSizeInBytes;
if (!types.isEmpty()) {
maxBlockSizeInBytes = (int) (1.0 * DEFAULT_MAX_PAGE_SIZE_IN_BYTES / types.size());
maxBlockSizeInBytes = Math.min(DEFAULT_MAX_BLOCK_SIZE_IN_BYTES, maxBlockSizeInBytes);
}
else {
maxBlockSizeInBytes = 0;
}
blockBuilderStatus = new BlockBuilderStatus(DEFAULT_MAX_PAGE_SIZE_IN_BYTES, maxBlockSizeInBytes);
blockBuilders = new BlockBuilder[types.size()];
for (int i = 0; i < blockBuilders.length; i++) {
blockBuilders[i] = types.get(i).createBlockBuilder(blockBuilderStatus);
}
}
public void reset()
{
if (isEmpty()) {
return;
}
declaredPositions = 0;
blockBuilderStatus = new BlockBuilderStatus(blockBuilderStatus);
for (int i = 0; i < types.size(); i++) {
blockBuilders[i] = types.get(i).createBlockBuilder(blockBuilderStatus);
}
}
public BlockBuilder getBlockBuilder(int channel)
{
return blockBuilders[channel];
}
/**
* Hack to declare positions when producing a page with no channels
*/
public void declarePosition()
{
declaredPositions++;
}
public boolean isFull()
{
return declaredPositions == Integer.MAX_VALUE || blockBuilderStatus.isFull();
}
public boolean isEmpty()
{
return blockBuilders.length == 0 ? declaredPositions == 0 : blockBuilderStatus.isEmpty();
}
public int getPositionCount()
{
return declaredPositions;
}
public long getSizeInBytes()
{
return blockBuilderStatus.getSizeInBytes();
}
public Page build()
{
if (blockBuilders.length == 0) {
return new Page(declaredPositions);
}
Block[] blocks = new Block[blockBuilders.length];
for (int i = 0; i < blocks.length; i++) {
blocks[i] = blockBuilders[i].build();
}
return new Page(blocks);
}
}