@Override public void setBody(Expression body) { query.setBody(body); }
@Override public Expression visit(Query q, ILangExpression arg) throws CompilationException { q.setBody(visit(q.getBody(), q)); return null; }
@Override public Expression visit(Query q, ILangExpression arg) throws CompilationException { q.setBody(visit(q.getBody(), q)); return null; }
@Override public Boolean visit(Query q, List<FunctionDecl> arg) throws CompilationException { Pair<Boolean, Expression> p = inlineUdfsInExpr(q.getBody(), arg); q.setBody(p.second); return p.first; }
final public Query Query() throws ParseException, ParseException { Query query = new Query(false); Expression expr; expr = Expression(); query.setBody(expr); query.setVarCounter(getVarCounter()); {if (true) return query;} throw new Error("Missing return statement in function"); }
final public Query Query() throws ParseException, ParseException { Query query = new Query(false); Expression expr; expr = Expression(); query.setBody(expr); query.setVarCounter(getVarCounter()); {if (true) return query;} throw new Error("Missing return statement in function"); }
@Override public Expression visit(Query q, ILangExpression arg) throws CompilationException { q.setBody(visit(q.getBody(), q)); q.setVarCounter(scopeChecker.getVarCounter()); return null; }
/** * Substitutes expression with replacement expressions according to the exprMap. * * @param expression * , * an input expression. * @param exprMap * a map that maps expressions to their corresponding replacement expressions. * @return an expression, where sub-expressions of the input expression (including the input expression itself) * are replaced with deep copies with their mapped replacements in the exprMap if there exists such a * replacement expression. * @throws CompilationException */ public static Expression substituteExpression(Expression expression, Map<Expression, Expression> exprMap, LangRewritingContext context) throws CompilationException { if (exprMap.isEmpty()) { return expression; } // Creates a wrapper query for the expression so that if the expression itself // is the key, it can also be replaced. Query wrapper = new Query(false); wrapper.setSourceLocation(expression.getSourceLocation()); wrapper.setBody(expression); // Creates a substitution visitor. SqlppSubstituteExpressionVisitor visitor = new SqlppSubstituteExpressionVisitor(context, exprMap); wrapper.accept(visitor, wrapper); return wrapper.getBody(); } }
@Override public Pair<ILangExpression, VariableSubstitutionEnvironment> visit(Query q, VariableSubstitutionEnvironment env) throws CompilationException { Query newQ = new Query(q.isExplain()); Pair<ILangExpression, VariableSubstitutionEnvironment> p1 = q.getBody().accept(this, env); newQ.setBody((Expression) p1.first); newQ.setSourceLocation(q.getSourceLocation()); return new Pair<>(newQ, p1.second); }
@Override public Void visit(DeleteStatement deleteStmt, Void visitArg) { List<Expression> arguments = new ArrayList<>(); Identifier dataverseName = deleteStmt.getDataverseName(); Identifier datasetName = deleteStmt.getDatasetName(); String arg = dataverseName == null ? datasetName.getValue() : dataverseName.getValue() + "." + datasetName.getValue(); LiteralExpr argumentLiteral = new LiteralExpr(new StringLiteral(arg)); arguments.add(argumentLiteral); CallExpr callExpression = new CallExpr(new FunctionSignature(BuiltinFunctions.DATASET), arguments); List<Clause> clauseList = new ArrayList<>(); VariableExpr var = deleteStmt.getVariableExpr(); Clause forClause = new ForClause(var, callExpression); clauseList.add(forClause); Clause whereClause = null; Expression condition = deleteStmt.getCondition(); if (condition != null) { whereClause = new WhereClause(condition); clauseList.add(whereClause); } VariableExpr returnExpr = new VariableExpr(var.getVar()); returnExpr.setIsNewVar(false); FLWOGRExpression flowgr = new FLWOGRExpression(clauseList, returnExpr); Query query = new Query(false); query.setBody(flowgr); deleteStmt.setQuery(query); return null; }
throw new ParseException(); query.setBody(expr); query.setSourceLocation(expr.getSourceLocation()); {if (true) return query;}
private Expression rewriteFunctionBody(FunctionDecl fnDecl) throws CompilationException { SourceLocation sourceLoc = fnDecl.getSourceLocation(); Query wrappedQuery = new Query(false); wrappedQuery.setSourceLocation(sourceLoc); wrappedQuery.setBody(fnDecl.getFuncBody()); wrappedQuery.setTopLevel(false); String fnNamespace = fnDecl.getSignature().getNamespace(); Dataverse defaultDataverse = metadataProvider.getDefaultDataverse(); Dataverse fnDataverse; if (fnNamespace == null || fnNamespace.equals(defaultDataverse.getDataverseName())) { fnDataverse = defaultDataverse; } else { try { fnDataverse = metadataProvider.findDataverse(fnNamespace); } catch (AlgebricksException e) { throw new CompilationException(ErrorCode.UNKNOWN_DATAVERSE, e, sourceLoc, fnNamespace); } } metadataProvider.setDefaultDataverse(fnDataverse); try { IQueryRewriter queryRewriter = rewriterFactory.createQueryRewriter(); queryRewriter.rewrite(declaredFunctions, wrappedQuery, metadataProvider, context, true, fnDecl.getParamList()); return wrappedQuery.getBody(); } finally { metadataProvider.setDefaultDataverse(defaultDataverse); } }
Query wrappedQuery = new Query(false); wrappedQuery.setSourceLocation(sourceLoc); wrappedQuery.setBody(cfs.getFunctionBodyExpression()); wrappedQuery.setTopLevel(false); List<VarIdentifier> paramVars = new ArrayList<>();
selectExpression.setSourceLocation(var.getSourceLocation()); Query query = new Query(false, false, selectExpression, 0); query.setBody(selectExpression); query.setSourceLocation(deleteStmt.getSourceLocation());