@Test public void testDoubleArrayApproxPercentile() { assertWindowQuery("approx_percentile(DOUBLE '42.3', array[0.5]) over ()", resultBuilder(TEST_SESSION, INTEGER, VARCHAR, new ArrayType(DOUBLE)) .row(1, "O", ImmutableList.of(42.3d)) .row(2, "O", ImmutableList.of(42.3d)) .row(3, "F", ImmutableList.of(42.3d)) .row(4, "O", ImmutableList.of(42.3d)) .row(5, "F", ImmutableList.of(42.3d)) .row(6, "F", ImmutableList.of(42.3d)) .row(7, "O", ImmutableList.of(42.3d)) .row(32, "O", ImmutableList.of(42.3d)) .row(33, "F", ImmutableList.of(42.3d)) .row(34, "O", ImmutableList.of(42.3d)) .build()); } }
public static Builder resultBuilder(ConnectorSession session, Iterable<? extends Type> types) { return new Builder(session, ImmutableList.copyOf(types)); }
resultBuilder.rows(CREATE_TABLE_PARTITIONED_DATA.getMaterializedRows()); MaterializedResult result = readTable(transaction, tableHandle, columnHandles, session, TupleDomain.all(), OptionalInt.empty(), Optional.of(storageFormat)); assertEqualsIgnoreOrder(result.getMaterializedRows(), resultBuilder.build().getMaterializedRows()); assertEqualsIgnoreOrder(result.getMaterializedRows(), resultBuilder.build().getMaterializedRows());
resultBuilder.rows(CREATE_TABLE_DATA.getMaterializedRows()); MaterializedResult result = readTable(transaction, tableHandle, columnHandles, session, TupleDomain.all(), OptionalInt.empty(), Optional.empty()); assertEqualsIgnoreOrder(result.getMaterializedRows(), resultBuilder.build().getMaterializedRows()); List<ColumnHandle> columnHandles = filterNonHiddenColumnHandles(metadata.getColumnHandles(session, tableHandle).values()); MaterializedResult result = readTable(transaction, tableHandle, columnHandles, session, TupleDomain.all(), OptionalInt.empty(), Optional.empty()); assertEqualsIgnoreOrder(result.getMaterializedRows(), resultBuilder.build().getMaterializedRows());
expectedResultBuilder.rows(CREATE_TABLE_PARTITIONED_DATA.getMaterializedRows()); List<ColumnHandle> columnHandles = filterNonHiddenColumnHandles(metadata.getColumnHandles(session, tableHandle).values()); MaterializedResult result = readTable(transaction, tableHandle, columnHandles, session, TupleDomain.all(), OptionalInt.empty(), Optional.of(storageFormat)); assertEqualsIgnoreOrder(result.getMaterializedRows(), expectedResultBuilder.build().getMaterializedRows()); ImmutableList<MaterializedRow> expectedRows = expectedResultBuilder.build().getMaterializedRows().stream() .filter(row -> !"2015-07-03".equals(row.getField(dsColumnOrdinalPosition))) .collect(toImmutableList());
.row(110L, "a", "alter1") .row(120L, "a", "insert1") .row(140L, "a", "drop1") .row(210L, "b", "drop2") .row(310L, "c", "alter2") .row(320L, "c", "alter3") .row(510L, "e", "drop3") .row(610L, "f", "insert2") .row(620L, "f", "insert3") .build(); Domain domainToDrop = Domain.create(ValueSet.of( createUnboundedVarcharType(), if (allowInsertExisting) { extraRowsForInsertExisting = MaterializedResult.resultBuilder(SESSION, BIGINT, createUnboundedVarcharType(), createUnboundedVarcharType()) .row(121L, "a", "insert1") .row(611L, "f", "insert2") .row(621L, "f", "insert3") .build() .getMaterializedRows(); .row(111L, "a", "alter1") .row(131L, "a", "add1") .row(221L, "b", "add2") .row(311L, "c", "alter2") .row(321L, "c", "alter3") .row(411L, "d", "add3") .rows(extraRowsForInsertExisting) .build();
expectedResultBuilder.rows(CREATE_TABLE_PARTITIONED_DATA.getMaterializedRows()); assertEqualsIgnoreOrder(result.getMaterializedRows(), expectedResultBuilder.build().getMaterializedRows()); ImmutableList<MaterializedRow> expectedRows = expectedResultBuilder.build().getMaterializedRows().stream() .filter(row -> !"2015-07-03".equals(row.getField(dsColumnOrdinalPosition))) .collect(ImmutableCollectors.toImmutableList());
resultBuilder.rows(CREATE_TABLE_DATA.getMaterializedRows()); MaterializedResult result = readTable(tableHandle, columnHandles, session, TupleDomain.<ColumnHandle>all(), OptionalInt.empty(), Optional.empty()); assertEqualsIgnoreOrder(result.getMaterializedRows(), resultBuilder.build().getMaterializedRows()); assertEqualsIgnoreOrder(result.getMaterializedRows(), resultBuilder.build().getMaterializedRows());
resultBuilder.rows(CREATE_TABLE_PARTITIONED_DATA.getMaterializedRows()); MaterializedResult result = readTable(tableHandle, columnHandles, session, TupleDomain.all(), OptionalInt.empty(), Optional.of(storageFormat)); assertEqualsIgnoreOrder(result.getMaterializedRows(), resultBuilder.build().getMaterializedRows()); assertEqualsIgnoreOrder(result.getMaterializedRows(), resultBuilder.build().getMaterializedRows());
public static Builder resultBuilder(ConnectorSession session, Iterable<? extends Type> types) { return new Builder(session, ImmutableList.copyOf(types)); }
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(); }
@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 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(); }
resultBuilder.rows(CREATE_TABLE_PARTITIONED_DATA.getMaterializedRows()); MaterializedResult result = readTable(transaction, tableHandle, columnHandles, session, TupleDomain.all(), OptionalInt.empty(), Optional.of(storageFormat)); assertEqualsIgnoreOrder(result.getMaterializedRows(), resultBuilder.build().getMaterializedRows()); assertEqualsIgnoreOrder(result.getMaterializedRows(), resultBuilder.build().getMaterializedRows());
expectedResultBuilder.rows(CREATE_TABLE_PARTITIONED_DATA.getMaterializedRows()); List<ColumnHandle> columnHandles = filterNonHiddenColumnHandles(metadata.getColumnHandles(session, tableHandle).values()); MaterializedResult result = readTable(transaction, tableHandle, columnHandles, session, TupleDomain.all(), OptionalInt.empty(), Optional.of(storageFormat)); assertEqualsIgnoreOrder(result.getMaterializedRows(), expectedResultBuilder.build().getMaterializedRows()); ImmutableList<MaterializedRow> expectedRows = expectedResultBuilder.build().getMaterializedRows().stream() .filter(row -> !"2015-07-03".equals(row.getField(dsColumnOrdinalPosition))) .collect(toImmutableList());
resultBuilder.rows(CREATE_TABLE_DATA.getMaterializedRows()); MaterializedResult result = readTable(transaction, tableHandle, columnHandles, session, TupleDomain.all(), OptionalInt.empty(), Optional.empty()); assertEqualsIgnoreOrder(result.getMaterializedRows(), resultBuilder.build().getMaterializedRows()); List<ColumnHandle> columnHandles = filterNonHiddenColumnHandles(metadata.getColumnHandles(session, tableHandle).values()); MaterializedResult result = readTable(transaction, tableHandle, columnHandles, session, TupleDomain.all(), OptionalInt.empty(), Optional.empty()); assertEqualsIgnoreOrder(result.getMaterializedRows(), resultBuilder.build().getMaterializedRows());
.row(110L, "a", "alter1") .row(120L, "a", "insert1") .row(140L, "a", "drop1") .row(210L, "b", "drop2") .row(310L, "c", "alter2") .row(320L, "c", "alter3") .row(510L, "e", "drop3") .row(610L, "f", "insert2") .row(620L, "f", "insert3") .build(); Domain domainToDrop = Domain.create(ValueSet.of( createUnboundedVarcharType(), if (allowInsertExisting) { extraRowsForInsertExisting = MaterializedResult.resultBuilder(SESSION, BIGINT, createUnboundedVarcharType(), createUnboundedVarcharType()) .row(121L, "a", "insert1") .row(611L, "f", "insert2") .row(621L, "f", "insert3") .build() .getMaterializedRows(); .row(111L, "a", "alter1") .row(131L, "a", "add1") .row(221L, "b", "add2") .row(311L, "c", "alter2") .row(321L, "c", "alter3") .row(411L, "d", "add3") .rows(extraRowsForInsertExisting) .build();