Examples with Scope io.trino.sql.analyzer.Scope used on opensource projects

Example 6 with Scope

use of io.trino.sql.analyzer.Scope in project trino by trinodb.

the class ScopeAware method scopeAwareComparison.

private Boolean scopeAwareComparison(Node left, Node right) {
    if (left instanceof Expression && right instanceof Expression) {
        Expression leftExpression = (Expression) left;
        Expression rightExpression = (Expression) right;
        if (analysis.isColumnReference(leftExpression) && analysis.isColumnReference(rightExpression)) {
            ResolvedField leftField = analysis.getResolvedField(leftExpression);
            ResolvedField rightField = analysis.getResolvedField(rightExpression);
            Scope leftScope = leftField.getScope();
            Scope rightScope = rightField.getScope();
            // For subqueries of the query associated with the current expression, compare by syntax
            // Note: it'd appear that hash() and equal() are inconsistent with each other in the case that:
            // * left.hasOuterParent(...) == true and right.hasOuterParent(...) == false
            // * leftField.getFieldId().equals(rightField.getFieldId()) == true
            // Both fields would seem to have different hashes but be equal to each other.
            // However, this cannot happen because we *require* that both expressions being compared by
            // rooted in the same "query scope" (i.e., sub-scopes that are local to each other) -- see ScopeAwareKey.equals().
            // If both fields have the same field id, by definition they will produce the same result for hasOuterParent().
            checkState(leftScope.hasOuterParent(queryScope) == rightScope.hasOuterParent(queryScope));
            if (leftScope.hasOuterParent(queryScope) && rightScope.hasOuterParent(queryScope)) {
                return treeEqual(leftExpression, rightExpression, CanonicalizationAware::canonicalizationAwareComparison);
            }
            // expression, compare by resolved field
            return leftField.getFieldId().equals(rightField.getFieldId());
        } else if (leftExpression instanceof Identifier && rightExpression instanceof Identifier) {
            return treeEqual(leftExpression, rightExpression, CanonicalizationAware::canonicalizationAwareComparison);
        }
    }
    if (!left.shallowEquals(right)) {
        return false;
    }
    return null;
}

Example 7 with Scope

use of io.trino.sql.analyzer.Scope in project trino by trinodb.

the class LogicalPlanner method createExplainAnalyzePlan.

private RelationPlan createExplainAnalyzePlan(Analysis analysis, ExplainAnalyze statement) {
    RelationPlan underlyingPlan = planStatementWithoutOutput(analysis, statement.getStatement());
    PlanNode root = underlyingPlan.getRoot();
    Scope scope = analysis.getScope(statement);
    Symbol outputSymbol = symbolAllocator.newSymbol(scope.getRelationType().getFieldByIndex(0));
    ImmutableList.Builder<Symbol> actualOutputs = ImmutableList.builder();
    RelationType outputDescriptor = analysis.getOutputDescriptor(statement.getStatement());
    for (Field field : outputDescriptor.getVisibleFields()) {
        int fieldIndex = outputDescriptor.indexOf(field);
        Symbol symbol = underlyingPlan.getSymbol(fieldIndex);
        actualOutputs.add(symbol);
    }
    root = new ExplainAnalyzeNode(idAllocator.getNextId(), root, outputSymbol, actualOutputs.build(), statement.isVerbose());
    return new RelationPlan(root, scope, ImmutableList.of(outputSymbol), Optional.empty());
}