/** Get variable name assigned in INTO or var=col clause by index */ String getIntoVariable(HplsqlParser.Select_stmtContext ctx, int idx) { HplsqlParser.Into_clauseContext into = getIntoClause(ctx); if (into != null) { return into.ident(idx).getText(); } HplsqlParser.Select_list_itemContext sl = ctx.fullselect_stmt() .fullselect_stmt_item(0) .subselect_stmt() .select_list() .select_list_item(idx); if (sl != null) { return sl.ident().getText(); } return null; }
/** Get number of elements in INTO or var=col assignment clause */ int getIntoCount(HplsqlParser.Select_stmtContext ctx) { HplsqlParser.Into_clauseContext into = getIntoClause(ctx); if (into != null) { return into.ident().size(); } List<HplsqlParser.Select_list_itemContext> sl = ctx.fullselect_stmt() .fullselect_stmt_item(0) .subselect_stmt() .select_list() .select_list_item(); if (sl.get(0).T_EQUAL() != null) { return sl.size(); } return 0; }