Best Java code snippets using com.facebook.presto.spi.SortingProperty (Showing top 20 results out of 315)

/**
 * Returns Optional.empty() if the column could not be translated
 */
@Override
public <T> Optional<LocalProperty<T>> translate(Function<E, Optional<T>> translator)
{
  return translator.apply(column)
      .map(translated -> new SortingProperty<>(translated, order));
}

private static SortingProperty<String> sorted(String column, SortOrder order)
{
  return new SortingProperty<>(column, order);
}

public Builder sorted(String column, SortOrder order)
{
  properties.add(new SortingProperty<>(column, order));
  return this;
}

private static SortingProperty<Symbol> sorted(String column, SortOrder order)
{
  return new SortingProperty<>(symbol(column), order);
}

@Override
public ActualProperties visitTopNRowNumber(TopNRowNumberNode node, List<ActualProperties> inputProperties)
{
  ActualProperties properties = Iterables.getOnlyElement(inputProperties);
  ImmutableList.Builder<LocalProperty<Symbol>> localProperties = ImmutableList.builder();
  localProperties.add(new GroupingProperty<>(node.getPartitionBy()));
  for (Symbol column : node.getOrderingScheme().getOrderBy()) {
    localProperties.add(new SortingProperty<>(column, node.getOrderingScheme().getOrdering(column)));
  }
  return ActualProperties.builderFrom(properties)
      .local(localProperties.build())
      .build();
}

@Override
public ActualProperties visitTopN(TopNNode node, List<ActualProperties> inputProperties)
{
  ActualProperties properties = Iterables.getOnlyElement(inputProperties);
  List<SortingProperty<Symbol>> localProperties = node.getOrderingScheme().getOrderBy().stream()
      .map(column -> new SortingProperty<>(column, node.getOrderingScheme().getOrdering(column)))
      .collect(toImmutableList());
  return ActualProperties.builderFrom(properties)
      .local(localProperties)
      .build();
}

@Override
public ActualProperties visitSort(SortNode node, List<ActualProperties> inputProperties)
{
  ActualProperties properties = Iterables.getOnlyElement(inputProperties);
  List<SortingProperty<Symbol>> localProperties = node.getOrderingScheme().getOrderBy().stream()
      .map(column -> new SortingProperty<>(column, node.getOrderingScheme().getOrdering(column)))
      .collect(toImmutableList());
  return ActualProperties.builderFrom(properties)
      .local(localProperties)
      .build();
}

@Override
public List<ConnectorTableLayoutResult> getTableLayouts(ConnectorSession session, ConnectorTableHandle table, Constraint<ColumnHandle> constraint, Optional<Set<ColumnHandle>> desiredColumns)
{
  MongoTableHandle tableHandle = (MongoTableHandle) table;
  Optional<Set<ColumnHandle>> partitioningColumns = Optional.empty(); //TODO: sharding key
  ImmutableList.Builder<LocalProperty<ColumnHandle>> localProperties = ImmutableList.builder();
  MongoTable tableInfo = mongoSession.getTable(tableHandle.getSchemaTableName());
  Map<String, ColumnHandle> columns = getColumnHandles(session, tableHandle);
  for (MongoIndex index : tableInfo.getIndexes()) {
    for (MongodbIndexKey key : index.getKeys()) {
      if (!key.getSortOrder().isPresent()) {
        continue;
      }
      if (columns.get(key.getName()) != null) {
        localProperties.add(new SortingProperty<>(columns.get(key.getName()), key.getSortOrder().get()));
      }
    }
  }
  ConnectorTableLayout layout = new ConnectorTableLayout(
      new MongoTableLayoutHandle(tableHandle, constraint.getSummary()),
      Optional.empty(),
      TupleDomain.all(),
      Optional.empty(),
      partitioningColumns,
      Optional.empty(),
      localProperties.build());
  return ImmutableList.of(new ConnectorTableLayoutResult(layout, constraint.getSummary()));
}

.map(column -> new SortingProperty<>(column, scheme.getOrdering(column)))
.forEach(localProperties::add));

@Test
public void testJsonSerialization()
    throws Exception
{
  ObjectMapper mapper = new ObjectMapperProvider().get()
      .registerModule(new SimpleModule()
          .addDeserializer(ColumnHandle.class, new JsonDeserializer<ColumnHandle>()
          {
            @Override
            public ColumnHandle deserialize(JsonParser jsonParser, DeserializationContext deserializationContext)
                throws IOException
            {
              return new ObjectMapperProvider().get().readValue(jsonParser, TestingColumnHandle.class);
            }
          }));
  TestingColumnHandle columnHandle = new TestingColumnHandle("a");
  LocalProperty<ColumnHandle> property1 = new ConstantProperty<>(columnHandle);
  assertEquals(property1, mapper.readValue(mapper.writeValueAsString(property1), new TypeReference<LocalProperty<ColumnHandle>>() {}));
  LocalProperty<ColumnHandle> property2 = new SortingProperty<>(columnHandle, SortOrder.ASC_NULLS_FIRST);
  assertEquals(property2, mapper.readValue(mapper.writeValueAsString(property2), new TypeReference<LocalProperty<ColumnHandle>>() {}));
  LocalProperty<ColumnHandle> property3 = new GroupingProperty<>(ImmutableList.of(columnHandle));
  assertEquals(property3, mapper.readValue(mapper.writeValueAsString(property3), new TypeReference<LocalProperty<ColumnHandle>>() {}));
}

        .map(symbol -> new SortingProperty<>(symbol, orderingScheme.getOrdering(symbol)))
        .forEach(desiredProperties::add));
Iterator<Optional<LocalProperty<Symbol>>> matchIterator = LocalProperties.match(child.getProperties().getLocalProperties(), desiredProperties).iterator();

    ImmutableList.of(orderKeyColumn)));
partitioningColumns = Optional.of(ImmutableSet.of(orderKeyColumn));
localProperties = ImmutableList.of(new SortingProperty<>(orderKeyColumn, SortOrder.ASC_NULLS_FIRST));
partitioningColumns = Optional.of(ImmutableSet.of(orderKeyColumn));
localProperties = ImmutableList.of(
    new SortingProperty<>(orderKeyColumn, SortOrder.ASC_NULLS_FIRST),
    new SortingProperty<>(columns.get(columnNaming.getName(LineItemColumn.LINE_NUMBER)), SortOrder.ASC_NULLS_FIRST));

if (node.getOrderingScheme().isPresent()) {
  node.getOrderingScheme().get().getOrderBy().stream()
      .map(column -> new SortingProperty<>(column, node.getOrderingScheme().get().getOrdering(column)))
      .forEach(localProperties::add);

@Override
public PlanWithProperties visitWindow(WindowNode node, PreferredProperties preferredProperties)
{
  List<LocalProperty<Symbol>> desiredProperties = new ArrayList<>();
  if (!node.getPartitionBy().isEmpty()) {
    desiredProperties.add(new GroupingProperty<>(node.getPartitionBy()));
  }
  node.getOrderingScheme().ifPresent(orderingScheme ->
      orderingScheme.getOrderBy().stream()
          .map(symbol -> new SortingProperty<>(symbol, orderingScheme.getOrdering(symbol)))
          .forEach(desiredProperties::add));
  PlanWithProperties child = planChild(
      node,
      PreferredProperties.partitionedWithLocal(ImmutableSet.copyOf(node.getPartitionBy()), desiredProperties)
          .mergeWithParent(preferredProperties));
  if (!child.getProperties().isStreamPartitionedOn(node.getPartitionBy()) &&
      !child.getProperties().isNodePartitionedOn(node.getPartitionBy())) {
    if (node.getPartitionBy().isEmpty()) {
      child = withDerivedProperties(
          gatheringExchange(idAllocator.getNextId(), REMOTE, child.getNode()),
          child.getProperties());
    }
    else {
      child = withDerivedProperties(
          partitionedExchange(idAllocator.getNextId(), REMOTE, child.getNode(), node.getPartitionBy(), node.getHashSymbol()),
          child.getProperties());
    }
  }
  return rebaseAndDeriveProperties(node, child);
}

desiredProperties.add(new SortingProperty<>(symbol, node.getOrderingScheme().getOrdering(symbol)));

/**
 * Returns Optional.empty() if the column could not be translated
 */
@Override
public <T> Optional<LocalProperty<T>> translate(Function<E, Optional<T>> translator)
{
  return translator.apply(column)
      .map(translated -> new SortingProperty<>(translated, order));
}

private static SortingProperty<String> sorted(String column, SortOrder order)
{
  return new SortingProperty<>(column, order);
}

public Builder sorted(String column, SortOrder order)
{
  properties.add(new SortingProperty<>(column, order));
  return this;
}

@Override
public ActualProperties visitSort(SortNode node, List<ActualProperties> inputProperties)
{
  ActualProperties properties = Iterables.getOnlyElement(inputProperties);
  List<SortingProperty<Symbol>> localProperties = node.getOrderBy().stream()
      .map(column -> new SortingProperty<>(column, node.getOrderings().get(column)))
      .collect(toImmutableList());
  return ActualProperties.builderFrom(properties)
      .local(localProperties)
      .build();
}

@Override
public ActualProperties visitTopN(TopNNode node, List<ActualProperties> inputProperties)
{
  ActualProperties properties = Iterables.getOnlyElement(inputProperties);
  List<SortingProperty<Symbol>> localProperties = node.getOrderBy().stream()
      .map(column -> new SortingProperty<>(column, node.getOrderings().get(column)))
      .collect(toImmutableList());
  return ActualProperties.builderFrom(properties)
      .local(localProperties)
      .build();
}