@Test public void testLimitWithPageAlignment() { List<Page> input = rowPagesBuilder(BIGINT) .addSequencePage(3, 1) .addSequencePage(2, 4) .addSequencePage(2, 6) .build(); OperatorFactory operatorFactory = new LimitOperatorFactory(0, new PlanNodeId("test"), 5); MaterializedResult expected = resultBuilder(driverContext.getSession(), BIGINT) .page(createSequencePage(ImmutableList.of(BIGINT), 3, 1)) .page(createSequencePage(ImmutableList.of(BIGINT), 2, 4)) .build(); OperatorAssertion.assertOperatorEquals(operatorFactory, driverContext, input, expected); }
@Override public void addInput(Page page) { requireNonNull(page, "page is null"); checkState(!finished, "operator finished"); resultBuilder.page(page); operatorContext.recordGeneratedOutput(page.getSizeInBytes(), page.getPositionCount()); }
public static MaterializedResult toMaterializedResult(ConnectorSession session, List<Type> types, List<Page> pages) { // materialize pages MaterializedResult.Builder resultBuilder = MaterializedResult.resultBuilder(session, types); for (Page outputPage : pages) { resultBuilder.page(outputPage); } return resultBuilder.build(); }
public synchronized Builder pages(Iterable<Page> pages) { for (Page page : pages) { this.page(page); } return this; }
public static MaterializedResult materializeSourceDataStream(ConnectorSession session, ConnectorPageSource pageSource, List<Type> types) { MaterializedResult.Builder builder = resultBuilder(session, types); while (!pageSource.isFinished()) { Page outputPage = pageSource.getNextPage(); if (outputPage == null) { break; } builder.page(outputPage); } return builder.build(); }
public static MaterializedResult toMaterializedResult(Session session, List<Type> types, List<Page> pages) { // materialize pages MaterializedResult.Builder resultBuilder = MaterializedResult.resultBuilder(session, types); for (Page outputPage : pages) { resultBuilder.page(outputPage); } return resultBuilder.build(); }
public static MaterializedResult toMaterializedResult(ConnectorSession session, List<Type> types, List<Page> pages) { // materialize pages MaterializedResult.Builder resultBuilder = MaterializedResult.resultBuilder(session, types); for (Page outputPage : pages) { resultBuilder.page(outputPage); } return resultBuilder.build(); }
public synchronized Builder pages(Iterable<Page> pages) { for (Page page : pages) { this.page(page); } return this; }
public static MaterializedResult toMaterializedResult(ConnectorSession session, List<Type> types, List<Page> pages) { // materialize pages MaterializedResult.Builder resultBuilder = MaterializedResult.resultBuilder(session, types); for (Page outputPage : pages) { resultBuilder.page(outputPage); } return resultBuilder.build(); }
public static MaterializedResult toMaterializedResult(Session session, List<Type> types, List<Page> pages) { // materialize pages MaterializedResult.Builder resultBuilder = MaterializedResult.resultBuilder(session, types); for (Page outputPage : pages) { resultBuilder.page(outputPage); } return resultBuilder.build(); }
public static MaterializedResult materializeSourceDataStream(ConnectorSession session, ConnectorPageSource pageSource, List<Type> types) { MaterializedResult.Builder builder = resultBuilder(session, types); while (!pageSource.isFinished()) { Page outputPage = pageSource.getNextPage(); if (outputPage == null) { break; } builder.page(outputPage); } return builder.build(); }