Search in sources :

Example 1 with UnnestClause

use of org.apache.asterix.lang.sqlpp.clause.UnnestClause in project asterixdb by apache.

the class SqlppCloneAndSubstituteVariablesVisitor method visit.

@Override
public Pair<ILangExpression, VariableSubstitutionEnvironment> visit(UnnestClause unnestClause, VariableSubstitutionEnvironment env) throws CompilationException {
    VariableExpr rightVar = unnestClause.getRightVariable();
    VariableExpr newRightVar = generateNewVariable(context, rightVar);
    VariableExpr newRightPosVar = unnestClause.hasPositionalVariable() ? generateNewVariable(context, unnestClause.getPositionalVariable()) : null;
    // Visits the right expression.
    Expression rightExpr = (Expression) visitUnnesBindingExpression(unnestClause.getRightExpression(), env).first;
    // Visits the condition.
    VariableSubstitutionEnvironment currentEnv = new VariableSubstitutionEnvironment(env);
    currentEnv.removeSubstitution(newRightVar);
    if (newRightPosVar != null) {
        currentEnv.removeSubstitution(newRightPosVar);
    }
    // The condition can refer to the newRightVar and newRightPosVar.
    UnnestClause newJoinClause = new UnnestClause(unnestClause.getJoinType(), rightExpr, newRightVar, newRightPosVar);
    return new Pair<>(newJoinClause, currentEnv);
}
Also used : VariableSubstitutionEnvironment(org.apache.asterix.lang.common.rewrites.VariableSubstitutionEnvironment) ILangExpression(org.apache.asterix.lang.common.base.ILangExpression) CaseExpression(org.apache.asterix.lang.sqlpp.expression.CaseExpression) Expression(org.apache.asterix.lang.common.base.Expression) SelectExpression(org.apache.asterix.lang.sqlpp.expression.SelectExpression) VariableExpr(org.apache.asterix.lang.common.expression.VariableExpr) UnnestClause(org.apache.asterix.lang.sqlpp.clause.UnnestClause) Pair(org.apache.hyracks.algebricks.common.utils.Pair)

Example 2 with UnnestClause

use of org.apache.asterix.lang.sqlpp.clause.UnnestClause in project asterixdb by apache.

the class DeepCopyVisitor method visit.

@Override
public UnnestClause visit(UnnestClause unnestClause, Void arg) throws CompilationException {
    Expression rightExpression = (Expression) unnestClause.getRightExpression().accept(this, arg);
    VariableExpr rightVar = (VariableExpr) unnestClause.getRightVariable().accept(this, arg);
    VariableExpr rightPositionVar = unnestClause.getPositionalVariable() == null ? null : (VariableExpr) unnestClause.getPositionalVariable().accept(this, arg);
    return new UnnestClause(unnestClause.getJoinType(), rightExpression, rightVar, rightPositionVar);
}
Also used : ILangExpression(org.apache.asterix.lang.common.base.ILangExpression) Expression(org.apache.asterix.lang.common.base.Expression) SelectExpression(org.apache.asterix.lang.sqlpp.expression.SelectExpression) CaseExpression(org.apache.asterix.lang.sqlpp.expression.CaseExpression) QuantifiedExpression(org.apache.asterix.lang.common.expression.QuantifiedExpression) VariableExpr(org.apache.asterix.lang.common.expression.VariableExpr) UnnestClause(org.apache.asterix.lang.sqlpp.clause.UnnestClause)

Aggregations

Expression (org.apache.asterix.lang.common.base.Expression)2 ILangExpression (org.apache.asterix.lang.common.base.ILangExpression)2 VariableExpr (org.apache.asterix.lang.common.expression.VariableExpr)2 UnnestClause (org.apache.asterix.lang.sqlpp.clause.UnnestClause)2 CaseExpression (org.apache.asterix.lang.sqlpp.expression.CaseExpression)2 SelectExpression (org.apache.asterix.lang.sqlpp.expression.SelectExpression)2 QuantifiedExpression (org.apache.asterix.lang.common.expression.QuantifiedExpression)1 VariableSubstitutionEnvironment (org.apache.asterix.lang.common.rewrites.VariableSubstitutionEnvironment)1 Pair (org.apache.hyracks.algebricks.common.utils.Pair)1