@Override protected List<? extends OperatorFactory> createOperatorFactories() { List<Type> tableScanTypes = getColumnTypes("orders", "totalprice", "clerk"); OperatorFactory tableScanOperator = createTableScanOperator(0, new PlanNodeId("test"), "orders", "totalprice", "clerk"); LimitOperatorFactory limitOperator = new LimitOperatorFactory(1, new PlanNodeId("test"), ROWS); OrderByOperatorFactory orderByOperator = new OrderByOperatorFactory( 2, new PlanNodeId("test"), tableScanTypes, ImmutableList.of(1), ROWS, ImmutableList.of(0), ImmutableList.of(ASC_NULLS_LAST), new PagesIndex.TestingFactory(false)); return ImmutableList.of(tableScanOperator, limitOperator, orderByOperator); }
@Test(expectedExceptions = ExceededMemoryLimitException.class, expectedExceptionsMessageRegExp = "Query exceeded per-node user memory limit of 10B.*") public void testMemoryLimit() { List<Page> input = rowPagesBuilder(BIGINT, DOUBLE) .row(1L, 0.1) .row(2L, 0.2) .pageBreak() .row(-1L, -0.1) .row(4L, 0.4) .build(); DriverContext driverContext = createTaskContext(executor, scheduledExecutor, TEST_SESSION, new DataSize(10, Unit.BYTE)) .addPipelineContext(0, true, true, false) .addDriverContext(); OrderByOperatorFactory operatorFactory = new OrderByOperatorFactory( 0, new PlanNodeId("test"), ImmutableList.of(BIGINT, DOUBLE), ImmutableList.of(1), 10, ImmutableList.of(0), ImmutableList.of(ASC_NULLS_LAST), new PagesIndex.TestingFactory(false)); toPages(operatorFactory, driverContext, input); } }
@Test public void testReverseOrder() { List<Page> input = rowPagesBuilder(BIGINT, DOUBLE) .row(1L, 0.1) .row(2L, 0.2) .pageBreak() .row(-1L, -0.1) .row(4L, 0.4) .build(); OrderByOperatorFactory operatorFactory = new OrderByOperatorFactory( 0, new PlanNodeId("test"), ImmutableList.of(BIGINT, DOUBLE), ImmutableList.of(0), 10, ImmutableList.of(0), ImmutableList.of(DESC_NULLS_LAST), new PagesIndex.TestingFactory(false)); MaterializedResult expected = resultBuilder(driverContext.getSession(), BIGINT) .row(4L) .row(2L) .row(1L) .row(-1L) .build(); assertOperatorEquals(operatorFactory, driverContext, input, expected); }
@Test public void testReverseOrder() throws Exception { List<Page> input = rowPagesBuilder(BIGINT, DOUBLE) .row(1, 0.1) .row(2, 0.2) .pageBreak() .row(-1, -0.1) .row(4, 0.4) .build(); OrderByOperatorFactory operatorFactory = new OrderByOperatorFactory( 0, new PlanNodeId("test"), ImmutableList.of(BIGINT, DOUBLE), ImmutableList.of(0), 10, ImmutableList.of(0), ImmutableList.of(DESC_NULLS_LAST)); Operator operator = operatorFactory.createOperator(driverContext); MaterializedResult expected = resultBuilder(driverContext.getSession(), BIGINT) .row(4) .row(2) .row(1) .row(-1) .build(); assertOperatorEquals(operator, input, expected); }
@Test(expectedExceptions = ExceededMemoryLimitException.class, expectedExceptionsMessageRegExp = "Query exceeded local memory limit of 10B") public void testMemoryLimit() throws Exception { List<Page> input = rowPagesBuilder(BIGINT, DOUBLE) .row(1, 0.1) .row(2, 0.2) .pageBreak() .row(-1, -0.1) .row(4, 0.4) .build(); DriverContext driverContext = createTaskContext(executor, TEST_SESSION, new DataSize(10, Unit.BYTE)) .addPipelineContext(true, true) .addDriverContext(); OrderByOperatorFactory operatorFactory = new OrderByOperatorFactory( 0, new PlanNodeId("test"), ImmutableList.of(BIGINT, DOUBLE), ImmutableList.of(1), 10, ImmutableList.of(0), ImmutableList.of(ASC_NULLS_LAST)); Operator operator = operatorFactory.createOperator(driverContext); toPages(operator, input); } }
@Test public void testMultiFieldKey() throws Exception { List<Page> input = rowPagesBuilder(VARCHAR, BIGINT) .row("a", 1) .row("b", 2) .pageBreak() .row("b", 3) .row("a", 4) .build(); OrderByOperatorFactory operatorFactory = new OrderByOperatorFactory( 0, new PlanNodeId("test"), ImmutableList.of(VARCHAR, BIGINT), ImmutableList.of(0, 1), 10, ImmutableList.of(0, 1), ImmutableList.of(ASC_NULLS_LAST, DESC_NULLS_LAST)); Operator operator = operatorFactory.createOperator(driverContext); MaterializedResult expected = MaterializedResult.resultBuilder(driverContext.getSession(), VARCHAR, BIGINT) .row("a", 4) .row("a", 1) .row("b", 3) .row("b", 2) .build(); assertOperatorEquals(operator, input, expected); }
@Test public void testSingleFieldKey() throws Exception { List<Page> input = rowPagesBuilder(BIGINT, DOUBLE) .row(1, 0.1) .row(2, 0.2) .pageBreak() .row(-1, -0.1) .row(4, 0.4) .build(); OrderByOperatorFactory operatorFactory = new OrderByOperatorFactory( 0, new PlanNodeId("test"), ImmutableList.of(BIGINT, DOUBLE), ImmutableList.of(1), 10, ImmutableList.of(0), ImmutableList.of(ASC_NULLS_LAST)); Operator operator = operatorFactory.createOperator(driverContext); MaterializedResult expected = resultBuilder(driverContext.getSession(), DOUBLE) .row(-0.1) .row(0.1) .row(0.2) .row(0.4) .build(); assertOperatorEquals(operator, input, expected); }
@Override protected List<? extends OperatorFactory> createOperatorFactories() { List<Type> tableScanTypes = getColumnTypes("orders", "totalprice", "clerk"); OperatorFactory tableScanOperator = createTableScanOperator(0, new PlanNodeId("test"), "orders", "totalprice", "clerk"); LimitOperatorFactory limitOperator = new LimitOperatorFactory(1, new PlanNodeId("test"), ROWS); OrderByOperatorFactory orderByOperator = new OrderByOperatorFactory( 2, new PlanNodeId("test"), tableScanTypes, ImmutableList.of(1), ROWS, ImmutableList.of(0), ImmutableList.of(ASC_NULLS_LAST), new PagesIndex.TestingFactory(false)); return ImmutableList.of(tableScanOperator, limitOperator, orderByOperator); }
@Override protected List<? extends OperatorFactory> createOperatorFactories() { OperatorFactory tableScanOperator = createTableScanOperator(0, new PlanNodeId("test"), "orders", "totalprice", "clerk"); LimitOperatorFactory limitOperator = new LimitOperatorFactory(1, new PlanNodeId("test"), tableScanOperator.getTypes(), ROWS); OrderByOperatorFactory orderByOperator = new OrderByOperatorFactory( 2, new PlanNodeId("test"), limitOperator.getTypes(), ImmutableList.of(1), ROWS, ImmutableList.of(0), ImmutableList.of(ASC_NULLS_LAST)); return ImmutableList.of(tableScanOperator, limitOperator, orderByOperator); }
@Override public OperatorFactory duplicate() { return new OrderByOperatorFactory(operatorId, planNodeId, sourceTypes, outputChannels, expectedPositions, sortChannels, sortOrder); } }
@Test public void testSingleFieldKey() { List<Page> input = rowPagesBuilder(BIGINT, DOUBLE) .row(1L, 0.1) .row(2L, 0.2) .pageBreak() .row(-1L, -0.1) .row(4L, 0.4) .build(); OrderByOperatorFactory operatorFactory = new OrderByOperatorFactory( 0, new PlanNodeId("test"), ImmutableList.of(BIGINT, DOUBLE), ImmutableList.of(1), 10, ImmutableList.of(0), ImmutableList.of(ASC_NULLS_LAST), new PagesIndex.TestingFactory(false)); MaterializedResult expected = resultBuilder(driverContext.getSession(), DOUBLE) .row(-0.1) .row(0.1) .row(0.2) .row(0.4) .build(); assertOperatorEquals(operatorFactory, driverContext, input, expected); }
@Test public void testMultiFieldKey() { List<Page> input = rowPagesBuilder(VARCHAR, BIGINT) .row("a", 1L) .row("b", 2L) .pageBreak() .row("b", 3L) .row("a", 4L) .build(); OrderByOperatorFactory operatorFactory = new OrderByOperatorFactory( 0, new PlanNodeId("test"), ImmutableList.of(VARCHAR, BIGINT), ImmutableList.of(0, 1), 10, ImmutableList.of(0, 1), ImmutableList.of(ASC_NULLS_LAST, DESC_NULLS_LAST), new PagesIndex.TestingFactory(false)); MaterializedResult expected = MaterializedResult.resultBuilder(driverContext.getSession(), VARCHAR, BIGINT) .row("a", 4L) .row("a", 1L) .row("b", 3L) .row("b", 2L) .build(); assertOperatorEquals(operatorFactory, driverContext, input, expected); }
@Override public OperatorFactory duplicate() { return new OrderByOperatorFactory(operatorId, planNodeId, sourceTypes, outputChannels, expectedPositions, sortChannels, sortOrder, pagesIndexFactory); } }