Examples with PINOT_UNSUPPORTED_EXPRESSION com.facebook.presto.pinot.PinotErrorCode.PINOT_UNSUPPORTED_EXPRESSION used on opensource projects

Example 1 with PINOT_UNSUPPORTED_EXPRESSION

use of com.facebook.presto.pinot.PinotErrorCode.PINOT_UNSUPPORTED_EXPRESSION in project presto by prestodb.

the class PinotQueryGeneratorContext method getIndicesMappingFromPinotSchemaToPrestoSchema.

private List<Integer> getIndicesMappingFromPinotSchemaToPrestoSchema(String query, Map<VariableReferenceExpression, PinotColumnHandle> assignments) {
    LinkedHashMap<VariableReferenceExpression, Selection> expressionsInPinotOrder = new LinkedHashMap<>();
    for (VariableReferenceExpression groupByColumn : groupByColumns) {
        Selection groupByColumnDefinition = selections.get(groupByColumn);
        if (groupByColumnDefinition == null) {
            throw new IllegalStateException(format("Group By column (%s) definition not found in input selections: %s", groupByColumn, Joiner.on(",").withKeyValueSeparator(":").join(selections)));
        }
        expressionsInPinotOrder.put(groupByColumn, groupByColumnDefinition);
    }
    for (VariableReferenceExpression outputColumn : outputs) {
        Selection outputColumnDefinition = selections.get(outputColumn);
        if (outputColumnDefinition == null) {
            throw new IllegalStateException(format("Output column (%s) definition not found in input selections: %s", outputColumn, Joiner.on(",").withKeyValueSeparator(":").join(selections)));
        }
        expressionsInPinotOrder.put(outputColumn, outputColumnDefinition);
    }
    if (useSqlSyntax) {
        checkSupported(assignments.size() <= expressionsInPinotOrder.keySet().stream().filter(key -> !hiddenColumnSet.contains(key)).count(), "Expected returned expressions %s is a superset of selections %s", Joiner.on(",").withKeyValueSeparator(":").join(expressionsInPinotOrder), Joiner.on(",").withKeyValueSeparator("=").join(assignments));
    } else {
        checkSupported(assignments.size() == expressionsInPinotOrder.keySet().stream().filter(key -> !hiddenColumnSet.contains(key)).count(), "Expected returned expressions %s to match selections %s", Joiner.on(",").withKeyValueSeparator(":").join(expressionsInPinotOrder), Joiner.on(",").withKeyValueSeparator("=").join(assignments));
    }
    Map<VariableReferenceExpression, Integer> assignmentToIndex = new HashMap<>();
    Iterator<Map.Entry<VariableReferenceExpression, PinotColumnHandle>> assignmentsIterator = assignments.entrySet().iterator();
    for (int i = 0; i < assignments.size(); i++) {
        VariableReferenceExpression key = assignmentsIterator.next().getKey();
        Integer previous = assignmentToIndex.put(key, i);
        if (previous != null) {
            throw new PinotException(PINOT_UNSUPPORTED_EXPRESSION, Optional.of(query), format("Expected Pinot column handle %s to occur only once, but we have: %s", key, Joiner.on(",").withKeyValueSeparator("=").join(assignments)));
        }
    }
    ImmutableList.Builder<Integer> outputIndices = ImmutableList.builder();
    for (Map.Entry<VariableReferenceExpression, Selection> expression : expressionsInPinotOrder.entrySet()) {
        Integer index;
        if (hiddenColumnSet.contains(expression.getKey())) {
            // negative output index means to skip this value returned by pinot at query time
            index = -1;
        } else {
            index = assignmentToIndex.get(expression.getKey());
        }
        if (index == null) {
            if (useSqlSyntax) {
                // negative output index means to skip this value returned by pinot at query time
                index = -1;
            } else {
                throw new PinotException(PINOT_UNSUPPORTED_EXPRESSION, Optional.of(query), format("Expected to find a Pinot column handle for the expression %s, but we have %s", expression, Joiner.on(",").withKeyValueSeparator(":").join(assignmentToIndex)));
            }
        }
        outputIndices.add(index);
    }
    return outputIndices.build();
}