You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tajo.apache.org by hy...@apache.org on 2014/04/18 13:44:57 UTC
[54/57] [abbrv] Merge branch 'master' of
https://git-wip-us.apache.org/repos/asf/tajo into window_function
http://git-wip-us.apache.org/repos/asf/tajo/blob/1b3ec373/tajo-core/src/main/java/org/apache/tajo/engine/parser/SQLAnalyzer.java
----------------------------------------------------------------------
diff --cc tajo-core/src/main/java/org/apache/tajo/engine/parser/SQLAnalyzer.java
index 0000000,cb356f8..14a383d
mode 000000,100644..100644
--- a/tajo-core/src/main/java/org/apache/tajo/engine/parser/SQLAnalyzer.java
+++ b/tajo-core/src/main/java/org/apache/tajo/engine/parser/SQLAnalyzer.java
@@@ -1,0 -1,1516 +1,1661 @@@
+ /**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+ package org.apache.tajo.engine.parser;
+
+ import com.google.common.base.Preconditions;
+ import com.google.common.collect.Lists;
+ import org.antlr.v4.runtime.ANTLRInputStream;
+ import org.antlr.v4.runtime.CommonTokenStream;
+ import org.antlr.v4.runtime.misc.NotNull;
+ import org.antlr.v4.runtime.tree.TerminalNode;
+ import org.apache.commons.lang.StringEscapeUtils;
+ import org.apache.tajo.algebra.*;
+ import org.apache.tajo.algebra.Aggregation.GroupType;
+ import org.apache.tajo.algebra.LiteralValue.LiteralType;
+ import org.apache.tajo.catalog.CatalogUtil;
+ import org.apache.tajo.engine.parser.SQLParser.*;
+ import org.apache.tajo.storage.StorageConstants;
+
+ import java.nio.charset.Charset;
+ import java.util.*;
+
+ import static org.apache.tajo.algebra.Aggregation.GroupElement;
+ import static org.apache.tajo.algebra.CreateTable.*;
++import static org.apache.tajo.algebra.WindowSpecExpr.WindowFrameEndBoundType;
++import static org.apache.tajo.algebra.WindowSpecExpr.WindowFrameStartBoundType;
+ import static org.apache.tajo.common.TajoDataTypes.Type;
+ import static org.apache.tajo.engine.parser.SQLParser.*;
+
+ public class SQLAnalyzer extends SQLParserBaseVisitor<Expr> {
+ private SQLParser parser;
+
+ public SQLAnalyzer() {
+ }
+
+ public Expr parse(String sql) {
+ ANTLRInputStream input = new ANTLRInputStream(sql);
+ SQLLexer lexer = new SQLLexer(input);
+ CommonTokenStream tokens = new CommonTokenStream(lexer);
+ this.parser = new SQLParser(tokens);
+ parser.setBuildParseTree(true);
+ parser.removeErrorListeners();
+
+ parser.setErrorHandler(new SQLErrorStrategy());
+ parser.addErrorListener(new SQLErrorListener());
+
+ SqlContext context;
+ try {
+ context = parser.sql();
+ } catch (SQLParseError e) {
+ throw new SQLSyntaxError(e);
+ }
+ return visitSql(context);
+ }
+
+ private static boolean checkIfExist(Object obj) {
+ return obj != null;
+ }
+
+ @Override
+ public Expr visitSql(SqlContext ctx) {
+ Expr statement = visit(ctx.statement());
+ if (checkIfExist(ctx.explain_clause())) {
+ return new Explain(statement);
+ } else {
+ return statement;
+ }
+ }
+
+ @Override
+ public Expr visitNon_join_query_expression(SQLParser.Non_join_query_expressionContext ctx) {
+
+ Expr current = visitNon_join_query_term(ctx.non_join_query_term());
+ if (ctx.getChildCount() == 1) {
+ return current;
+ }
+
+ OpType operatorType;
+ Expr left;
+ for (int i = 1; i < ctx.getChildCount(); i++) {
+ int idx = i;
+ boolean distinct = true;
+
+ if (ctx.getChild(idx) instanceof TerminalNode) {
+ if (((TerminalNode) ctx.getChild(idx)).getSymbol().getType() == UNION) {
+ operatorType = OpType.Union;
+ } else {
+ operatorType = OpType.Except;
+ }
+
+ idx++;
+
+ if (ctx.getChild(idx) instanceof TerminalNode) {
+ if (((TerminalNode) ctx.getChild(idx)).getSymbol().getType() == ALL) {
+ distinct = false;
+ }
+
+ idx++;
+ }
+
+ SQLParser.Query_termContext queryTermContext =
+ (SQLParser.Query_termContext) ctx.getChild(idx);
+ Expr right = visitQuery_term(queryTermContext);
+
+ left = current;
+ current = new SetOperation(operatorType, left, right, distinct);
+
+ i = idx;
+ }
+ }
+
+ return current;
+ }
+
+ @Override
+ public Expr visitNon_join_query_term(Non_join_query_termContext ctx) {
+
+ Expr current = visitNon_join_query_primary(ctx.non_join_query_primary());
+ Expr left;
+
+ for (int i = 1; i < ctx.getChildCount(); ) {
+ int idx = i;
+ boolean distinct = true;
+
+ if (ctx.getChild(idx) instanceof TerminalNode) {
+ if (((TerminalNode) ctx.getChild(idx)).getSymbol().getType() == INTERSECT) {
+ idx++;
+ }
+
+ if (ctx.getChild(idx) instanceof TerminalNode) {
+ if (((TerminalNode) ctx.getChild(idx)).getSymbol().getType() == ALL) {
+ distinct = false;
+ idx++;
+ }
+ }
+
+ Query_primaryContext queryPrimaryContext = (Query_primaryContext) ctx.getChild(idx);
+ Expr right = visitQuery_primary(queryPrimaryContext);
+
+ left = current;
+ current = new SetOperation(OpType.Intersect, left, right, distinct);
+
+ i += idx;
+ }
+ }
+
+ return current;
+
+ }
+
+ @Override
+ public Expr visitQuery_specification(SQLParser.Query_specificationContext ctx) {
+ Expr current = null;
+ if (ctx.table_expression() != null) {
+ current = visitFrom_clause(ctx.table_expression().from_clause());
+
+ if (ctx.table_expression().where_clause() != null) {
+ Selection selection = visitWhere_clause(ctx.table_expression().where_clause());
+ selection.setChild(current);
+ current = selection;
+ }
+
+ if (ctx.table_expression().groupby_clause() != null) {
+ Aggregation aggregation = visitGroupby_clause(ctx.table_expression().groupby_clause());
+ aggregation.setChild(current);
+ current = aggregation;
+
+ if (ctx.table_expression().having_clause() != null) {
+ Expr havingCondition = visitBoolean_value_expression(
+ ctx.table_expression().having_clause().boolean_value_expression());
+ Having having = new Having(havingCondition);
+ having.setChild(current);
+ current = having;
+ }
+ }
+
+ if (ctx.table_expression().orderby_clause() != null) {
+ Sort sort = visitOrderby_clause(ctx.table_expression().orderby_clause());
+ sort.setChild(current);
+ current = sort;
+ }
+
++ if (checkIfExist(ctx.table_expression().window_clause())) {
++ Window window = visitWindow_clause(ctx.table_expression().window_clause());
++ window.setChild(current);
++ current = window;
++ }
++
+ if (ctx.table_expression().limit_clause() != null) {
+ Limit limit = visitLimit_clause(ctx.table_expression().limit_clause());
+ limit.setChild(current);
+ current = limit;
+ }
+ }
+
+ Projection projection = visitSelect_list(ctx.select_list());
+
+ if (ctx.set_qualifier() != null && ctx.set_qualifier().DISTINCT() != null) {
+ projection.setDistinct();
+ }
+
+ if (current != null) {
+ projection.setChild(current);
+ }
+
+ current = projection;
+
+ return current;
+ }
+
+ /**
+ * <pre>
+ * select_list
+ * : select_sublist (COMMA select_sublist)*
+ * ;
+ * </pre>
+ *
+ * @param ctx
+ * @return
+ */
+ @Override
+ public Projection visitSelect_list(SQLParser.Select_listContext ctx) {
+ Projection projection = new Projection();
+ NamedExpr[] targets = new NamedExpr[ctx.select_sublist().size()];
+ for (int i = 0; i < targets.length; i++) {
+ targets[i] = visitSelect_sublist(ctx.select_sublist(i));
+ }
+ projection.setNamedExprs(targets);
+
+ return projection;
+ }
+
+ /**
+ * <pre>
+ * select_sublist
+ * : derived_column
+ * | asterisked_qualifier
+ * ;
+ * </pre>
+ *
+ * @param ctx
+ * @return
+ */
+ @Override
+ public NamedExpr visitSelect_sublist(SQLParser.Select_sublistContext ctx) {
+ if (ctx.qualified_asterisk() != null) {
+ return visitQualified_asterisk(ctx.qualified_asterisk());
+ } else {
+ return visitDerived_column(ctx.derived_column());
+ }
+ }
+
+ @Override
+ public RelationList visitFrom_clause(SQLParser.From_clauseContext ctx) {
+ Expr[] relations = new Expr[ctx.table_reference_list().table_reference().size()];
+ for (int i = 0; i < relations.length; i++) {
+ relations[i] = visitTable_reference(ctx.table_reference_list().table_reference(i));
+ }
+ return new RelationList(relations);
+ }
+
+ @Override
+ public Selection visitWhere_clause(SQLParser.Where_clauseContext ctx) {
+ return new Selection(visitSearch_condition(ctx.search_condition()));
+ }
+
+ @Override
+ public Aggregation visitGroupby_clause(SQLParser.Groupby_clauseContext ctx) {
+ Aggregation clause = new Aggregation();
+
+ // If grouping group is not empty
+ if (ctx.grouping_element_list().grouping_element().get(0).empty_grouping_set() == null) {
+ int elementSize = ctx.grouping_element_list().grouping_element().size();
+ ArrayList<GroupElement> groups = new ArrayList<GroupElement>(elementSize + 1);
+ ArrayList<Expr> ordinaryExprs = null;
+ int groupSize = 1;
+ groups.add(null);
+
+ for (int i = 0; i < elementSize; i++) {
+ SQLParser.Grouping_elementContext element =
+ ctx.grouping_element_list().grouping_element().get(i);
+ if (element.ordinary_grouping_set() != null) {
+ if (ordinaryExprs == null) {
+ ordinaryExprs = new ArrayList<Expr>();
+ }
+ Collections.addAll(ordinaryExprs, getRowValuePredicandsFromOrdinaryGroupingSet(element.ordinary_grouping_set()));
+ } else if (element.rollup_list() != null) {
+ groupSize++;
+ groups.add(new GroupElement(GroupType.Rollup,
+ getRowValuePredicandsFromOrdinaryGroupingSetList(element.rollup_list().c)));
+ } else if (element.cube_list() != null) {
+ groupSize++;
+ groups.add(new GroupElement(GroupType.Cube,
+ getRowValuePredicandsFromOrdinaryGroupingSetList(element.cube_list().c)));
+ }
+ }
+
+ if (ordinaryExprs != null) {
+ groups.set(0, new GroupElement(GroupType.OrdinaryGroup, ordinaryExprs.toArray(new Expr[ordinaryExprs.size()])));
+ clause.setGroups(groups.subList(0, groupSize).toArray(new GroupElement[groupSize]));
+ } else if (groupSize > 1) {
+ clause.setGroups(groups.subList(1, groupSize).toArray(new GroupElement[groupSize - 1]));
+ }
+ }
+
+ return clause;
+ }
+
++ @Override public WindowFunctionExpr visitWindow_function(@NotNull SQLParser.Window_functionContext context) {
++ WindowFunctionExpr windowFunction = null;
++
++ Window_function_typeContext functionType = context.window_function_type();
++ GeneralSetFunctionExpr functionBody;
++ if (checkIfExist(functionType.rank_function_type())) {
++ Rank_function_typeContext rankFunction = functionType.rank_function_type();
++ if (checkIfExist(rankFunction.RANK())) {
++ functionBody = new GeneralSetFunctionExpr("rank", false, new Expr[] {});
++ } else if (checkIfExist(rankFunction.DENSE_RANK())) {
++ functionBody = new GeneralSetFunctionExpr("dense_rank", false, new Expr[] {});
++ } else if (checkIfExist(rankFunction.PERCENT_RANK())) {
++ functionBody = new GeneralSetFunctionExpr("percent_rank", false, new Expr[] {});
++ } else {
++ functionBody = new GeneralSetFunctionExpr("cume_dist", false, new Expr[] {});
++ }
++ } else if (checkIfExist(functionType.ROW_NUMBER())) {
++ functionBody = new GeneralSetFunctionExpr("row_number", false, new Expr[] {});
++ } else {
++ functionBody = visitAggregate_function(functionType.aggregate_function());
++ }
++ windowFunction = new WindowFunctionExpr(functionBody);
++
++ Window_name_or_specificationContext windowNameOrSpec = context.window_name_or_specification();
++ if (checkIfExist(windowNameOrSpec.window_name())) {
++ windowFunction.setWindowName(windowNameOrSpec.window_name().getText());
++ } else {
++ windowFunction.setWindowSpec(visitWindow_specification(windowNameOrSpec.window_specification()));
++ }
++
++ return windowFunction;
++ }
++
+ @Override
- public Sort visitOrderby_clause(SQLParser.Orderby_clauseContext ctx) {
- int size = ctx.sort_specifier_list().sort_specifier().size();
++ public Window visitWindow_clause(@NotNull SQLParser.Window_clauseContext ctx) {
++ Window.WindowDefinition [] definitions =
++ new Window.WindowDefinition[ctx.window_definition_list().window_definition().size()];
++ for (int i = 0; i < definitions.length; i++) {
++ Window_definitionContext windowDefinitionContext = ctx.window_definition_list().window_definition(i);
++ String windowName = windowDefinitionContext.window_name().identifier().getText();
++ WindowSpecExpr windowSpec = visitWindow_specification(windowDefinitionContext.window_specification());
++ definitions[i] = new Window.WindowDefinition(windowName, windowSpec);
++ }
++ return new Window(definitions);
++ }
++
++ @Override public WindowSpecExpr visitWindow_specification(@NotNull SQLParser.Window_specificationContext ctx) {
++ WindowSpecExpr windowSpec = new WindowSpecExpr();
++ if (checkIfExist(ctx.window_specification_details())) {
++ Window_specification_detailsContext windowSpecDetail = ctx.window_specification_details();
++
++ if (checkIfExist(windowSpecDetail.existing_window_name())) {
++ windowSpec.setWindowName(windowSpecDetail.existing_window_name().getText());
++ }
++
++ if (checkIfExist(windowSpecDetail.window_partition_clause())) {
++ windowSpec.setPartitionKeys(
++ buildRowValuePredicands(windowSpecDetail.window_partition_clause().row_value_predicand_list()));
++ }
++
++ if (checkIfExist(windowSpecDetail.window_order_clause())) {
++ windowSpec.setSortSpecs(
++ buildSortSpecs(windowSpecDetail.window_order_clause().orderby_clause().sort_specifier_list()));
++ }
++
++ if (checkIfExist(windowSpecDetail.window_frame_clause())) {
++ Window_frame_clauseContext frameContext = windowSpecDetail.window_frame_clause();
++
++ WindowSpecExpr.WindowFrameUnit unit;
++ // frame unit - there are only two cases: RANGE and ROW
++ if (checkIfExist(frameContext.window_frame_units().RANGE())) {
++ unit = WindowSpecExpr.WindowFrameUnit.RANGE;
++ } else {
++ unit = WindowSpecExpr.WindowFrameUnit.ROW;
++ }
++
++ WindowSpecExpr.WindowFrame windowFrame;
++
++ if (checkIfExist(frameContext.window_frame_extent().window_frame_between())) { // when 'between' is given
++ Window_frame_betweenContext between = frameContext.window_frame_extent().window_frame_between();
++ WindowSpecExpr.WindowStartBound startBound = buildWindowStartBound(between.window_frame_start_bound());
++ WindowSpecExpr.WindowEndBound endBound = buildWindowEndBound(between.window_frame_end_bound());
++
++ windowFrame = new WindowSpecExpr.WindowFrame(unit, startBound, endBound);
++ } else { // if there is only start bound
++ WindowSpecExpr.WindowStartBound startBound =
++ buildWindowStartBound(frameContext.window_frame_extent().window_frame_start_bound());
++ windowFrame = new WindowSpecExpr.WindowFrame(unit, startBound);
++ }
++
++ windowSpec.setWindowFrame(windowFrame);
++ }
++ }
++ return windowSpec;
++ }
++
++ public WindowSpecExpr.WindowStartBound buildWindowStartBound(Window_frame_start_boundContext context) {
++ WindowFrameStartBoundType boundType = null;
++ if (checkIfExist(context.UNBOUNDED())) {
++ boundType = WindowFrameStartBoundType.UNBOUNDED_PRECEDING;
++ } else if (checkIfExist(context.unsigned_value_specification())) {
++ boundType = WindowFrameStartBoundType.PRECEDING;
++ } else {
++ boundType = WindowFrameStartBoundType.CURRENT_ROW;
++ }
++
++ WindowSpecExpr.WindowStartBound bound = new WindowSpecExpr.WindowStartBound(boundType);
++ if (boundType == WindowFrameStartBoundType.PRECEDING) {
++ bound.setNumber(visitUnsigned_value_specification(context.unsigned_value_specification()));
++ }
++
++ return bound;
++ }
++
++ public WindowSpecExpr.WindowEndBound buildWindowEndBound(Window_frame_end_boundContext context) {
++ WindowFrameEndBoundType boundType;
++ if (checkIfExist(context.UNBOUNDED())) {
++ boundType = WindowFrameEndBoundType.UNBOUNDED_FOLLOWING;
++ } else if (checkIfExist(context.unsigned_value_specification())) {
++ boundType = WindowFrameEndBoundType.FOLLOWING;
++ } else {
++ boundType = WindowFrameEndBoundType.CURRENT_ROW;
++ }
++
++ WindowSpecExpr.WindowEndBound endBound = new WindowSpecExpr.WindowEndBound(boundType);
++ if (boundType == WindowFrameEndBoundType.FOLLOWING) {
++ endBound.setNumber(visitUnsigned_value_specification(context.unsigned_value_specification()));
++ }
++
++ return endBound;
++ }
++
++ public Sort.SortSpec[] buildSortSpecs(Sort_specifier_listContext context) {
++ int size = context.sort_specifier().size();
++
+ Sort.SortSpec specs[] = new Sort.SortSpec[size];
+ for (int i = 0; i < size; i++) {
- SQLParser.Sort_specifierContext specContext = ctx.sort_specifier_list().sort_specifier(i);
- Expr column = visitRow_value_predicand(specContext.key);
- specs[i] = new Sort.SortSpec(column);
++ SQLParser.Sort_specifierContext specContext = context.sort_specifier(i);
++ Expr sortKeyExpr = visitRow_value_predicand(specContext.key);
++ specs[i] = new Sort.SortSpec(sortKeyExpr);
+ if (specContext.order_specification() != null) {
+ if (specContext.order.DESC() != null) {
+ specs[i].setDescending();
+ }
+ }
+
+ if (specContext.null_ordering() != null) {
+ if (specContext.null_ordering().FIRST() != null) {
+ specs[i].setNullFirst();
+ }
+ }
+ }
+
- return new Sort(specs);
++ return specs;
++ }
++
++ @Override
++ public Sort visitOrderby_clause(SQLParser.Orderby_clauseContext ctx) {
++ return new Sort(buildSortSpecs(ctx.sort_specifier_list()));
+ }
+
+ @Override
+ public Limit visitLimit_clause(SQLParser.Limit_clauseContext ctx) {
+ return new Limit(visitNumeric_value_expression(ctx.numeric_value_expression()));
+ }
+
+ @Override
+ public Expr visitJoined_table(SQLParser.Joined_tableContext ctx) {
+ Expr top = visitTable_primary(ctx.table_primary());
+
+ // The following loop builds a left deep join tree.
+ Join join;
+ for (int i = 0; i < ctx.joined_table_primary().size(); i++) {
+ join = visitJoined_table_primary(ctx.joined_table_primary(i));
+ join.setLeft(top);
+ top = join;
+ }
+
+ return top;
+ }
+
+ @Override
+ public Join visitJoined_table_primary(SQLParser.Joined_table_primaryContext ctx) {
+ Join join;
+ if (ctx.CROSS() != null) {
+ join = new Join(JoinType.CROSS);
+ } else if (ctx.UNION() != null) {
+ join = new Join(JoinType.UNION);
+ } else { // qualified join or natural
+ if (ctx.join_type() != null && ctx.join_type().outer_join_type() != null) {
+ Outer_join_type_part2Context outer_join_typeContext = ctx.join_type().outer_join_type()
+ .outer_join_type_part2();
+ if (outer_join_typeContext.FULL() != null) {
+ join = new Join(JoinType.FULL_OUTER);
+ } else if (outer_join_typeContext.LEFT() != null) {
+ join = new Join(JoinType.LEFT_OUTER);
+ } else {
+ join = new Join(JoinType.RIGHT_OUTER);
+ }
+ } else {
+ join = new Join(JoinType.INNER);
+ }
+
+ if (ctx.NATURAL() != null) {
+ join.setNatural();
+ }
+
+ if (ctx.join_specification() != null) { // only for qualified join
+ if (ctx.join_specification().join_condition() != null) {
+ Expr searchCondition = visitSearch_condition(ctx.join_specification().
+ join_condition().search_condition());
+ join.setQual(searchCondition);
+ } else if (ctx.join_specification().named_columns_join() != null) {
- ColumnReferenceExpr[] columns = getColumnReferences(ctx.join_specification().
++ ColumnReferenceExpr[] columns = buildColumnReferenceList(ctx.join_specification().
+ named_columns_join().column_reference_list());
+ join.setJoinColumns(columns);
+ }
+ }
+ }
+
+ join.setRight(visitTable_primary(ctx.right));
+ return join;
+ }
+
+ private Expr[] getRowValuePredicandsFromOrdinaryGroupingSetList(Ordinary_grouping_set_listContext ctx) {
+ ArrayList<Expr> rowValuePredicands = new ArrayList<Expr>();
+ for (int i = 0; i < ctx.ordinary_grouping_set().size(); i++) {
+ Collections.addAll(rowValuePredicands, getRowValuePredicandsFromOrdinaryGroupingSet(ctx.ordinary_grouping_set(i)));
+ }
+ return rowValuePredicands.toArray(new Expr[rowValuePredicands.size()]);
+ }
+
+ private Expr[] getRowValuePredicandsFromOrdinaryGroupingSet(Ordinary_grouping_setContext ctx) {
+ ArrayList<Expr> rowValuePredicands = new ArrayList<Expr>();
+ if (ctx.row_value_predicand() != null) {
+ rowValuePredicands.add(visitRow_value_predicand(ctx.row_value_predicand()));
+ }
+ if (ctx.row_value_predicand_list() != null) {
- Collections.addAll(rowValuePredicands, getRowValuePredicands(ctx.row_value_predicand_list()));
++ Collections.addAll(rowValuePredicands, buildRowValuePredicands(ctx.row_value_predicand_list()));
+ }
+ return rowValuePredicands.toArray(new Expr[rowValuePredicands.size()]);
+ }
+
- private Expr[] getRowValuePredicands(Row_value_predicand_listContext ctx) {
++ private Expr[] buildRowValuePredicands(Row_value_predicand_listContext ctx) {
+ Expr[] rowValuePredicands = new Expr[ctx.row_value_predicand().size()];
+ for (int i = 0; i < rowValuePredicands.length; i++) {
+ rowValuePredicands[i] = visitRow_value_predicand(ctx.row_value_predicand(i));
+ }
+ return rowValuePredicands;
+ }
+
- private ColumnReferenceExpr[] getColumnReferences(Column_reference_listContext ctx) {
++ private ColumnReferenceExpr[] buildColumnReferenceList(Column_reference_listContext ctx) {
+ ColumnReferenceExpr[] columnRefs = new ColumnReferenceExpr[ctx.column_reference().size()];
+ for (int i = 0; i < columnRefs.length; i++) {
+ columnRefs[i] = visitColumn_reference(ctx.column_reference(i));
+ }
+ return columnRefs;
+ }
+
+ @Override
+ public Expr visitTable_primary(SQLParser.Table_primaryContext ctx) {
+ if (ctx.table_or_query_name() != null) {
+ Relation relation = new Relation(ctx.table_or_query_name().getText());
+ if (ctx.alias != null) {
+ relation.setAlias(ctx.alias.getText());
+ }
+ return relation;
+ } else if (ctx.derived_table() != null) {
+ return new TablePrimarySubQuery(ctx.name.getText(), visit(ctx.derived_table().table_subquery()));
+ } else {
+ return null;
+ }
+ }
+
+
+ @Override
+ public Expr visitSubquery(SQLParser.SubqueryContext ctx) {
+ return visitQuery_expression(ctx.query_expression());
+ }
+
+ @Override
+ public BetweenPredicate visitBetween_predicate(SQLParser.Between_predicateContext ctx) {
+ Expr predicand = visitRow_value_predicand(ctx.predicand);
+ Expr begin = visitRow_value_predicand(ctx.between_predicate_part_2().begin);
+ Expr end = visitRow_value_predicand(ctx.between_predicate_part_2().end);
+ return new BetweenPredicate(checkIfExist(ctx.between_predicate_part_2().NOT()),
+ checkIfExist(ctx.between_predicate_part_2().SYMMETRIC()), predicand, begin, end);
+ }
+
+ @Override
+ public CaseWhenPredicate visitSimple_case(SQLParser.Simple_caseContext ctx) {
+ Expr leftTerm = visitBoolean_value_expression(ctx.boolean_value_expression());
+ CaseWhenPredicate caseWhen = new CaseWhenPredicate();
+
+ for (int i = 0; i < ctx.simple_when_clause().size(); i++) {
+ Simple_when_clauseContext simpleWhenCtx = ctx.simple_when_clause(i);
+ BinaryOperator bin = new BinaryOperator(OpType.Equals, leftTerm,
+ visitValue_expression(simpleWhenCtx.search_condition().value_expression()));
+ caseWhen.addWhen(bin, buildCaseResult(simpleWhenCtx.result()));
+ }
+ if (ctx.else_clause() != null) {
+ caseWhen.setElseResult(buildCaseResult(ctx.else_clause().result()));
+ }
+ return caseWhen;
+ }
+
+ private Expr buildCaseResult(ResultContext result) {
+ if (result.NULL() != null) {
+ return new NullLiteral();
+ } else {
+ return visitValue_expression(result.value_expression());
+ }
+ }
+
+ @Override
+ public CaseWhenPredicate visitSearched_case(SQLParser.Searched_caseContext ctx) {
+ CaseWhenPredicate caseWhen = new CaseWhenPredicate();
+
+ for (int i = 0; i < ctx.searched_when_clause().size(); i++) {
+ Searched_when_clauseContext searchedWhenCtx = ctx.searched_when_clause(i);
+ caseWhen.addWhen(
+ visitSearch_condition(searchedWhenCtx.search_condition()),
+ buildCaseResult(searchedWhenCtx.result()));
+ }
+ if (ctx.else_clause() != null) {
+ caseWhen.setElseResult(buildCaseResult(ctx.else_clause().result()));
+ }
+ return caseWhen;
+ }
+
+ @Override
+ public Expr visitCommon_value_expression(SQLParser.Common_value_expressionContext ctx) {
+ if (checkIfExist(ctx.NULL())) {
+ return new NullLiteral();
+ } else {
+ return visitChildren(ctx);
+ }
+ }
+
+ @Override
+ public Expr visitParenthesized_value_expression(SQLParser.Parenthesized_value_expressionContext ctx) {
+ return visitValue_expression(ctx.value_expression());
+ }
+
+ @Override
+ public Expr visitBoolean_value_expression(SQLParser.Boolean_value_expressionContext ctx) {
+ Expr current = visitOr_predicate(ctx.or_predicate());
+ return current;
+ }
+
+ @Override
+ public Expr visitOr_predicate(SQLParser.Or_predicateContext ctx) {
+ Expr current = visitAnd_predicate(ctx.and_predicate());
+
+ Expr left;
+ Expr right;
+ for (int i = 0; i < ctx.or_predicate().size(); i++) {
+ left = current;
+ right = visitOr_predicate(ctx.or_predicate(i));
+ current = new BinaryOperator(OpType.Or, left, right);
+ }
+
+ return current;
+ }
+
+ @Override
+ public Expr visitAnd_predicate(SQLParser.And_predicateContext ctx) {
+ Expr current = visitBoolean_factor(ctx.boolean_factor());
+
+ Expr left;
+ Expr right;
+ for (int i = 0; i < ctx.and_predicate().size(); i++) {
+ left = current;
+ right = visitAnd_predicate(ctx.and_predicate(i));
+ current = new BinaryOperator(OpType.And, left, right);
+ }
+
+ return current;
+ }
+
+ @Override
+ public Expr visitBoolean_factor(SQLParser.Boolean_factorContext ctx) {
+ if (ctx.NOT() != null) {
+ return new NotExpr(visitBoolean_test(ctx.boolean_test()));
+ } else {
+ return visitBoolean_test(ctx.boolean_test());
+ }
+ }
+
+ @Override
+ public Expr visitBoolean_test(SQLParser.Boolean_testContext ctx) {
+ if (checkIfExist(ctx.is_clause())) {
+ Is_clauseContext isClauseContext = ctx.is_clause();
+ if (checkIfExist(isClauseContext.NOT())) {
+ if (checkIfExist(ctx.is_clause().truth_value().TRUE())) {
+ return new NotExpr(visitBoolean_primary(ctx.boolean_primary()));
+ } else {
+ return visitBoolean_primary(ctx.boolean_primary());
+ }
+ } else {
+ if (checkIfExist(ctx.is_clause().truth_value().TRUE())) {
+ return visitBoolean_primary(ctx.boolean_primary());
+ } else {
+ return new NotExpr(visitBoolean_primary(ctx.boolean_primary()));
+ }
+ }
+ } else {
+ return visitBoolean_primary(ctx.boolean_primary());
+ }
+ }
+
+ @Override
+ public Expr visitBoolean_primary(SQLParser.Boolean_primaryContext ctx) {
+ if (ctx.predicate() != null) {
+ return visitPredicate(ctx.predicate());
+ } else {
+ return visitBoolean_predicand(ctx.boolean_predicand());
+ }
+ }
+
+ @Override
+ public Expr visitBoolean_predicand(SQLParser.Boolean_predicandContext ctx) {
+ if (checkIfExist(ctx.nonparenthesized_value_expression_primary())) {
+ return visitNonparenthesized_value_expression_primary(ctx.nonparenthesized_value_expression_primary());
+ } else {
+ return visitBoolean_value_expression(ctx.parenthesized_boolean_value_expression().boolean_value_expression());
+ }
+ }
+
+ @Override
+ public Expr visitNonparenthesized_value_expression_primary(
+ SQLParser.Nonparenthesized_value_expression_primaryContext ctx) {
+ return visitChildren(ctx);
+ }
+
+ @Override
+ public Expr visitRow_value_predicand(SQLParser.Row_value_predicandContext ctx) {
+ if (checkIfExist(ctx.row_value_special_case())) {
+ return visitRow_value_special_case(ctx.row_value_special_case());
+ } else {
+ return visitRow_value_constructor_predicand(ctx.row_value_constructor_predicand());
+ }
+ }
+
+ @Override
+ public Expr visitRow_value_constructor_predicand(SQLParser.Row_value_constructor_predicandContext ctx) {
+ if (checkIfExist(ctx.boolean_predicand())) {
+ return visitBoolean_predicand(ctx.boolean_predicand());
+ } else {
+ return visitCommon_value_expression(ctx.common_value_expression());
+ }
+ }
+
+ @Override
+ public BinaryOperator visitComparison_predicate(SQLParser.Comparison_predicateContext ctx) {
+ TerminalNode operator = (TerminalNode) ctx.comp_op().getChild(0);
+ return new BinaryOperator(tokenToExprType(operator.getSymbol().getType()),
+ visitRow_value_predicand(ctx.left),
+ visitRow_value_predicand(ctx.right));
+ }
+
+ @Override
+ public Expr visitNumeric_value_expression(SQLParser.Numeric_value_expressionContext ctx) {
+ Expr current = visitTerm(ctx.term(0));
+
+ Expr left;
+ Expr right;
+ for (int i = 1; i < ctx.getChildCount(); i++) {
+ left = current;
+ TerminalNode operator = (TerminalNode) ctx.getChild(i++);
+ right = visitTerm((TermContext) ctx.getChild(i));
+
+ if (operator.getSymbol().getType() == PLUS) {
+ current = new BinaryOperator(OpType.Plus, left, right);
+ } else {
+ current = new BinaryOperator(OpType.Minus, left, right);
+ }
+ }
+
+ return current;
+ }
+
+ @Override
+ public Expr visitTerm(SQLParser.TermContext ctx) {
+ Expr current = visitFactor(ctx.factor(0));
+
+ Expr left;
+ Expr right;
+ for (int i = 1; i < ctx.getChildCount(); i++) {
+ left = current;
+ TerminalNode operator = (TerminalNode) ctx.getChild(i++);
+ right = visitFactor((FactorContext) ctx.getChild(i));
+
+ if (operator.getSymbol().getType() == MULTIPLY) {
+ current = new BinaryOperator(OpType.Multiply, left, right);
+ } else if (operator.getSymbol().getType() == DIVIDE) {
+ current = new BinaryOperator(OpType.Divide, left, right);
+ } else {
+ current = new BinaryOperator(OpType.Modular, left, right);
+ }
+ }
+
+ return current;
+ }
+
+ @Override
+ public Expr visitFactor(SQLParser.FactorContext ctx) {
+ Expr current = visitNumeric_primary(ctx.numeric_primary());
+ if (checkIfExist(ctx.sign()) && checkIfExist(ctx.sign().MINUS())) {
+ current = new SignedExpr(true, current);
+ }
+ return current;
+ }
+
+ @Override
+ public Expr visitNumeric_primary(SQLParser.Numeric_primaryContext ctx) {
+ Expr current = null;
+ if (checkIfExist(ctx.value_expression_primary())) {
+ current = visitValue_expression_primary(ctx.value_expression_primary());
+ for (int i = 0; i < ctx.CAST_EXPRESSION().size(); i++) {
+ current = new CastExpr(current, visitData_type(ctx.cast_target(i).data_type()));
+ }
+ } else if (checkIfExist(ctx.numeric_value_function())) {
+ current = visitNumeric_value_function(ctx.numeric_value_function());
+ }
+
+ return current;
+ }
+
+ public static OpType tokenToExprType(int tokenId) {
+ switch (tokenId) {
+ case SQLParser.UNION:
+ return OpType.Union;
+ case SQLParser.EXCEPT:
+ return OpType.Except;
+ case SQLParser.INTERSECT:
+ return OpType.Intersect;
+
+ case SQLParser.AND:
+ return OpType.And;
+ case SQLParser.OR:
+ return OpType.Or;
+
+ case SQLParser.EQUAL:
+ return OpType.Equals;
+ case SQLParser.NOT_EQUAL:
+ return OpType.NotEquals;
+ case SQLParser.LTH:
+ return OpType.LessThan;
+ case SQLParser.LEQ:
+ return OpType.LessThanOrEquals;
+ case SQLParser.GTH:
+ return OpType.GreaterThan;
+ case SQLParser.GEQ:
+ return OpType.GreaterThanOrEquals;
+
+ case SQLParser.MULTIPLY:
+ return OpType.Multiply;
+ case SQLParser.DIVIDE:
+ return OpType.Divide;
+ case SQLParser.MODULAR:
+ return OpType.Modular;
+ case SQLParser.PLUS:
+ return OpType.Plus;
+ case SQLParser.MINUS:
+ return OpType.Minus;
+
+ default:
+ throw new RuntimeException("Unknown Token Id: " + tokenId);
+ }
+ }
+
+ @Override
+ public InPredicate visitIn_predicate(SQLParser.In_predicateContext ctx) {
+ return new InPredicate(visitChildren(ctx.numeric_value_expression()),
+ visitIn_predicate_value(ctx.in_predicate_value()), ctx.NOT() != null);
+ }
+
+ @Override
+ public Expr visitIn_predicate_value(SQLParser.In_predicate_valueContext ctx) {
+ if (checkIfExist(ctx.in_value_list())) {
+ int size = ctx.in_value_list().row_value_predicand().size();
+ Expr [] exprs = new Expr[size];
+ for (int i = 0; i < size; i++) {
+ exprs[i] = visitRow_value_predicand(ctx.in_value_list().row_value_predicand(i));
+ }
+ return new ValueListExpr(exprs);
+ } else {
+ return new SimpleTableSubQuery(visitChildren(ctx.table_subquery()));
+ }
+ }
+
+ @Override
+ public Expr visitArray(SQLParser.ArrayContext ctx) {
+ int size = ctx.numeric_value_expression().size();
+ Expr[] exprs = new Expr[size];
+ for (int i = 0; i < size; i++) {
+ exprs[i] = visit(ctx.numeric_value_expression(i));
+ }
+ return new ValueListExpr(exprs);
+ }
+
+ @Override
+ public Expr visitPattern_matching_predicate(SQLParser.Pattern_matching_predicateContext ctx) {
+ Expr predicand = visitChildren(ctx.row_value_predicand());
+ Expr pattern = new LiteralValue(stripQuote(ctx.Character_String_Literal().getText()),
+ LiteralType.String);
+
+ if (checkIfExist(ctx.pattern_matcher().negativable_matcher())) {
+ boolean not = ctx.pattern_matcher().NOT() != null;
+ Negativable_matcherContext matcher = ctx.pattern_matcher().negativable_matcher();
+ if (checkIfExist(matcher.LIKE())) {
+ return new PatternMatchPredicate(OpType.LikePredicate, not, predicand, pattern);
+ } else if (checkIfExist(matcher.ILIKE())) {
+ return new PatternMatchPredicate(OpType.LikePredicate, not, predicand, pattern, true);
+ } else if (checkIfExist(matcher.SIMILAR())) {
+ return new PatternMatchPredicate(OpType.SimilarToPredicate, not, predicand, pattern);
+ } else if (checkIfExist(matcher.REGEXP()) || checkIfExist(matcher.RLIKE())) {
+ return new PatternMatchPredicate(OpType.Regexp, not, predicand, pattern);
+ } else {
+ throw new SQLSyntaxError("Unsupported predicate: " + matcher.getText());
+ }
+ } else if (checkIfExist(ctx.pattern_matcher().regex_matcher())) {
+ Regex_matcherContext matcher = ctx.pattern_matcher().regex_matcher();
+ if (checkIfExist(matcher.Similar_To())) {
+ return new PatternMatchPredicate(OpType.Regexp, false, predicand, pattern, false);
+ } else if (checkIfExist(matcher.Not_Similar_To())) {
+ return new PatternMatchPredicate(OpType.Regexp, true, predicand, pattern, false);
+ } else if (checkIfExist(matcher.Similar_To_Case_Insensitive())) {
+ return new PatternMatchPredicate(OpType.Regexp, false, predicand, pattern, true);
+ } else if (checkIfExist(matcher.Not_Similar_To_Case_Insensitive())) {
+ return new PatternMatchPredicate(OpType.Regexp, true, predicand, pattern, true);
+ } else {
+ throw new SQLSyntaxError("Unsupported predicate: " + matcher.getText());
+ }
+ } else {
+ throw new SQLSyntaxError("Unsupported predicate: " + ctx.pattern_matcher().getText());
+ }
+ }
+
+ @Override
+ public IsNullPredicate visitNull_predicate(SQLParser.Null_predicateContext ctx) {
+ Expr predicand = visitRow_value_predicand(ctx.row_value_predicand());
+ return new IsNullPredicate(ctx.NOT() != null, predicand);
+ }
+
+ @Override
+ public ExistsPredicate visitExists_predicate(SQLParser.Exists_predicateContext ctx) {
+ return new ExistsPredicate(new SimpleTableSubQuery(visitTable_subquery(ctx.table_subquery())), ctx.NOT() != null);
+ }
+
+ @Override
+ public ColumnReferenceExpr visitColumn_reference(SQLParser.Column_referenceContext ctx) {
+ ColumnReferenceExpr column = new ColumnReferenceExpr(ctx.name.getText());
+ if (checkIfExist(ctx.db_name)) {
+ column.setQualifier(CatalogUtil.buildFQName(ctx.db_name.getText(), ctx.tb_name.getText()));
+ } else if (ctx.tb_name != null) {
+ column.setQualifier(ctx.tb_name.getText());
+ }
+
+ return column;
+ }
+
+ @Override
+ public LiteralValue visitUnsigned_numeric_literal(@NotNull SQLParser.Unsigned_numeric_literalContext ctx) {
+ if (ctx.NUMBER() != null) {
+ long lValue = Long.parseLong(ctx.getText());
+ if (lValue >= Integer.MIN_VALUE && lValue <= Integer.MAX_VALUE) {
+ return new LiteralValue(ctx.getText(), LiteralType.Unsigned_Integer);
+ } else {
+ return new LiteralValue(ctx.getText(), LiteralType.Unsigned_Large_Integer);
+ }
+ } else {
+ return new LiteralValue(ctx.getText(), LiteralType.Unsigned_Float);
+ }
+ }
+
+ @Override
- public FunctionExpr visitAggregate_function(SQLParser.Aggregate_functionContext ctx) {
++ public GeneralSetFunctionExpr visitAggregate_function(SQLParser.Aggregate_functionContext ctx) {
+ if (ctx.COUNT() != null && ctx.MULTIPLY() != null) {
+ return new CountRowsFunctionExpr();
+ } else {
+ return visitGeneral_set_function(ctx.general_set_function());
+ }
+ }
+
+ @Override
- public FunctionExpr visitGeneral_set_function(SQLParser.General_set_functionContext ctx) {
++ public GeneralSetFunctionExpr visitGeneral_set_function(SQLParser.General_set_functionContext ctx) {
+ String signature = ctx.set_function_type().getText();
+ boolean distinct = checkIfExist(ctx.set_qualifier()) && checkIfExist(ctx.set_qualifier().DISTINCT());
+ Expr param = visitValue_expression(ctx.value_expression());
+
- return new GeneralSetFunctionExpr(signature, distinct, param);
++ return new GeneralSetFunctionExpr(signature, distinct, new Expr [] {param});
+ }
+
+ @Override
+ public FunctionExpr visitRoutine_invocation(SQLParser.Routine_invocationContext ctx) {
+ String signature = ctx.function_name().getText();
+ FunctionExpr function = new FunctionExpr(signature);
+ if (ctx.sql_argument_list() != null) {
+ int numArgs = ctx.sql_argument_list().value_expression().size();
+ Expr[] argument_list = new Expr[numArgs];
+ for (int i = 0; i < numArgs; i++) {
+ argument_list[i] = visitValue_expression(ctx.sql_argument_list().
+ value_expression().get(i));
+ }
+
+ function.setParams(argument_list);
+ }
+ return function;
+ }
+
+ @Override
+ public NamedExpr visitDerived_column(Derived_columnContext ctx) {
+ NamedExpr target = new NamedExpr(visitValue_expression(ctx.value_expression()));
+ if (ctx.as_clause() != null) {
+ target.setAlias(ctx.as_clause().identifier().getText());
+ }
+ return target;
+ }
+
+ @Override
+ public NamedExpr visitQualified_asterisk(Qualified_asteriskContext ctx) {
+ QualifiedAsteriskExpr target = new QualifiedAsteriskExpr();
+ if (ctx.tb_name != null) {
+ target.setQualifier(ctx.tb_name.getText());
+ }
+
+ return new NamedExpr(target);
+ }
+
+ @Override
+ public Expr visitCharacter_string_type(SQLParser.Character_string_typeContext ctx) {
+ return new LiteralValue(stripQuote(ctx.getText()), LiteralType.String);
+ }
+
+ @Override
+ public Expr visitCharacter_value_expression(SQLParser.Character_value_expressionContext ctx) {
+ Expr current = visitCharacter_factor(ctx.character_factor(0));
+
+ Expr left;
+ Expr right;
+ for (int i = 1; i < ctx.getChildCount(); i++) {
+ left = current;
+ i++; // skip '||' operator
+ right = visitCharacter_factor((Character_factorContext) ctx.getChild(i));
+
+ if (left.getType() == OpType.Literal && right.getType() == OpType.Literal) {
+ current = new LiteralValue(((LiteralValue) left).getValue() + ((LiteralValue) right).getValue(),
+ LiteralType.String);
+ } else {
+ current = new BinaryOperator(OpType.Concatenate, left, right);
+ }
+ }
+
+ return current;
+ }
+
+
+ @Override
+ public Expr visitNumeric_value_function(Numeric_value_functionContext ctx) {
+ if (checkIfExist(ctx.extract_expression())) {
+ return visitExtract_expression(ctx.extract_expression());
+ }
+
+ return null;
+ }
+
+ @Override
+ public Expr visitExtract_expression(Extract_expressionContext ctx) {
+ Expr extractTarget = new LiteralValue(ctx.extract_field_string.getText(), LiteralType.String);
+ Expr extractSource;
+ if (checkIfExist(ctx.extract_source().column_reference())) {
+ extractSource = visitColumn_reference(ctx.extract_source().column_reference());
+ } else if (checkIfExist(ctx.extract_source().datetime_literal())) {
+ extractSource = visitDatetime_literal(ctx.extract_source().datetime_literal());
+ } else {
+ return null;
+ }
+
+ String functionName = "date_part";
+ Expr[] params = new Expr[]{extractTarget, extractSource};
+
+ return new FunctionExpr(functionName, params);
+ }
+
+ @Override
+ public Expr visitTrim_function(SQLParser.Trim_functionContext ctx) {
+ Expr trimSource = visitChildren(ctx.trim_operands().trim_source);
+ String functionName = "trim";
+ if (checkIfExist(ctx.trim_operands().FROM())) {
+ if (checkIfExist(ctx.trim_operands().trim_specification())) {
+ Trim_specificationContext specification = ctx.trim_operands().trim_specification();
+ if (checkIfExist(specification.LEADING())) {
+ functionName = "ltrim";
+ } else if (checkIfExist(specification.TRAILING())) {
+ functionName = "rtrim";
+ } else {
+ functionName = "trim";
+ }
+ }
+ }
+
+ Expr trimCharacters = null;
+ if (checkIfExist(ctx.trim_operands().trim_character)) {
+ trimCharacters = visitCharacter_value_expression(ctx.trim_operands().trim_character);
+ }
+
+ Expr[] params;
+ if (trimCharacters != null) {
+ params = new Expr[]{trimSource, trimCharacters};
+ } else {
+ params = new Expr[]{trimSource};
+ }
+
+ return new FunctionExpr(functionName, params);
+ }
+
+ @Override
+ public Expr visitDatabase_definition(@NotNull SQLParser.Database_definitionContext ctx) {
+ return new CreateDatabase(ctx.identifier().getText(), null, checkIfExist(ctx.if_not_exists()));
+ }
+
+ @Override
+ public Expr visitDrop_database_statement(@NotNull SQLParser.Drop_database_statementContext ctx) {
+ return new DropDatabase(ctx.identifier().getText(), checkIfExist(ctx.if_exists()));
+ }
+
+ @Override
+ public Expr visitCreate_table_statement(SQLParser.Create_table_statementContext ctx) {
+ String tableName = ctx.table_name().getText();
+ CreateTable createTable = new CreateTable(tableName, checkIfExist(ctx.if_not_exists()));
+
+ if (checkIfExist(ctx.EXTERNAL())) {
+ createTable.setExternal();
+
+ ColumnDefinition[] elements = getDefinitions(ctx.table_elements());
+ String fileType = ctx.file_type.getText();
+ String path = stripQuote(ctx.path.getText());
+
+ createTable.setTableElements(elements);
+ createTable.setStorageType(fileType);
+ createTable.setLocation(path);
+ } else {
+ if (checkIfExist(ctx.table_elements())) {
+ ColumnDefinition[] elements = getDefinitions(ctx.table_elements());
+ createTable.setTableElements(elements);
+ }
+
+ if (checkIfExist(ctx.USING())) {
+ String fileType = ctx.file_type.getText();
+ createTable.setStorageType(fileType);
+ }
+
+ if (checkIfExist(ctx.query_expression())) {
+ Expr subquery = visitQuery_expression(ctx.query_expression());
+ createTable.setSubQuery(subquery);
+ }
+ }
+
+ if (checkIfExist(ctx.param_clause())) {
+ Map<String, String> params = escapeTableMeta(getParams(ctx.param_clause()));
+ createTable.setParams(params);
+ }
+
+ if (checkIfExist(ctx.table_partitioning_clauses())) {
+ PartitionMethodDescExpr partitionMethodDesc =
+ parseTablePartitioningClause(ctx.table_partitioning_clauses());
+ createTable.setPartitionMethod(partitionMethodDesc);
+ }
+ return createTable;
+ }
+
+ private ColumnDefinition[] getDefinitions(SQLParser.Table_elementsContext ctx) {
+ int size = ctx.field_element().size();
+ ColumnDefinition[] elements = new ColumnDefinition[size];
+ for (int i = 0; i < size; i++) {
+ String name = ctx.field_element(i).name.getText();
+ DataTypeExpr typeDef = visitData_type(ctx.field_element(i).field_type().data_type());
+ elements[i] = new ColumnDefinition(name, typeDef);
+ }
+
+ return elements;
+ }
+
+ public PartitionMethodDescExpr parseTablePartitioningClause(SQLParser.Table_partitioning_clausesContext ctx) {
+
+ if (checkIfExist(ctx.range_partitions())) { // For Range Partition
+ Range_partitionsContext rangePartitionsContext = ctx.range_partitions();
+ List<Range_value_clauseContext> rangeValueClause = rangePartitionsContext.
+ range_value_clause_list().range_value_clause();
+
+ List<RangePartitionSpecifier> specifiers = Lists.newArrayList();
+
+ for (Range_value_clauseContext rangeValue : rangeValueClause) {
+ if (checkIfExist(rangeValue.MAXVALUE())) { // LESS THAN (MAXVALUE)
+ specifiers.add(new RangePartitionSpecifier(rangeValue.partition_name().getText()));
+ } else { // LESS THAN (expr)
+ specifiers.add(new RangePartitionSpecifier(rangeValue.partition_name().getText(),
+ visitValue_expression(rangeValue.value_expression())));
+ }
+ }
- return new CreateTable.RangePartition(getColumnReferences(ctx.range_partitions().column_reference_list()),
++ return new CreateTable.RangePartition(buildColumnReferenceList(ctx.range_partitions().column_reference_list()),
+ specifiers);
+
+ } else if (checkIfExist(ctx.hash_partitions())) { // For Hash Partition
+ Hash_partitionsContext hashPartitions = ctx.hash_partitions();
+
+ if (checkIfExist(hashPartitions.hash_partitions_by_quantity())) { // PARTITIONS (num)
- return new HashPartition(getColumnReferences(hashPartitions.column_reference_list()),
++ return new HashPartition(buildColumnReferenceList(hashPartitions.column_reference_list()),
+ visitNumeric_value_expression(hashPartitions.hash_partitions_by_quantity().quantity));
+
+ } else { // ( PARTITION part_name , ...)
+ List<CreateTable.PartitionSpecifier> specifiers = Lists.newArrayList();
+ for (Individual_hash_partitionContext partition :
+ hashPartitions.individual_hash_partitions().individual_hash_partition()) {
+ specifiers.add(new CreateTable.PartitionSpecifier(partition.partition_name().getText()));
+ }
- return new HashPartition(getColumnReferences(hashPartitions.column_reference_list()), specifiers);
++ return new HashPartition(buildColumnReferenceList(hashPartitions.column_reference_list()), specifiers);
+ }
+
+ } else if (checkIfExist(ctx.list_partitions())) { // For List Partition
+ List_partitionsContext listPartitions = ctx.list_partitions();
+ List<List_value_partitionContext> partitions = listPartitions.list_value_clause_list().list_value_partition();
+ List<ListPartitionSpecifier> specifiers = Lists.newArrayList();
+
+ for (List_value_partitionContext listValuePartition : partitions) {
+ int size = listValuePartition.in_value_list().row_value_predicand().size();
+ Expr [] exprs = new Expr[size];
+ for (int i = 0; i < size; i++) {
+ exprs[i] = visitRow_value_predicand(listValuePartition.in_value_list().row_value_predicand(i));
+ }
+ specifiers.add(new ListPartitionSpecifier(listValuePartition.partition_name().getText(),
+ new ValueListExpr(exprs)));
+ }
- return new ListPartition(getColumnReferences(ctx.list_partitions().column_reference_list()), specifiers);
++ return new ListPartition(buildColumnReferenceList(ctx.list_partitions().column_reference_list()), specifiers);
+
+ } else if (checkIfExist(ctx.column_partitions())) { // For Column Partition (Hive Style)
+ return new CreateTable.ColumnPartition(getDefinitions(ctx.column_partitions().table_elements()), true);
+ } else {
+ throw new SQLSyntaxError("Invalid Partition Type: " + ctx.toStringTree());
+ }
+ }
+
+ @Override
+ public DataTypeExpr visitData_type(SQLParser.Data_typeContext ctx) {
+ SQLParser.Predefined_typeContext predefined_type = ctx.predefined_type();
+
+ DataTypeExpr typeDefinition = null;
+ if (predefined_type.character_string_type() != null) {
+ SQLParser.Character_string_typeContext character_string_type =
+ predefined_type.character_string_type();
+
+ if ((character_string_type.CHARACTER() != null || character_string_type.CHAR() != null) &&
+ character_string_type.VARYING() == null) {
+
+ typeDefinition = new DataTypeExpr(Type.CHAR.name());
+
+ if (character_string_type.type_length() != null) {
+ typeDefinition.setLengthOrPrecision(
+ Integer.parseInt(character_string_type.type_length().NUMBER().getText()));
+ }
+
+ } else if (character_string_type.VARCHAR() != null
+ || character_string_type.VARYING() != null) {
+
+ typeDefinition = new DataTypeExpr(Type.VARCHAR.name());
+
+ if (character_string_type.type_length() != null) {
+ typeDefinition.setLengthOrPrecision(
+ Integer.parseInt(character_string_type.type_length().NUMBER().getText()));
+ }
+
+ } else if (character_string_type.TEXT() != null) {
+ typeDefinition = new DataTypeExpr(Type.TEXT.name());
+ }
+
+ } else if (predefined_type.national_character_string_type() != null) {
+ SQLParser.National_character_string_typeContext nchar_type =
+ predefined_type.national_character_string_type();
+ if ((nchar_type.CHAR() != null || nchar_type.CHARACTER() != null
+ || nchar_type.NCHAR() != null) && nchar_type.VARYING() == null) {
+ typeDefinition = new DataTypeExpr(Type.NCHAR.name());
+ } else if (nchar_type.NVARCHAR() != null || nchar_type.VARYING() != null) {
+ typeDefinition = new DataTypeExpr(Type.NVARCHAR.name());
+ }
+
+ if (nchar_type.type_length() != null) {
+ typeDefinition.setLengthOrPrecision(
+ Integer.parseInt(nchar_type.type_length().NUMBER().getText()));
+ }
+
+ } else if (predefined_type.binary_large_object_string_type() != null) {
+ SQLParser.Binary_large_object_string_typeContext blob_type =
+ predefined_type.binary_large_object_string_type();
+ typeDefinition = new DataTypeExpr(Type.BLOB.name());
+ if (blob_type.type_length() != null) {
+ typeDefinition.setLengthOrPrecision(
+ Integer.parseInt(blob_type.type_length().NUMBER().getText()));
+ }
+ } else if (predefined_type.numeric_type() != null) {
+ // exact number
+ if (predefined_type.numeric_type().exact_numeric_type() != null) {
+ SQLParser.Exact_numeric_typeContext exactType =
+ predefined_type.numeric_type().exact_numeric_type();
+ if (exactType.TINYINT() != null || exactType.INT1() != null) {
+ typeDefinition = new DataTypeExpr(Type.INT1.name());
+ } else if (exactType.INT2() != null || exactType.SMALLINT() != null) {
+ typeDefinition = new DataTypeExpr(Type.INT2.name());
+ } else if (exactType.INT4() != null || exactType.INTEGER() != null ||
+ exactType.INT() != null) {
+ typeDefinition = new DataTypeExpr(Type.INT4.name());
+ } else if (exactType.INT8() != null || exactType.BIGINT() != null) {
+ typeDefinition = new DataTypeExpr(Type.INT8.name());
+ } else if (exactType.NUMERIC() != null) {
+ typeDefinition = new DataTypeExpr(Type.NUMERIC.name());
+ } else if (exactType.DECIMAL() != null || exactType.DEC() != null) {
+ typeDefinition = new DataTypeExpr(Type.NUMERIC.name());
+ }
+
+ if (typeDefinition.getTypeName().equals(Type.NUMERIC.name())) {
+ if (exactType.precision_param() != null) {
+ if (exactType.precision_param().scale != null) {
+ typeDefinition.setScale(
+ Integer.parseInt(exactType.precision_param().scale.getText()));
+ }
+ typeDefinition.setLengthOrPrecision(
+ Integer.parseInt(exactType.precision_param().precision.getText()));
+ }
+ }
+ } else { // approximate number
+ SQLParser.Approximate_numeric_typeContext approximateType =
+ predefined_type.numeric_type().approximate_numeric_type();
+ if (approximateType.FLOAT() != null || approximateType.FLOAT4() != null
+ || approximateType.REAL() != null) {
+ typeDefinition = new DataTypeExpr(Type.FLOAT4.name());
+ } else if (approximateType.FLOAT8() != null || approximateType.DOUBLE() != null) {
+ typeDefinition = new DataTypeExpr(Type.FLOAT8.name());
+ }
+ }
+ } else if (predefined_type.boolean_type() != null) {
+ typeDefinition = new DataTypeExpr(Type.BOOLEAN.name());
+ } else if (predefined_type.datetime_type() != null) {
+ SQLParser.Datetime_typeContext dateTimeType = predefined_type.datetime_type();
+ if (dateTimeType.DATE() != null) {
+ typeDefinition = new DataTypeExpr(Type.DATE.name());
+ } else if (dateTimeType.TIME(0) != null && dateTimeType.ZONE() == null) {
+ typeDefinition = new DataTypeExpr(Type.TIME.name());
+ } else if ((dateTimeType.TIME(0) != null && dateTimeType.ZONE() != null) ||
+ dateTimeType.TIMETZ() != null) {
+ typeDefinition = new DataTypeExpr(Type.TIMEZ.name());
+ } else if (dateTimeType.TIMESTAMP() != null && dateTimeType.ZONE() == null) {
+ typeDefinition = new DataTypeExpr(Type.TIMESTAMP.name());
+ } else if ((dateTimeType.TIMESTAMP() != null && dateTimeType.ZONE() != null) ||
+ dateTimeType.TIMESTAMPTZ() != null) {
+ typeDefinition = new DataTypeExpr(Type.TIMESTAMPZ.name());
+ }
+ } else if (predefined_type.bit_type() != null) {
+ SQLParser.Bit_typeContext bitType = predefined_type.bit_type();
+ if (bitType.VARBIT() != null || bitType.VARYING() != null) {
+ typeDefinition = new DataTypeExpr(Type.VARBIT.name());
+ } else {
+ typeDefinition = new DataTypeExpr(Type.BIT.name());
+ }
+ if (bitType.type_length() != null) {
+ typeDefinition.setLengthOrPrecision(
+ Integer.parseInt(bitType.type_length().NUMBER().getText()));
+ }
+ } else if (predefined_type.binary_type() != null) {
+ SQLParser.Binary_typeContext binaryType = predefined_type.binary_type();
+ if (binaryType.VARBINARY() != null || binaryType.VARYING() != null) {
+ typeDefinition = new DataTypeExpr(Type.VARBINARY.name());
+ } else {
+ typeDefinition = new DataTypeExpr(Type.BINARY.name());
+ }
+
+ if (binaryType.type_length() != null) {
+ typeDefinition.setLengthOrPrecision(
+ Integer.parseInt(binaryType.type_length().NUMBER().getText()));
+ }
+ } else if (predefined_type.network_type() != null) {
+ typeDefinition = new DataTypeExpr(Type.INET4.name());
+ }
+
+ return typeDefinition;
+ }
+
+ @Override
+ public Expr visitInsert_statement(SQLParser.Insert_statementContext ctx) {
+ Insert insertExpr = new Insert();
+
+ if (ctx.OVERWRITE() != null) {
+ insertExpr.setOverwrite();
+ }
+
+ if (ctx.table_name() != null) {
+ insertExpr.setTableName(ctx.table_name().getText());
+
+ if (ctx.column_name_list() != null) {
+ String[] targetColumns = new String[ctx.column_name_list().identifier().size()];
+ for (int i = 0; i < targetColumns.length; i++) {
+ targetColumns[i] = ctx.column_name_list().identifier().get(i).getText();
+ }
+
+ insertExpr.setTargetColumns(targetColumns);
+ }
+ }
+
+ if (ctx.LOCATION() != null) {
+ insertExpr.setLocation(stripQuote(ctx.path.getText()));
+
+ if (ctx.USING() != null) {
+ insertExpr.setStorageType(ctx.file_type.getText());
+
+ if (ctx.param_clause() != null) {
+ insertExpr.setParams(escapeTableMeta(getParams(ctx.param_clause())));
+ }
+ }
+ }
+
+ insertExpr.setSubQuery(visitQuery_expression(ctx.query_expression()));
+
+ Preconditions.checkState(insertExpr.hasTableName() || insertExpr.hasLocation(),
+ "Either a table name or a location should be given.");
+ Preconditions.checkState(insertExpr.hasTableName() ^ insertExpr.hasLocation(),
+ "A table name and a location cannot coexist.");
+ return insertExpr;
+ }
+
+ @Override
+ public Expr visitDrop_table_statement(SQLParser.Drop_table_statementContext ctx) {
+ return new DropTable(ctx.table_name().getText(), checkIfExist(ctx.if_exists()), checkIfExist(ctx.PURGE()));
+ }
+
+
+ private Map<String, String> getParams(SQLParser.Param_clauseContext ctx) {
+ Map<String, String> params = new HashMap<String, String>();
+ for (int i = 0; i < ctx.param().size(); i++) {
+ params.put(stripQuote(ctx.param(i).key.getText()), stripQuote(ctx.param(i).value.getText()));
+ }
+
+ return params;
+ }
+
+ public Map<String, String> escapeTableMeta(Map<String, String> map) {
+ Map<String, String> params = new HashMap<String, String>();
+ for (Map.Entry<String, String> entry : map.entrySet()) {
+ if (entry.getKey().equals(StorageConstants.CSVFILE_DELIMITER)) {
+ params.put(entry.getKey(), escapeDelimiter(entry.getValue()));
+ } else {
+ params.put(entry.getKey(), entry.getValue());
+ }
+ }
+ return params;
+ }
+
+ public static String escapeDelimiter(String value) {
+ try {
+ String delimiter = StringEscapeUtils.unescapeJava(value);
+ delimiter = new String(new byte[]{Byte.valueOf(delimiter).byteValue()}, Charset.defaultCharset());
+ return StringEscapeUtils.escapeJava(delimiter);
+ } catch (NumberFormatException e) {
+ }
+ return value;
+ }
+
+ private static String stripQuote(String str) {
+ return str.substring(1, str.length() - 1);
+ }
+
+ @Override
+ public Expr visitCast_specification(SQLParser.Cast_specificationContext ctx) {
+ Expr operand = visitChildren(ctx.cast_operand());
+ DataTypeExpr castTarget = visitData_type(ctx.cast_target().data_type());
+ return new CastExpr(operand, castTarget);
+ }
+
+ @Override
+ public Expr visitUnsigned_value_specification(@NotNull SQLParser.Unsigned_value_specificationContext ctx) {
+ return visitChildren(ctx);
+ }
+
+ @Override
+ public Expr visitUnsigned_literal(@NotNull SQLParser.Unsigned_literalContext ctx) {
+ if (checkIfExist(ctx.unsigned_numeric_literal())) {
+ return visitUnsigned_numeric_literal(ctx.unsigned_numeric_literal());
+ } else {
+ return visitGeneral_literal(ctx.general_literal());
+ }
+ }
+
+ @Override
+ public Expr visitGeneral_literal(SQLParser.General_literalContext ctx) {
+ if (checkIfExist(ctx.Character_String_Literal())) {
+ return new LiteralValue(stripQuote(ctx.Character_String_Literal().getText()), LiteralType.String);
+ } else if (checkIfExist(ctx.datetime_literal())) {
+ return visitDatetime_literal(ctx.datetime_literal());
+ } else {
+ return new BooleanLiteral(checkIfExist(ctx.boolean_literal().TRUE()));
+ }
+ }
+
+ @Override
+ public Expr visitDatetime_literal(@NotNull SQLParser.Datetime_literalContext ctx) {
+ if (checkIfExist(ctx.time_literal())) {
+ return visitTime_literal(ctx.time_literal());
+ } else if (checkIfExist(ctx.date_literal())) {
+ return visitDate_literal(ctx.date_literal());
+ } else {
+ return visitTimestamp_literal(ctx.timestamp_literal());
+ }
+ }
+
+ @Override
+ public Expr visitTime_literal(SQLParser.Time_literalContext ctx) {
+ String timePart = stripQuote(ctx.time_string.getText());
+ return new TimeLiteral(parseTime(timePart));
+ }
+
+ @Override
+ public Expr visitDate_literal(SQLParser.Date_literalContext ctx) {
+ String datePart = stripQuote(ctx.date_string.getText());
+ return new DateLiteral(parseDate(datePart));
+ }
+
+ @Override
+ public Expr visitTimestamp_literal(SQLParser.Timestamp_literalContext ctx) {
+ String timestampStr = stripQuote(ctx.timestamp_string.getText());
+ String[] parts = timestampStr.split(" ");
+ String datePart = parts[0];
+ String timePart = parts[1];
+ return new TimestampLiteral(parseDate(datePart), parseTime(timePart));
+ }
+
+ private DateValue parseDate(String datePart) {
+ // e.g., 1980-04-01
+ String[] parts = datePart.split("-");
+ return new DateValue(parts[0], parts[1], parts[2]);
+ }
+
+ private TimeValue parseTime(String timePart) {
+ // e.g., 12:01:50.399
+ String[] parts = timePart.split(":");
+
+ TimeValue time;
+ boolean hasFractionOfSeconds = parts[2].indexOf('.') > 0;
+ if (hasFractionOfSeconds) {
+ String[] secondsParts = parts[2].split("\\.");
+ time = new TimeValue(parts[0], parts[1], secondsParts[0]);
+ if (secondsParts.length == 2) {
+ time.setSecondsFraction(secondsParts[1]);
+ }
+ } else {
+ time = new TimeValue(parts[0], parts[1], parts[2]);
+ }
+ return time;
+ }
+
+ @Override
+ public Expr visitAlter_tablespace_statement(@NotNull SQLParser.Alter_tablespace_statementContext ctx) {
+ AlterTablespace alter = new AlterTablespace(ctx.space_name.getText());
+ alter.setLocation(stripQuote(ctx.uri.getText()));
+ return alter;
+ }
+
+ @Override
+ public Expr visitAlter_table_statement(SQLParser.Alter_table_statementContext ctx) {
+
+ final List<Table_nameContext> tables = ctx.table_name();
+
+ final AlterTable alterTable = new AlterTable(tables.get(0).getText());
+
+ if (tables.size() == 2) {
+ alterTable.setNewTableName(tables.get(1).getText());
+ }
+
+ if (checkIfExist(ctx.column_name()) && ctx.column_name().size() == 2) {
+ final List<Column_nameContext> columns = ctx.column_name();
+ alterTable.setColumnName(columns.get(0).getText());
+ alterTable.setNewColumnName(columns.get(1).getText());
+ }
+
+ Field_elementContext field_elementContext = ctx.field_element();
+ if (checkIfExist(field_elementContext)) {
+ final String name = field_elementContext.name.getText();
+ final DataTypeExpr typeDef = visitData_type(field_elementContext.field_type().data_type());
+ final ColumnDefinition columnDefinition = new ColumnDefinition(name, typeDef);
+ alterTable.setAddNewColumn(columnDefinition);
+ }
+
+ alterTable.setAlterTableOpType(determineAlterTableType(ctx));
+
+ return alterTable;
+ }
+
+ private AlterTableOpType determineAlterTableType(SQLParser.Alter_table_statementContext ctx) {
+
+ final int RENAME_MASK = 00000001;
+ final int COLUMN_MASK = 00000010;
+ final int TO_MASK = 00000100;
+ final int ADD_MASK = 00001000;
+
+ int val = 00000000;
+
+ for (int idx = 1; idx < ctx.getChildCount(); idx++) {
+
+ if (ctx.getChild(idx) instanceof TerminalNode) {
+ if (((TerminalNode) ctx.getChild(idx)).getSymbol().getType() == RENAME) {
+ val = val | RENAME_MASK;
+ }
+ if (((TerminalNode) ctx.getChild(idx)).getSymbol().getType() == COLUMN) {
+ val = val | COLUMN_MASK;
+ }
+ if (((TerminalNode) ctx.getChild(idx)).getSymbol().getType() == TO) {
+ val = val | TO_MASK;
+ }
+ if (((TerminalNode) ctx.getChild(idx)).getSymbol().getType() == ADD) {
+ val = val | ADD_MASK;
+ }
+ }
+ }
+ return evaluateAlterTableOperationTye(val);
+ }
+
+ private AlterTableOpType evaluateAlterTableOperationTye(final int value) {
+
+ switch (value) {
+ case 65:
+ return AlterTableOpType.RENAME_TABLE;
+ case 73:
+ return AlterTableOpType.RENAME_COLUMN;
+ case 520:
+ return AlterTableOpType.ADD_COLUMN;
+ default:
+ return null;
+ }
+ }
+ }
http://git-wip-us.apache.org/repos/asf/tajo/blob/1b3ec373/tajo-core/src/test/java/org/apache/tajo/engine/parser/TestSQLAnalyzer.java
----------------------------------------------------------------------
diff --cc tajo-core/src/test/java/org/apache/tajo/engine/parser/TestSQLAnalyzer.java
index 0000000,2010502..f372ade
mode 000000,100644..100644
--- a/tajo-core/src/test/java/org/apache/tajo/engine/parser/TestSQLAnalyzer.java
+++ b/tajo-core/src/test/java/org/apache/tajo/engine/parser/TestSQLAnalyzer.java
@@@ -1,0 -1,473 +1,526 @@@
+ /**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+ package org.apache.tajo.engine.parser;
+
+ import org.antlr.v4.runtime.ANTLRInputStream;
+ import org.antlr.v4.runtime.CommonTokenStream;
+ import org.apache.tajo.algebra.CreateTable;
+ import org.apache.tajo.algebra.Expr;
+ import org.apache.tajo.algebra.LiteralValue;
+ import org.apache.tajo.algebra.OpType;
+ import org.apache.tajo.engine.parser.SQLParser.SqlContext;
+ import org.apache.tajo.util.FileUtil;
+ import org.junit.Test;
+
+ import java.io.File;
+ import java.io.IOException;
+ import java.util.Iterator;
+
+ import static org.junit.Assert.assertEquals;
+ import static org.junit.Assert.assertTrue;
+
+ /**
+ * This unit tests uses a number of query files located in tajo/tajo-core/src/test/resources/queries.
+ * So, you must set tajo/tajo-core/ as the working directory.
+ */
+ public class TestSQLAnalyzer {
+
+ public static Expr parseQuery(String sql) {
+ ANTLRInputStream input = new ANTLRInputStream(sql);
+ SQLLexer lexer = new SQLLexer(input);
+ CommonTokenStream tokens = new CommonTokenStream(lexer);
+ SQLParser parser = new SQLParser(tokens);
+ parser.setBuildParseTree(true);
+ SQLAnalyzer visitor = new SQLAnalyzer();
+ SqlContext context = parser.sql();
+ return visitor.visitSql(context);
+ }
+
++ public void assertParseResult(String sqlFileName, String resultFileName) throws IOException {
++ String sql = FileUtil.readTextFileFromResource("queries/TestSQLAnalyzer/" + sqlFileName);
++ String result = FileUtil.readTextFileFromResource("results/TestSQLAnalyzer/" + resultFileName);
++
++ Expr expr = parseQuery(sql);
++ assertEquals(expr.toJson().trim(), result.trim());
++ }
++
+
+ @Test
+ public void testSelect1() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/select_1.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testSelect2() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/select_2.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testSelect3() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/select_3.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testSelect4() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/select_4.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testSelect5() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/select_5.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testAsterisk1() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/asterisk_1.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testAsterisk2() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/asterisk_2.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testAsterisk3() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/asterisk_3.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testAsterisk4() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/asterisk_4.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testGroupby1() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/groupby_1.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testJoin1() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/join_1.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testJoin2() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/join_2.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testJoin3() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/join_3.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testJoin4() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/join_4.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testJoin5() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/join_5.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testJoin6() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/join_6.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testJoin7() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/join_7.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testJoin8() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/join_8.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testJoin9() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/join_9.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testJoin10() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/join_10.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testJoin11() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/join_11.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testSet1() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/set_1.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testSet2() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/set_2.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testSet3() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/set_3.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testSet4() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/set_4.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testDropTable() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/drop_table.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testCreateTable1() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_1.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testCreateTable2() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_2.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testCreateTable3() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_3.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testCreateTable4() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_4.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testCreateTable5() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_5.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testCreateTable6() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_6.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testCreateTable7() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_7.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testCreateTable8() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_8.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testCreateTable9() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_9.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testCreateTable10() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_10.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testCreateTablePartitionByHash1() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_partition_by_hash_1.sql"));
+ Expr expr = parseQuery(sql);
+ assertEquals(OpType.CreateTable, expr.getType());
+ CreateTable createTable = (CreateTable) expr;
+ assertTrue(createTable.hasPartition());
+ assertEquals(CreateTable.PartitionType.HASH, createTable.getPartitionMethod().getPartitionType());
+ CreateTable.HashPartition hashPartition = createTable.getPartitionMethod();
+ assertEquals("col1", hashPartition.getColumns()[0].getCanonicalName());
+ assertTrue(hashPartition.hasQuantifier());
+ }
+
+ @Test
+ public void testCreateTablePartitionByHash2() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_partition_by_hash_2.sql"));
+ Expr expr = parseQuery(sql);
+ assertEquals(OpType.CreateTable, expr.getType());
+ CreateTable createTable = (CreateTable) expr;
+ assertTrue(createTable.hasPartition());
+ assertEquals(CreateTable.PartitionType.HASH, createTable.getPartitionMethod().getPartitionType());
+ CreateTable.HashPartition hashPartition = createTable.getPartitionMethod();
+ assertEquals("col1", hashPartition.getColumns()[0].getCanonicalName());
+ assertTrue(hashPartition.hasSpecifiers());
+ assertEquals(3, hashPartition.getSpecifiers().size());
+ }
+
+ @Test
+ public void testCreateTablePartitionByRange() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_partition_by_range.sql"));
+ Expr expr = parseQuery(sql);
+ assertEquals(OpType.CreateTable, expr.getType());
+ CreateTable createTable = (CreateTable) expr;
+ assertTrue(createTable.hasPartition());
+ assertEquals(CreateTable.PartitionType.RANGE, createTable.getPartitionMethod().getPartitionType());
+ CreateTable.RangePartition rangePartition = createTable.getPartitionMethod();
+ assertEquals("col1", rangePartition.getColumns()[0].getCanonicalName());
+ assertEquals(3, rangePartition.getSpecifiers().size());
+ }
+
+ @Test
+ public void testCreateTablePartitionByList() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_partition_by_list.sql"));
+ Expr expr = parseQuery(sql);
+ assertEquals(OpType.CreateTable, expr.getType());
+ CreateTable createTable = (CreateTable) expr;
+ assertTrue(createTable.hasPartition());
+ assertEquals(CreateTable.PartitionType.LIST, createTable.getPartitionMethod().getPartitionType());
+ CreateTable.ListPartition listPartition = createTable.getPartitionMethod();
+ assertEquals("col1", listPartition.getColumns()[0].getCanonicalName());
+ assertEquals(2, listPartition.getSpecifiers().size());
+ Iterator<CreateTable.ListPartitionSpecifier> iterator = listPartition.getSpecifiers().iterator();
+ CreateTable.ListPartitionSpecifier specifier = iterator.next();
+ LiteralValue value1 = (LiteralValue) specifier.getValueList().getValues()[0];
+ LiteralValue value2 = (LiteralValue) specifier.getValueList().getValues()[1];
+ assertEquals("Seoul", value1.getValue());
+ assertEquals("서울", value2.getValue());
+
+ specifier = iterator.next();
+ value1 = (LiteralValue) specifier.getValueList().getValues()[0];
+ value2 = (LiteralValue) specifier.getValueList().getValues()[1];
+ assertEquals("Busan", value1.getValue());
+ assertEquals("부산", value2.getValue());
+ }
+
+ @Test
+ public void testCreateTablePartitionByColumn() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/create_table_partition_by_column.sql"));
+ Expr expr = parseQuery(sql);
+ assertEquals(OpType.CreateTable, expr.getType());
+ CreateTable createTable = (CreateTable) expr;
+ assertTrue(createTable.hasPartition());
+ assertEquals(CreateTable.PartitionType.COLUMN, createTable.getPartitionMethod().getPartitionType());
+ CreateTable.ColumnPartition columnPartition = createTable.getPartitionMethod();
+ assertEquals(3, columnPartition.getColumns().length);
+ assertEquals("col3", columnPartition.getColumns()[0].getColumnName());
+ assertEquals("col4", columnPartition.getColumns()[1].getColumnName());
+ assertEquals("col5", columnPartition.getColumns()[2].getColumnName());
+ }
+
+ @Test
+ public void testTableSubQuery1() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/table_subquery1.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testTableSubQuery2() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/table_subquery2.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testInSubquery1() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/in_subquery_1.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testInSubquery2() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/in_subquery_2.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testExistsPredicate1() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/exists_predicate_1.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testExistsPredicate2() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/exists_predicate_2.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testInsertIntoTable() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/insert_into_select_1.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testInsertIntoLocation() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/insert_into_select_2.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testInsertIntoTable2() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/insert_into_select_3.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testInsertOverwriteIntoTable() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/insert_overwrite_into_select_1.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testInsertOverwriteIntoLocation() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/insert_overwrite_into_select_2.sql"));
+ parseQuery(sql);
+ }
+
+ @Test
+ public void testInsertOverwriteIntoTable2() throws IOException {
+ String sql = FileUtil.readTextFile(new File("src/test/resources/queries/default/insert_overwrite_into_select_3.sql"));
+ parseQuery(sql);
+ }
+
+ static String[] exprs = {
+ "1 + 2", // 0
+ "3 - 4", // 1
+ "5 * 6", // 2
+ "7 / 8", // 3
+ "10 % 2", // 4
+ "1 * 2 > 3 / 4", // 5
+ "1 * 2 < 3 / 4", // 6
+ "1 * 2 = 3 / 4", // 7
+ "1 * 2 != 3 / 4", // 8
+ "1 * 2 <> 3 / 4", // 9
+ "gender in ('male', 'female')", // 10
+ "gender not in ('male', 'female')", // 11
+ "score > 90 and age < 20", // 12
+ "score > 90 and age < 20 and name != 'hyunsik'", // 13
+ "score > 90 or age < 20", // 14
+ "score > 90 or age < 20 and name != 'hyunsik'", // 15
+ "((a+3 > 1) or 1=1) and (3 != (abc + 4) and type in (3,4))", // 16
+ "3", // 17
+ "1.2", // 18
+ "sum(age)", // 19
+ "now()", // 20
+ "not (90 > 100)", // 21
+ "type like '%top'", // 22
+ "type not like 'top%'", // 23
+ "col = 'value'", // 24
+ "col is null", // 25
+ "col is not null", // 26
+ "col = null", // 27
+ "col != null", // 38
+ };
+
+ public static Expr parseExpr(String sql) {
+ ANTLRInputStream input = new ANTLRInputStream(sql);
+ SQLLexer lexer = new SQLLexer(input);
+ CommonTokenStream tokens = new CommonTokenStream(lexer);
+ SQLParser parser = new SQLParser(tokens);
+ parser.setBuildParseTree(true);
+ SQLAnalyzer visitor = new SQLAnalyzer();
+ SQLParser.Value_expressionContext context = parser.value_expression();
+ return visitor.visitValue_expression(context);
+ }
+
+ @Test
+ public void testExprs() {
+ for (int i = 0; i < exprs.length; i++) {
+ parseExpr(exprs[i]);
+ }
+ }
++
++ @Test
++ public void windowFunction1() throws IOException {
++ assertParseResult("window1.sql", "window1.result");
++ }
++
++ @Test
++ public void windowFunction2() throws IOException {
++ assertParseResult("window2.sql", "window2.result");
++ }
++
++ @Test
++ public void windowFunction3() throws IOException {
++ assertParseResult("window3.sql", "window3.result");
++ }
++
++ @Test
++ public void windowFunction4() throws IOException {
++ assertParseResult("window4.sql", "window4.result");
++ }
++
++ @Test
++ public void windowFunction5() throws IOException {
++ assertParseResult("window5.sql", "window5.result");
++ }
++
++ @Test
++ public void windowFunction6() throws IOException {
++ assertParseResult("window6.sql", "window6.result");
++ }
++
++ @Test
++ public void windowFunction7() throws IOException {
++ assertParseResult("window7.sql", "window7.result");
++ }
++
++ @Test
++ public void windowFunction8() throws IOException {
++ assertParseResult("window8.sql", "window8.result");
++ }
++
++ @Test
++ public void windowFunction9() throws IOException {
++ assertParseResult("window9.sql", "window9.result");
++ }
+ }