Examples with ColumnSelector org.springframework.data.cassandra.core.query.Columns.ColumnSelector used on opensource projects

Example 1 with ColumnSelector

use of org.springframework.data.cassandra.core.query.Columns.ColumnSelector in project spring-data-cassandra by spring-projects.

the class QueryMapper method getMappedSelector.

private Selector getMappedSelector(Selector selector, CqlIdentifier cqlIdentifier) {
    if (selector instanceof ColumnSelector) {
        ColumnSelector columnSelector = (ColumnSelector) selector;
        ColumnSelector mappedColumnSelector = ColumnSelector.from(cqlIdentifier);
        return columnSelector.getAlias().map(mappedColumnSelector::as).orElse(mappedColumnSelector);
    }
    if (selector instanceof FunctionCall) {
        FunctionCall functionCall = (FunctionCall) selector;
        FunctionCall mappedFunctionCall = FunctionCall.from(functionCall.getExpression(), functionCall.getParameters().stream().map(obj -> {
            if (obj instanceof Selector) {
                return getMappedSelector((Selector) obj, cqlIdentifier);
            }
            return obj;
        }).toArray());
        return // 
        functionCall.getAlias().map(// 
        mappedFunctionCall::as).orElse(mappedFunctionCall);
    }
    throw new IllegalArgumentException(String.format("Selector [%s] not supported", selector));
}

Example 2 with ColumnSelector

use of org.springframework.data.cassandra.core.query.Columns.ColumnSelector in project spring-data-cassandra by spring-projects.

the class QueryMapper method getMappedSelectors.

/**
 * Map {@link Columns} with a {@link CassandraPersistentEntity type hint} to {@link ColumnSelector}s.
 *
 * @param columns must not be {@literal null}.
 * @param entity must not be {@literal null}.
 * @return the mapped {@link Selector}s.
 */
public List<Selector> getMappedSelectors(Columns columns, CassandraPersistentEntity<?> entity) {
    Assert.notNull(columns, "Columns must not be null");
    Assert.notNull(entity, "CassandraPersistentEntity must not be null");
    if (columns.isEmpty()) {
        return Collections.emptyList();
    }
    List<Selector> selectors = new ArrayList<>();
    for (ColumnName column : columns) {
        Field field = createPropertyField(entity, column);
        columns.getSelector(column).ifPresent(selector -> {
            List<CqlIdentifier> mappedColumnNames = getCqlIdentifier(column, field);
            for (CqlIdentifier mappedColumnName : mappedColumnNames) {
                selectors.add(getMappedSelector(selector, mappedColumnName));
            }
        });
    }
    if (columns.isEmpty()) {
        addColumns(entity, selectors);
    }
    return selectors;
}

Example 3 with ColumnSelector

use of org.springframework.data.cassandra.core.query.Columns.ColumnSelector in project spring-data-cassandra by spring-projects.

the class QueryMapper method getMappedColumnNames.

/**
 * Map {@link Columns} with a {@link CassandraPersistentEntity type hint} to column names for included columns.
 * Function call selectors or other {@link org.springframework.data.cassandra.core.query.Columns.Selector} types are
 * not included.
 *
 * @param columns must not be {@literal null}.
 * @param entity must not be {@literal null}.
 * @return the mapped column names.
 */
public List<CqlIdentifier> getMappedColumnNames(Columns columns, CassandraPersistentEntity<?> entity) {
    Assert.notNull(columns, "Columns must not be null");
    Assert.notNull(entity, "CassandraPersistentEntity must not be null");
    if (columns.isEmpty()) {
        return Collections.emptyList();
    }
    List<CqlIdentifier> columnNames = new ArrayList<>();
    Set<PersistentProperty<?>> seen = new HashSet<>();
    for (ColumnName column : columns) {
        Field field = createPropertyField(entity, column);
        field.getProperty().ifPresent(seen::add);
        columns.getSelector(column).filter(selector -> selector instanceof ColumnSelector).ifPresent(columnSelector -> columnNames.addAll(getCqlIdentifier(column, field)));
    }
    if (columns.isEmpty()) {
        entity.doWithProperties((PropertyHandler<CassandraPersistentProperty>) property -> {
            if (property.isCompositePrimaryKey()) {
                return;
            }
            if (seen.add(property)) {
                columnNames.add(property.getRequiredColumnName());
            }
        });
    }
    return columnNames;
}