public static Block createTestBlock() { BlockBuilder blockBuilder = BIGINT.createBlockBuilder(null, 15); BIGINT.writeLong(blockBuilder, 1111); BIGINT.writeLong(blockBuilder, 1111); BIGINT.writeLong(blockBuilder, 1111); BIGINT.writeLong(blockBuilder, 2222); BIGINT.writeLong(blockBuilder, 2222); BIGINT.writeLong(blockBuilder, 2222); BIGINT.writeLong(blockBuilder, 2222); BIGINT.writeLong(blockBuilder, 2222); BIGINT.writeLong(blockBuilder, 3333); BIGINT.writeLong(blockBuilder, 3333); BIGINT.writeLong(blockBuilder, 4444); return blockBuilder.build(); }
public static Block getHashBlock(List<? extends Type> hashTypes, Block... hashBlocks) { checkArgument(hashTypes.size() == hashBlocks.length); int[] hashChannels = new int[hashBlocks.length]; for (int i = 0; i < hashBlocks.length; i++) { hashChannels[i] = i; } HashGenerator hashGenerator = new InterpretedHashGenerator(ImmutableList.copyOf(hashTypes), hashChannels); int positionCount = hashBlocks[0].getPositionCount(); BlockBuilder builder = BIGINT.createFixedSizeBlockBuilder(positionCount); Page page = new Page(hashBlocks); for (int i = 0; i < positionCount; i++) { BIGINT.writeLong(builder, hashGenerator.hashPosition(i, page)); } return builder.build(); }
private static BlockBuilder writeValues(long[][] expectedValues, BlockBuilder blockBuilder) { for (long[] expectedValue : expectedValues) { if (expectedValue == null) { blockBuilder.appendNull(); } else { BlockBuilder elementBlockBuilder = BIGINT.createBlockBuilder(null, expectedValue.length); for (long v : expectedValue) { BIGINT.writeLong(elementBlockBuilder, v); } blockBuilder.appendStructure(elementBlockBuilder); } } return blockBuilder; }
public static Block createTestBlock() { BlockBuilder blockBuilder = BIGINT.createBlockBuilder(new BlockBuilderStatus(), 15); BIGINT.writeLong(blockBuilder, 1111); BIGINT.writeLong(blockBuilder, 1111); BIGINT.writeLong(blockBuilder, 1111); BIGINT.writeLong(blockBuilder, 2222); BIGINT.writeLong(blockBuilder, 2222); BIGINT.writeLong(blockBuilder, 2222); BIGINT.writeLong(blockBuilder, 2222); BIGINT.writeLong(blockBuilder, 2222); BIGINT.writeLong(blockBuilder, 3333); BIGINT.writeLong(blockBuilder, 3333); BIGINT.writeLong(blockBuilder, 4444); return blockBuilder.build(); }
public Block serialize() { Block valuesBlock = values.build(); BlockBuilder blockBuilder = new InterleavedBlockBuilder(ImmutableList.of(type, BIGINT), new BlockBuilderStatus(), valuesBlock.getPositionCount() * 2); for (int i = 0; i < valuesBlock.getPositionCount(); i++) { type.appendTo(valuesBlock, i, blockBuilder); BIGINT.writeLong(blockBuilder, counts.get(i)); } return blockBuilder.build(); }
private BlockBuilder createBlockBuilderWithValues(long[][] expectedValues) { BlockBuilder blockBuilder = new ArrayBlockBuilder(BIGINT, new BlockBuilderStatus(), 100, 100); for (long[] expectedValue : expectedValues) { if (expectedValue == null) { blockBuilder.appendNull(); } else { BlockBuilder elementBlockBuilder = BIGINT.createBlockBuilder(new BlockBuilderStatus(), expectedValue.length); for (long v : expectedValue) { BIGINT.writeLong(elementBlockBuilder, v); } blockBuilder.writeObject(elementBlockBuilder).closeEntry(); } } return blockBuilder; }
private static Block createTestBlock() { ArrayBlockBuilder blockBuilder = (ArrayBlockBuilder) TYPE.createBlockBuilder(new BlockBuilderStatus(), 3); ArrayElementBlockWriter arrayElementBlockWriter; arrayElementBlockWriter = blockBuilder.beginBlockEntry(); BIGINT.writeLong(arrayElementBlockWriter, 1); VARCHAR.writeSlice(arrayElementBlockWriter, utf8Slice("cat")); blockBuilder.closeEntry(); arrayElementBlockWriter = blockBuilder.beginBlockEntry(); BIGINT.writeLong(arrayElementBlockWriter, 2); VARCHAR.writeSlice(arrayElementBlockWriter, utf8Slice("cats")); blockBuilder.closeEntry(); arrayElementBlockWriter = blockBuilder.beginBlockEntry(); BIGINT.writeLong(arrayElementBlockWriter, 3); VARCHAR.writeSlice(arrayElementBlockWriter, utf8Slice("dog")); blockBuilder.closeEntry(); return blockBuilder.build(); }
@Override public Block[] getSequenceBlocks(int start, int length) { BlockBuilder blockBuilder = BIGINT.createBlockBuilder(new BlockBuilderStatus(), length); for (int i = start; i < start + length; i++) { BIGINT.writeLong(blockBuilder, i); } return new Block[] {blockBuilder.build()}; }
@Override public Block[] getSequenceBlocks(int start, int length) { BlockBuilder blockBuilder = BIGINT.createBlockBuilder(null, length); for (int i = start; i < start + length; i++) { BIGINT.writeLong(blockBuilder, i); } return new Block[] {blockBuilder.build()}; }
@Override public Block[] getSequenceBlocks(int start, int length) { BlockBuilder blockBuilder = getType().createBlockBuilder(new BlockBuilderStatus(), length); for (int i = start; i < start + length; i++) { if (getType().equals(BIGINT)) { BIGINT.writeLong(blockBuilder, (long) i); } else { DOUBLE.writeDouble(blockBuilder, (double) i); } } return new Block[] {blockBuilder.build()}; }
BlockBuilder blockBuilder = BIGINT.createBlockBuilder(null, NUM_ROUNDS * (NUM_RANDOM_VALUES_IN_ROUND + NUM_EXISTING_VALUES_IN_ROUND)); HashTable hashTable = new HashTable(BIGINT, blockBuilder); Set<Long> valueSet = new HashSet<>(); for (int j = 0; j < NUM_RANDOM_VALUES_IN_ROUND; j++) { long value = rand.nextLong(); BIGINT.writeLong(blockBuilder, value); if (valueSet.contains(value)) { assertTrue(hashTable.contains(blockBuilder.getPositionCount() - 1)); BIGINT.writeLong(blockBuilder, value); assertTrue(hashTable.contains(blockBuilder.getPositionCount() - 1)); assertFalse(hashTable.addIfAbsent(blockBuilder.getPositionCount() - 1));
public static Block createLongsBlock(int... values) { BlockBuilder builder = BIGINT.createBlockBuilder(new BlockBuilderStatus(), 100); for (int value : values) { BIGINT.writeLong(builder, (long) value); } return builder.build(); }
@Override public Block[] getSequenceBlocks(int start, int length) { BlockBuilder blockBuilder = BIGINT.createBlockBuilder(null, length); for (int i = start; i < start + length; i++) { BIGINT.writeLong(blockBuilder, i); } return new Block[] {blockBuilder.build()}; }
blockBuilder = type.createBlockBuilder(new BlockBuilderStatus(), MAX_VECTOR_LENGTH); blockBuilder = type.createBlockBuilder(new BlockBuilderStatus(), MAX_VECTOR_LENGTH, bytes.length); long value = bigintPartitionKey(partitionKey.getValue(), name); for (int i = 0; i < MAX_VECTOR_LENGTH; i++) { BIGINT.writeLong(blockBuilder, value); constantBlocks[columnIndex] = blockBuilder.build(); BlockBuilder blockBuilder = type.createBlockBuilder(new BlockBuilderStatus(), MAX_VECTOR_LENGTH); for (int i = 0; i < MAX_VECTOR_LENGTH; i++) { blockBuilder.appendNull(); constantBlocks[columnIndex] = blockBuilder.build();
@Override public Block[] getSequenceBlocks(int start, int length) { BlockBuilder blockBuilder = BIGINT.createBlockBuilder(new BlockBuilderStatus(), length); for (int i = start; i < start + length; i++) { BIGINT.writeLong(blockBuilder, i); } return new Block[] {blockBuilder.build()}; }
@Override public Block[] getSequenceBlocks(int start, int length) { BlockBuilder blockBuilder = BIGINT.createBlockBuilder(null, length); for (int i = start; i < start + length; i++) { BIGINT.writeLong(blockBuilder, i); } return new Block[] {blockBuilder.build()}; }
private Block generateIdColumn() { BlockBuilder block = BIGINT.createFixedSizeBlockBuilder(inputPage.getPositionCount()); for (int currentPosition = 0; currentPosition < inputPage.getPositionCount(); currentPosition++) { if (rowIdCounter >= maxRowIdCounterValue) { requestValues(); } long rowId = rowIdCounter++; verify((rowId & uniqueValueMask) == 0, "RowId and uniqueValue mask overlaps"); BIGINT.writeLong(block, uniqueValueMask | rowId); } return block.build(); } }
blockBuilder = type.createBlockBuilder(new BlockBuilderStatus(), MAX_BATCH_SIZE); blockBuilder = type.createBlockBuilder(new BlockBuilderStatus(), MAX_BATCH_SIZE, bytes.length); long value = bigintPartitionKey(partitionKey.getValue(), name); for (int i = 0; i < MAX_BATCH_SIZE; i++) { BIGINT.writeLong(blockBuilder, value); constantBlocks[columnIndex] = blockBuilder.build(); BlockBuilder blockBuilder = type.createBlockBuilder(new BlockBuilderStatus(), MAX_BATCH_SIZE, NULL_ENTRY_SIZE); for (int i = 0; i < MAX_BATCH_SIZE; i++) { blockBuilder.appendNull(); constantBlocks[columnIndex] = blockBuilder.build();