@Override public BlockBuilder createFixedSizeBlockBuilder(int positionCount) { return new LongArrayBlockBuilder(null, positionCount); }
@Override public boolean isNull(int position) { checkReadablePosition(position); return valueIsNull[position]; }
private void checkReadablePosition(int position) { if (position < 0 || position >= getPositionCount()) { throw new IllegalArgumentException("position is not valid"); } } }
@Test public void testBuildingFromLongArrayBlockBuilder() { LongArrayBlockBuilder blockBuilder = new LongArrayBlockBuilder(null, 100); populateNullValues(blockBuilder, 100); assertEquals(blockBuilder.build().getEncodingName(), RunLengthBlockEncoding.NAME); }
assertEquals(tableFinisher.getComputedStatistics().size(), 1); assertEquals(getOnlyElement(tableFinisher.getComputedStatistics()).getColumnStatistics().size(), 1); Block expectedStatisticsBlock = new LongArrayBlockBuilder(null, 1) .writeLong(7) .closeEntry() .build();
public LongArrayBlockBuilder(@Nullable BlockBuilderStatus blockBuilderStatus, int expectedEntries) { this.blockBuilderStatus = blockBuilderStatus; this.initialEntryCount = max(expectedEntries, 1); updateDataSize(); }
@Override public BlockBuilder writeLong(long value) { if (values.length <= positionCount) { growCapacity(); } values[positionCount] = value; hasNonNullValue = true; positionCount++; if (blockBuilderStatus != null) { blockBuilderStatus.addBytes(Byte.BYTES + Long.BYTES); } return this; }
@Override public long getEstimatedDataSizeForStats(int position) { return isNull(position) ? 0 : Long.BYTES; }
@Test public void testBuildingFromLongArrayBlockBuilder() { LongArrayBlockBuilder blockBuilder = new LongArrayBlockBuilder(null, 100); populateNullValues(blockBuilder, 100); assertEquals(blockBuilder.build().getEncodingName(), RunLengthBlockEncoding.NAME); }
assertEquals(tableFinisher.getComputedStatistics().size(), 1); assertEquals(getOnlyElement(tableFinisher.getComputedStatistics()).getColumnStatistics().size(), 1); Block expectedStatisticsBlock = new LongArrayBlockBuilder(null, 1) .writeLong(7) .closeEntry() .build();
private void growCapacity() { int newSize; if (initialized) { newSize = BlockUtil.calculateNewArraySize(values.length); } else { newSize = initialEntryCount; initialized = true; } valueIsNull = Arrays.copyOf(valueIsNull, newSize); values = Arrays.copyOf(values, newSize); updateDataSize(); }
@Override public BlockBuilder appendNull() { if (values.length <= positionCount) { growCapacity(); } valueIsNull[positionCount] = true; hasNullValue = true; positionCount++; if (blockBuilderStatus != null) { blockBuilderStatus.addBytes(Byte.BYTES + Long.BYTES); } return this; }
@Override public final BlockBuilder createFixedSizeBlockBuilder(int positionCount) { return new LongArrayBlockBuilder(null, positionCount); }
@Override public long getLong(int position, int offset) { checkReadablePosition(position); if (offset != 0) { throw new IllegalArgumentException("offset must be zero"); } return values[position]; }
@Override public String toString() { StringBuilder sb = new StringBuilder("LongArrayBlockBuilder{"); sb.append("positionCount=").append(getPositionCount()); sb.append('}'); return sb.toString(); }
@Override public final BlockBuilder createFixedSizeBlockBuilder(int positionCount) { return new LongArrayBlockBuilder(null, positionCount); }
@Override @Deprecated // TODO: Remove when we fix intermediate types on aggregations. public int getInt(int position, int offset) { checkReadablePosition(position); if (offset != 0) { throw new IllegalArgumentException("offset must be zero"); } return toIntExact(values[position]); }
@Override public Block copyRegion(int positionOffset, int length) { checkValidRegion(getPositionCount(), positionOffset, length); if (!hasNonNullValue) { return new RunLengthEncodedBlock(NULL_VALUE_BLOCK, length); } boolean[] newValueIsNull = null; if (hasNullValue) { newValueIsNull = Arrays.copyOfRange(valueIsNull, positionOffset, positionOffset + length); } long[] newValues = Arrays.copyOfRange(values, positionOffset, positionOffset + length); return new LongArrayBlock(0, length, newValueIsNull, newValues); }
@Override public BlockBuilder newBlockBuilderLike(BlockBuilderStatus blockBuilderStatus) { return new LongArrayBlockBuilder(blockBuilderStatus, calculateBlockResetSize(positionCount)); }
@Override @Deprecated // TODO: Remove when we fix intermediate types on aggregations. public short getShort(int position, int offset) { checkReadablePosition(position); if (offset != 0) { throw new IllegalArgumentException("offset must be zero"); } short value = (short) (values[position]); if (value != values[position]) { throw new ArithmeticException("short overflow"); } return value; }