Examples with TopNProjection io.crate.execution.dsl.projection.TopNProjection used on opensource projects

Example 6 with TopNProjection

use of io.crate.execution.dsl.projection.TopNProjection in project crate by crate.

the class GroupByPlannerTest method testCountDistinctWithGroupBy.

@Test
public void testCountDistinctWithGroupBy() throws Exception {
    var e = SQLExecutor.builder(clusterService, 2, RandomizedTest.getRandom(), List.of()).addTable(TableDefinitions.USER_TABLE_DEFINITION).build();
    Merge distributedGroupByMerge = e.plan("select count(distinct id), name from users group by name order by count(distinct id)");
    Merge reducerMerge = (Merge) distributedGroupByMerge.subPlan();
    CollectPhase collectPhase = ((Collect) reducerMerge.subPlan()).collectPhase();
    // collect
    assertThat(collectPhase.toCollect().get(0), instanceOf(Reference.class));
    assertThat(collectPhase.toCollect().size(), is(2));
    assertThat(((Reference) collectPhase.toCollect().get(0)).column().name(), is("id"));
    assertThat(((Reference) collectPhase.toCollect().get(1)).column().name(), is("name"));
    Projection projection = collectPhase.projections().get(0);
    assertThat(projection, instanceOf(GroupProjection.class));
    GroupProjection groupProjection = (GroupProjection) projection;
    Symbol groupKey = groupProjection.keys().get(0);
    assertThat(groupKey, instanceOf(InputColumn.class));
    assertThat(((InputColumn) groupKey).index(), is(1));
    assertThat(groupProjection.values().size(), is(1));
    assertThat(groupProjection.mode(), is(AggregateMode.ITER_PARTIAL));
    Aggregation aggregation = groupProjection.values().get(0);
    Symbol aggregationInput = aggregation.inputs().get(0);
    assertThat(aggregationInput.symbolType(), is(SymbolType.INPUT_COLUMN));
    // reducer
    MergePhase mergePhase = reducerMerge.mergePhase();
    assertThat(mergePhase.projections(), contains(instanceOf(GroupProjection.class), instanceOf(OrderedTopNProjection.class), instanceOf(EvalProjection.class)));
    Projection groupProjection1 = mergePhase.projections().get(0);
    groupProjection = (GroupProjection) groupProjection1;
    assertThat(groupProjection.keys().get(0), instanceOf(InputColumn.class));
    assertThat(((InputColumn) groupProjection.keys().get(0)).index(), is(0));
    assertThat(groupProjection.mode(), is(AggregateMode.PARTIAL_FINAL));
    assertThat(groupProjection.values().get(0), instanceOf(Aggregation.class));
    OrderedTopNProjection topNProjection = (OrderedTopNProjection) mergePhase.projections().get(1);
    Symbol collection_count = topNProjection.outputs().get(0);
    assertThat(collection_count, SymbolMatchers.isInputColumn(0));
    // handler
    MergePhase localMergeNode = distributedGroupByMerge.mergePhase();
    assertThat(localMergeNode.projections(), Matchers.emptyIterable());
}

Example 7 with TopNProjection

use of io.crate.execution.dsl.projection.TopNProjection in project crate by crate.

the class GroupByPlannerTest method testGroupByWithAggregationAndLimit.

@Test
public void testGroupByWithAggregationAndLimit() throws Exception {
    var e = SQLExecutor.builder(clusterService, 2, RandomizedTest.getRandom(), List.of()).addTable(TableDefinitions.USER_TABLE_DEFINITION).build();
    Merge distributedGroupByMerge = e.plan("select count(*), name from users group by name limit 1 offset 1");
    Merge reducerMerge = (Merge) distributedGroupByMerge.subPlan();
    // distributed merge
    MergePhase distributedMergePhase = reducerMerge.mergePhase();
    assertThat(distributedMergePhase.projections().get(0), instanceOf(GroupProjection.class));
    assertThat(distributedMergePhase.projections().get(1), instanceOf(TopNProjection.class));
    // limit must include offset because the real limit can only be applied on the handler
    // after all rows have been gathered.
    TopNProjection topN = (TopNProjection) distributedMergePhase.projections().get(1);
    assertThat(topN.limit(), is(2));
    assertThat(topN.offset(), is(0));
    // local merge
    MergePhase localMergePhase = distributedGroupByMerge.mergePhase();
    assertThat(localMergePhase.projections().get(0), instanceOf(TopNProjection.class));
    topN = (TopNProjection) localMergePhase.projections().get(0);
    assertThat(topN.limit(), is(1));
    assertThat(topN.offset(), is(1));
    assertThat(topN.outputs().get(0), instanceOf(InputColumn.class));
    assertThat(((InputColumn) topN.outputs().get(0)).index(), is(0));
    assertThat(topN.outputs().get(1), instanceOf(InputColumn.class));
    assertThat(((InputColumn) topN.outputs().get(1)).index(), is(1));
}

Example 8 with TopNProjection

use of io.crate.execution.dsl.projection.TopNProjection in project crate by crate.

the class MergeNodeTest method testSerialization.

@Test
public void testSerialization() throws Exception {
    List<Symbol> keys = Collections.singletonList(new InputColumn(0, DataTypes.STRING));
    List<Aggregation> aggregations = Collections.singletonList(new Aggregation(CountAggregation.COUNT_STAR_SIGNATURE, CountAggregation.COUNT_STAR_SIGNATURE.getReturnType().createType(), Collections.emptyList()));
    GroupProjection groupProjection = new GroupProjection(keys, aggregations, AggregateMode.PARTIAL_FINAL, RowGranularity.CLUSTER);
    TopNProjection topNProjection = new TopNProjection(10, 0, Symbols.typeView(groupProjection.outputs()));
    List<Projection> projections = Arrays.asList(groupProjection, topNProjection);
    MergePhase node = new MergePhase(UUID.randomUUID(), 0, "merge", 2, 1, Set.of("node1", "node2"), List.of(DataTypes.UNDEFINED, DataTypes.STRING), projections, DistributionInfo.DEFAULT_BROADCAST, null);
    BytesStreamOutput output = new BytesStreamOutput();
    node.writeTo(output);
    StreamInput input = output.bytes().streamInput();
    MergePhase node2 = new MergePhase(input);
    assertThat(node.numUpstreams(), is(node2.numUpstreams()));
    assertThat(node.nodeIds(), is(node2.nodeIds()));
    assertThat(node.jobId(), is(node2.jobId()));
    assertEquals(node.inputTypes(), node2.inputTypes());
    assertThat(node.phaseId(), is(node2.phaseId()));
    assertThat(node.distributionInfo(), is(node2.distributionInfo()));
}

Example 9 with TopNProjection

use of io.crate.execution.dsl.projection.TopNProjection in project crate by crate.

the class ProjectionToProjectorVisitorTest method testSimpleTopNProjection.

@Test
public void testSimpleTopNProjection() throws Exception {
    TopNProjection projection = new TopNProjection(10, 2, Collections.singletonList(DataTypes.LONG));
    Projector projector = visitor.create(projection, txnCtx, RamAccounting.NO_ACCOUNTING, memoryManager, UUID.randomUUID());
    assertThat(projector, instanceOf(SimpleTopNProjector.class));
    TestingRowConsumer consumer = new TestingRowConsumer();
    consumer.accept(projector.apply(TestingBatchIterators.range(0, 20)), null);
    List<Object[]> result = consumer.getResult();
    assertThat(result.size(), is(10));
    assertThat(result.get(0), is(new Object[] { 2 }));
}