Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* 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.block;
import com.facebook.presto.jdbc.internal.jol.info.ClassLayout;
import java.util.List;
import java.util.function.BiConsumer;
import static com.facebook.presto.jdbc.internal.spi.block.BlockUtil.checkValidRegion;
import static com.facebook.presto.jdbc.internal.spi.block.BlockUtil.compactArray;
import static com.facebook.presto.jdbc.internal.airlift.slice.SizeOf.sizeOf;
public class ShortArrayBlock
implements Block
{
private static final int INSTANCE_SIZE = ClassLayout.parseClass(ShortArrayBlock.class).instanceSize();
private final int arrayOffset;
private final int positionCount;
private final boolean[] valueIsNull;
private final short[] values;
private final long sizeInBytes;
private final long retainedSizeInBytes;
public ShortArrayBlock(int positionCount, boolean[] valueIsNull, short[] values)
{
this(0, positionCount, valueIsNull, values);
}
ShortArrayBlock(int arrayOffset, int positionCount, boolean[] valueIsNull, short[] values)
{
if (arrayOffset < 0) {
throw new IllegalArgumentException("arrayOffset is negative");
}
this.arrayOffset = arrayOffset;
if (positionCount < 0) {
throw new IllegalArgumentException("positionCount is negative");
}
this.positionCount = positionCount;
if (values.length - arrayOffset < positionCount) {
throw new IllegalArgumentException("values length is less than positionCount");
}
this.values = values;
if (valueIsNull.length - arrayOffset < positionCount) {
throw new IllegalArgumentException("isNull length is less than positionCount");
}
this.valueIsNull = valueIsNull;
sizeInBytes = (Short.BYTES + Byte.BYTES) * (long) positionCount;
retainedSizeInBytes = INSTANCE_SIZE + sizeOf(valueIsNull) + sizeOf(values);
}
@Override
public long getSizeInBytes()
{
return sizeInBytes;
}
@Override
public long getRegionSizeInBytes(int position, int length)
{
return (Short.BYTES + Byte.BYTES) * (long) length;
}
@Override
public long getRetainedSizeInBytes()
{
return retainedSizeInBytes;
}
@Override
public void retainedBytesForEachPart(BiConsumer consumer)
{
consumer.accept(values, sizeOf(values));
consumer.accept(valueIsNull, sizeOf(valueIsNull));
consumer.accept(this, (long) INSTANCE_SIZE);
}
@Override
public int getPositionCount()
{
return positionCount;
}
@Override
public short getShort(int position, int offset)
{
checkReadablePosition(position);
if (offset != 0) {
throw new IllegalArgumentException("offset must be zero");
}
return values[position + arrayOffset];
}
@Override
public boolean isNull(int position)
{
checkReadablePosition(position);
return valueIsNull[position + arrayOffset];
}
@Override
public void writePositionTo(int position, BlockBuilder blockBuilder)
{
checkReadablePosition(position);
blockBuilder.writeShort(values[position + arrayOffset]);
}
@Override
public Block getSingleValueBlock(int position)
{
checkReadablePosition(position);
return new ShortArrayBlock(
1,
new boolean[] {valueIsNull[position + arrayOffset]},
new short[] {values[position + arrayOffset]});
}
@Override
public Block copyPositions(List positions)
{
boolean[] newValueIsNull = new boolean[positions.size()];
short[] newValues = new short[positions.size()];
for (int i = 0; i < positions.size(); i++) {
int position = positions.get(i);
checkReadablePosition(position);
newValueIsNull[i] = valueIsNull[position + arrayOffset];
newValues[i] = values[position + arrayOffset];
}
return new ShortArrayBlock(positions.size(), newValueIsNull, newValues);
}
@Override
public Block getRegion(int positionOffset, int length)
{
checkValidRegion(getPositionCount(), positionOffset, length);
return new ShortArrayBlock(positionOffset + arrayOffset, length, valueIsNull, values);
}
@Override
public Block copyRegion(int positionOffset, int length)
{
checkValidRegion(getPositionCount(), positionOffset, length);
positionOffset += arrayOffset;
boolean[] newValueIsNull = compactArray(valueIsNull, positionOffset, length);
short[] newValues = compactArray(values, positionOffset, length);
if (newValueIsNull == valueIsNull && newValues == values) {
return this;
}
return new ShortArrayBlock(length, newValueIsNull, newValues);
}
@Override
public BlockEncoding getEncoding()
{
return new ShortArrayBlockEncoding();
}
@Override
public String toString()
{
StringBuilder sb = new StringBuilder("ShortArrayBlock{");
sb.append("positionCount=").append(getPositionCount());
sb.append('}');
return sb.toString();
}
private void checkReadablePosition(int position)
{
if (position < 0 || position >= getPositionCount()) {
throw new IllegalArgumentException("position is not valid");
}
}
}
