use of org.apache.carbondata.core.metadata.blocklet.DataFileFooter in project carbondata by apache.
the class BTreeBlockFinderTest method testBtreeSearchIsWorkingAndGivingPorperBlockletWithDictionaryKey.
/**
* Below method will test when key which is not present and key which is
* more than
* last node key is passes for searching it should give first block
*/
@Test
public void testBtreeSearchIsWorkingAndGivingPorperBlockletWithDictionaryKey() throws KeyGenException {
BtreeBuilder builder = new BlockBTreeBuilder();
List<DataFileFooter> footerList = getFileFooterListWithOnlyDictionaryKey();
BTreeBuilderInfo infos = new BTreeBuilderInfo(footerList, null);
builder.build(infos);
DataRefNode dataBlock = builder.get();
assertTrue(dataBlock != null);
DataRefNodeFinder finder = new BTreeDataRefNodeFinder(new int[] { 2, 2 }, 2, 0);
int[] dimensionBitLength = CarbonUtil.getDimensionBitLength(new int[] { 10000, 10000 }, new int[] { 1, 1 });
KeyGenerator multiDimKeyVarLengthGenerator = new MultiDimKeyVarLengthGenerator(dimensionBitLength);
IndexKey key = new IndexKey(multiDimKeyVarLengthGenerator.generateKey(new int[] { 10001, 10001 }), null);
DataRefNode findFirstBlock = finder.findFirstDataBlock(dataBlock, key);
assertEquals(99, findFirstBlock.nodeNumber());
DataRefNode findLastBlock = finder.findLastDataBlock(dataBlock, key);
assertEquals(99, findLastBlock.nodeNumber());
}
use of org.apache.carbondata.core.metadata.blocklet.DataFileFooter in project carbondata by apache.
the class BTreeBlockFinderTest method getFileMatadataWithOnlyNoDictionaryKey.
private DataFileFooter getFileMatadataWithOnlyNoDictionaryKey(byte[] startKey, byte[] endKey, byte[] noDictionaryStartKey, byte[] noDictionaryEndKey) {
DataFileFooter footer = new DataFileFooter();
BlockletIndex index = new BlockletIndex();
BlockletBTreeIndex btreeIndex = new BlockletBTreeIndex();
ByteBuffer buffer = ByteBuffer.allocate(4 + 0 + 4 + noDictionaryStartKey.length);
buffer.putInt(0);
buffer.putInt(noDictionaryStartKey.length);
buffer.put(noDictionaryStartKey);
buffer.rewind();
btreeIndex.setStartKey(buffer.array());
ByteBuffer buffer1 = ByteBuffer.allocate(4 + 0 + 4 + noDictionaryEndKey.length);
buffer1.putInt(0);
buffer1.putInt(noDictionaryEndKey.length);
buffer1.put(noDictionaryEndKey);
buffer1.rewind();
btreeIndex.setEndKey(buffer1.array());
BlockletMinMaxIndex minMax = new BlockletMinMaxIndex();
minMax.setMaxValues(new byte[][] { endKey, noDictionaryEndKey });
minMax.setMinValues(new byte[][] { startKey, noDictionaryStartKey });
index.setBtreeIndex(btreeIndex);
index.setMinMaxIndex(minMax);
footer.setBlockletIndex(index);
return footer;
}
use of org.apache.carbondata.core.metadata.blocklet.DataFileFooter in project carbondata by apache.
the class BTreeBlockFinderTest method testBtreeBuilderGetMethodIsGivingNotNullRootNode.
@Test
public void testBtreeBuilderGetMethodIsGivingNotNullRootNode() {
BtreeBuilder builder = new BlockBTreeBuilder();
List<DataFileFooter> footerList = getDataFileFooterList();
BTreeBuilderInfo infos = new BTreeBuilderInfo(footerList, null);
builder.build(infos);
DataRefNode dataBlock = builder.get();
assertTrue(dataBlock != null);
}
use of org.apache.carbondata.core.metadata.blocklet.DataFileFooter in project carbondata by apache.
the class BTreeBlockFinderTest method testBtreeSearchIsWorkingAndGivingPorperBlockletWithDictionaryKey2.
@Test
public void testBtreeSearchIsWorkingAndGivingPorperBlockletWithDictionaryKey2() throws KeyGenException {
BtreeBuilder builder = new BlockBTreeBuilder();
List<DataFileFooter> footerList = getFileFooterListWithOnlyDictionaryKey();
BTreeBuilderInfo infos = new BTreeBuilderInfo(footerList, null);
builder.build(infos);
DataRefNode dataBlock = builder.get();
assertTrue(dataBlock != null);
DataRefNodeFinder finder = new BTreeDataRefNodeFinder(new int[] { 2, 2 }, 2, 0);
int[] dimensionBitLength = CarbonUtil.getDimensionBitLength(new int[] { 10000, 10000 }, new int[] { 1, 1 });
KeyGenerator multiDimKeyVarLengthGenerator = new MultiDimKeyVarLengthGenerator(dimensionBitLength);
IndexKey key = new IndexKey(multiDimKeyVarLengthGenerator.generateKey(new int[] { 0, 0 }), null);
DataRefNode findFirstBlock = finder.findFirstDataBlock(dataBlock, key);
assertEquals(0, findFirstBlock.nodeNumber());
DataRefNode findLastBlock = finder.findLastDataBlock(dataBlock, key);
assertEquals(0, findLastBlock.nodeNumber());
}
use of org.apache.carbondata.core.metadata.blocklet.DataFileFooter in project carbondata by apache.
the class BTreeBlockFinderTest method getDataFileFooterList.
private List<DataFileFooter> getDataFileFooterList() {
List<DataFileFooter> list = new ArrayList<DataFileFooter>();
try {
int[] dimensionBitLength = CarbonUtil.getDimensionBitLength(new int[] { 10000, 10000 }, new int[] { 1, 1 });
KeyGenerator multiDimKeyVarLengthGenerator = new MultiDimKeyVarLengthGenerator(dimensionBitLength);
int i = 1;
while (i < 1001) {
byte[] startKey = multiDimKeyVarLengthGenerator.generateKey(new int[] { i, i });
byte[] endKey = multiDimKeyVarLengthGenerator.generateKey(new int[] { i + 10, i + 10 });
ByteBuffer buffer = ByteBuffer.allocate(4 + 1);
buffer.rewind();
buffer.put((byte) 1);
buffer.putInt(i);
buffer.array();
byte[] noDictionaryStartKey = buffer.array();
ByteBuffer buffer1 = ByteBuffer.allocate(4 + 1);
buffer1.rewind();
buffer1.put((byte) 1);
buffer1.putInt(i + 10);
buffer1.array();
byte[] noDictionaryEndKey = buffer.array();
DataFileFooter footer = getFileFooter(startKey, endKey, noDictionaryStartKey, noDictionaryEndKey);
list.add(footer);
i = i + 10;
}
} catch (Exception e) {
LOGGER.error(e);
}
return list;
}
Aggregations