@Test
public void testAggregationGroupBysWithCombine() {
final MAggregationGroupByOperator aggregationGroupByOperator = getmAggregationGroupByOperator();
final IntermediateResultsBlock block =
(IntermediateResultsBlock) aggregationGroupByOperator.nextBlock();
for (int i = 0; i < _numAggregations; ++i) {
LOGGER.debug("Result1: {}", block.getAggregationGroupByOperatorResult().get(i));
}
////////////////////////////////////////////////////////////////////////////////
final MAggregationGroupByOperator aggregationGroupByOperator1 =
getmAggregationGroupByOperator();
final IntermediateResultsBlock block1 =
(IntermediateResultsBlock) aggregationGroupByOperator1.nextBlock();
for (int i = 0; i < _numAggregations; ++i) {
LOGGER.debug("Result 2: {}", block1.getAggregationGroupByOperatorResult().get(i));
}
CombineService.mergeTwoBlocks(getAggregationGroupByNoFilterBrokerRequest(), block, block1);
for (int i = 0; i < _numAggregations; ++i) {
LOGGER.debug("Combined Result: {}", block.getAggregationGroupByOperatorResult().get(i));
}
}
@Test
public void testAggregationGroupBysWithDataTableEncodeAndDecode() throws Exception {
final MAggregationGroupByOperator aggregationGroupByOperator = getmAggregationGroupByOperator();
final IntermediateResultsBlock block =
(IntermediateResultsBlock) aggregationGroupByOperator.nextBlock();
for (int i = 0; i < _numAggregations; ++i) {
LOGGER.debug("Result 1: {}", block.getAggregationGroupByOperatorResult().get(i));
}
////////////////////////////////////////////////////////////////////////////////
final MAggregationGroupByOperator aggregationGroupByOperator1 =
getmAggregationGroupByOperator();
final IntermediateResultsBlock block1 =
(IntermediateResultsBlock) aggregationGroupByOperator1.nextBlock();
for (int i = 0; i < _numAggregations; ++i) {
LOGGER.debug("Result 2: {}", block1.getAggregationGroupByOperatorResult().get(i));
}
CombineService.mergeTwoBlocks(getAggregationGroupByNoFilterBrokerRequest(), block, block1);
for (int i = 0; i < _numAggregations; ++i) {
LOGGER.debug("Combined Result: {}", block.getAggregationGroupByOperatorResult().get(i));
}
final DataTable dataTable = block.getAggregationGroupByResultDataTable();
final List<Map<String, Serializable>> results =
AggregationGroupByOperatorService.transformDataTableToGroupByResult(dataTable);
for (int i = 0; i < _numAggregations; ++i) {
LOGGER.debug("Decoded Result: {}", results.get(i));
}
}
@Test
public void testAggregationGroupBysWithDataTableEncodeAndDecode() throws Exception {
final List<AggregationFunctionGroupByOperator> aggregationFunctionGroupByOperatorList =
new ArrayList<AggregationFunctionGroupByOperator>();
final BReusableFilteredDocIdSetOperator docIdSetOperator =
new BReusableFilteredDocIdSetOperator(null, _indexSegment.getTotalDocs(), 5000);
final Map<String, DataSource> dataSourceMap = getDataSourceMap();
final MProjectionOperator projectionOperator =
new MProjectionOperator(dataSourceMap, docIdSetOperator);
for (int i = 0; i < _numAggregations; ++i) {
final MDefaultAggregationFunctionGroupByOperator aggregationFunctionGroupByOperator =
new MDefaultAggregationFunctionGroupByOperator(
_aggregationInfos.get(i),
_groupBy,
new UReplicatedProjectionOperator(projectionOperator),
true);
aggregationFunctionGroupByOperatorList.add(aggregationFunctionGroupByOperator);
}
final MAggregationGroupByOperator aggregationGroupByOperator =
new MAggregationGroupByOperator(
_indexSegment,
_aggregationInfos,
_groupBy,
projectionOperator,
aggregationFunctionGroupByOperatorList);
final IntermediateResultsBlock block =
(IntermediateResultsBlock) aggregationGroupByOperator.nextBlock();
for (int i = 0; i < _numAggregations; ++i) {
LOGGER.info("Result 1: {}", block.getAggregationGroupByOperatorResult().get(i));
}
/////////////////////////////////////////////////////////////////////////
final List<AggregationFunctionGroupByOperator> aggregationFunctionGroupByOperatorList1 =
new ArrayList<AggregationFunctionGroupByOperator>();
final BReusableFilteredDocIdSetOperator docIdSetOperator1 =
new BReusableFilteredDocIdSetOperator(null, _indexSegment.getTotalDocs(), 5000);
final Map<String, DataSource> dataSourceMap1 = getDataSourceMap();
final MProjectionOperator projectionOperator1 =
new MProjectionOperator(dataSourceMap1, docIdSetOperator1);
for (int i = 0; i < _numAggregations; ++i) {
final MDefaultAggregationFunctionGroupByOperator aggregationFunctionGroupByOperator1 =
new MDefaultAggregationFunctionGroupByOperator(
_aggregationInfos.get(i),
_groupBy,
new UReplicatedProjectionOperator(projectionOperator1),
true);
aggregationFunctionGroupByOperatorList1.add(aggregationFunctionGroupByOperator1);
}
final MAggregationGroupByOperator aggregationGroupByOperator1 =
new MAggregationGroupByOperator(
_indexSegment,
_aggregationInfos,
_groupBy,
projectionOperator1,
aggregationFunctionGroupByOperatorList1);
final IntermediateResultsBlock block1 =
(IntermediateResultsBlock) aggregationGroupByOperator1.nextBlock();
for (int i = 0; i < _numAggregations; ++i) {
LOGGER.info("Result 2: {}", block1.getAggregationGroupByOperatorResult().get(i));
}
CombineService.mergeTwoBlocks(getAggregationGroupByNoFilterBrokerRequest(), block, block1);
for (int i = 0; i < _numAggregations; ++i) {
LOGGER.info("Combined Result: {}", block.getAggregationGroupByOperatorResult().get(i));
}
final DataTable dataTable = block.getAggregationGroupByResultDataTable();
final List<Map<String, Serializable>> results =
AggregationGroupByOperatorService.transformDataTableToGroupByResult(dataTable);
for (int i = 0; i < _numAggregations; ++i) {
LOGGER.info("Decode AggregationResult from DataTable: {}", results.get(i));
}
}
