/** Create a temporary table statement */ public Integer createTemporaryTable( ParserRuleContext ctx, Create_table_columnsContext colCtx, String name) { String managedName = null; String sql = null; String columns = exec.getFormattedText(colCtx); if (conf.tempTables == Conf.TempTables.NATIVE) { sql = "CREATE TEMPORARY TABLE " + name + "\n(" + columns + "\n)"; } else if (conf.tempTables == Conf.TempTables.MANAGED) { managedName = name + "_" + UUID.randomUUID().toString().replace("-", ""); if (!conf.tempTablesSchema.isEmpty()) { managedName = conf.tempTablesSchema + "." + managedName; } sql = "CREATE TABLE " + managedName + "\n(" + columns + "\n)"; if (!conf.tempTablesLocation.isEmpty()) { sql += "\nLOCATION '" + conf.tempTablesLocation + "/" + managedName + "'"; } if (trace) { trace(ctx, "Managed table name: " + managedName); } } if (sql != null) { Query query = exec.executeSql(ctx, sql, exec.conf.defaultConnection); if (query.error()) { exec.signal(query); return 1; } if (managedName != null) { exec.addManagedTable(name, managedName); } exec.setSqlSuccess(); exec.closeQuery(query, exec.conf.defaultConnection); } return 0; }
/** Binary boolean expression in executable SQL statement */ public Integer execBoolBinarySql(HplsqlParser.Bool_expr_binaryContext ctx) { StringBuilder sql = new StringBuilder(); sql.append(evalPop(ctx.expr(0)).toString()); sql.append(" " + exec.getFormattedText(ctx.bool_expr_binary_operator()) + " "); sql.append(evalPop(ctx.expr(1)).toString()); exec.stackPush(sql); return 0; }
/** SQL INSERT VALUES statement */ public Integer insertValues(HplsqlParser.Insert_stmtContext ctx) { trace(ctx, "INSERT VALUES"); String table = evalPop(ctx.table_name()).toString(); String conn = exec.getObjectConnection(ctx.table_name().getText()); Conn.Type type = exec.getConnectionType(conn); StringBuilder sql = new StringBuilder(); if (type == Conn.Type.HIVE) { sql.append("INSERT INTO TABLE " + table + " "); if (conf.insertValues == Conf.InsertValues.NATIVE) { sql.append("VALUES\n("); } } else { sql.append("INSERT INTO " + table); if (ctx.insert_stmt_cols() != null) { sql.append(" " + exec.getFormattedText(ctx.insert_stmt_cols())); } sql.append(" VALUES\n("); } int rows = ctx.insert_stmt_rows().insert_stmt_row().size(); for (int i = 0; i < rows; i++) { HplsqlParser.Insert_stmt_rowContext row = ctx.insert_stmt_rows().insert_stmt_row(i); int cols = row.expr().size(); for (int j = 0; j < cols; j++) { String value = evalPop(row.expr(j)).toSqlString(); if (j == 0 && type == Conn.Type.HIVE && conf.insertValues == Conf.InsertValues.SELECT) { sql.append("SELECT "); } sql.append(value); if (j + 1 != cols) { sql.append(", "); } } if (type != Conn.Type.HIVE || conf.insertValues == Conf.InsertValues.NATIVE) { if (i + 1 == rows) { sql.append(")"); } else { sql.append("),\n("); } } else if (type == Conn.Type.HIVE && conf.insertValues == Conf.InsertValues.SELECT) { sql.append(" FROM " + conf.dualTable); if (i + 1 < rows) { sql.append("\nUNION ALL\n"); } } } if (trace) { trace(ctx, sql.toString()); } Query query = exec.executeSql(ctx, sql.toString(), conn); if (query.error()) { exec.signal(query); return 1; } exec.setSqlSuccess(); exec.closeQuery(query, exec.conf.defaultConnection); return 0; }
/** MERGE statement */ public Integer merge(HplsqlParser.Merge_stmtContext ctx) { trace(ctx, "MERGE"); String sql = exec.getFormattedText(ctx); trace(ctx, sql); Query query = exec.executeSql(ctx, sql, exec.conf.defaultConnection); if (query.error()) { exec.signal(query); return 1; } exec.setSqlSuccess(); exec.closeQuery(query, exec.conf.defaultConnection); return 0; }
/** Common table expression (WITH clause) */ public Integer cte(HplsqlParser.Cte_select_stmtContext ctx) { int cnt = ctx.cte_select_stmt_item().size(); StringBuilder sql = new StringBuilder(); sql.append("WITH "); for (int i = 0; i < cnt; i++) { HplsqlParser.Cte_select_stmt_itemContext c = ctx.cte_select_stmt_item(i); sql.append(c.ident().getText()); if (c.cte_select_cols() != null) { sql.append(" " + exec.getFormattedText(c.cte_select_cols())); } sql.append(" AS ("); sql.append(evalPop(ctx.cte_select_stmt_item(i).fullselect_stmt()).toString()); sql.append(")"); if (i + 1 != cnt) { sql.append(",\n"); } } exec.stackPush(sql); return 0; }
/** Evaluate an expression in executable SQL statement */ public void execSql(HplsqlParser.ExprContext ctx) { StringBuilder sql = new StringBuilder(); if (ctx.T_OPEN_P() != null) { sql.append("("); if (ctx.select_stmt() != null) { exec.append( sql, evalPop(ctx.select_stmt()).toString(), ctx.T_OPEN_P().getSymbol(), ctx.select_stmt().getStart()); exec.append( sql, ctx.T_CLOSE_P().getText(), ctx.select_stmt().stop, ctx.T_CLOSE_P().getSymbol()); } else { sql.append(evalPop(ctx.expr(0)).toString()); sql.append(")"); } } else if (ctx.T_MUL() != null) { sql.append(evalPop(ctx.expr(0)).toString()); sql.append(" * "); sql.append(evalPop(ctx.expr(1)).toString()); } else if (ctx.T_DIV() != null) { sql.append(evalPop(ctx.expr(0)).toString()); sql.append(" / "); sql.append(evalPop(ctx.expr(1)).toString()); } else if (ctx.T_ADD() != null) { sql.append(evalPop(ctx.expr(0)).toString()); sql.append(" + "); sql.append(evalPop(ctx.expr(1)).toString()); } else if (ctx.T_SUB() != null) { sql.append(evalPop(ctx.expr(0)).toString()); sql.append(" - "); sql.append(evalPop(ctx.expr(1)).toString()); } else if (ctx.interval_item() != null) { sql.append(exec.getFormattedText(ctx)); } else { visitChildren(ctx); sql.append(exec.stackPop().toString()); } exec.stackPush(sql); }