From e6cc96a21f8bc1ea9b2b869ad50a2bf7edb8ce10 Mon Sep 17 00:00:00 2001
From: XudongXie <xiexudong23@otcaix.iscas.ac.cn>
Date: Wed, 11 Sep 2024 15:55:54 +0800
Subject: [PATCH] add parseSelect

---
 sqlTranslate/.idea/workspace.xml              |  32 ++++-
 .../src/main/java/Parser/OracleParser.java    | 122 +++++++++++++++++-
 2 files changed, 146 insertions(+), 8 deletions(-)

diff --git a/sqlTranslate/.idea/workspace.xml b/sqlTranslate/.idea/workspace.xml
index 7fd597c4..3c8fca74 100644
--- a/sqlTranslate/.idea/workspace.xml
+++ b/sqlTranslate/.idea/workspace.xml
@@ -4,9 +4,9 @@
     <option name="autoReloadType" value="SELECTIVE" />
   </component>
   <component name="ChangeListManager">
-    <list default="true" id="d8d12963-4777-48e3-9669-309cf5e41d60" name="更改" comment="revise">
+    <list default="true" id="d8d12963-4777-48e3-9669-309cf5e41d60" name="更改" comment="add parseSelect">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/src/main/java/Lexer/OracleLexer.java" beforeDir="false" afterPath="$PROJECT_DIR$/src/main/java/Lexer/OracleLexer.java" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/src/main/java/Parser/OracleParser.java" beforeDir="false" afterPath="$PROJECT_DIR$/src/main/java/Parser/OracleParser.java" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -307,7 +307,28 @@
       <option name="project" value="LOCAL" />
       <updated>1725978582486</updated>
     </task>
-    <option name="localTasksCounter" value="30" />
+    <task id="LOCAL-00030" summary="add pattern of functions in select">
+      <created>1725981063195</created>
+      <option name="number" value="00030" />
+      <option name="presentableId" value="LOCAL-00030" />
+      <option name="project" value="LOCAL" />
+      <updated>1725981063195</updated>
+    </task>
+    <task id="LOCAL-00031" summary="add some comments">
+      <created>1725981511612</created>
+      <option name="number" value="00031" />
+      <option name="presentableId" value="LOCAL-00031" />
+      <option name="project" value="LOCAL" />
+      <updated>1725981511612</updated>
+    </task>
+    <task id="LOCAL-00032" summary="add parseSelect">
+      <created>1725981975960</created>
+      <option name="number" value="00032" />
+      <option name="presentableId" value="LOCAL-00032" />
+      <option name="project" value="LOCAL" />
+      <updated>1725981975960</updated>
+    </task>
+    <option name="localTasksCounter" value="33" />
     <servers />
   </component>
   <component name="Vcs.Log.Tabs.Properties">
@@ -336,7 +357,10 @@
     <MESSAGE value="add select" />
     <MESSAGE value="add case when" />
     <MESSAGE value="revise" />
-    <option name="LAST_COMMIT_MESSAGE" value="revise" />
+    <MESSAGE value="add pattern of functions in select" />
+    <MESSAGE value="add some comments" />
+    <MESSAGE value="add parseSelect" />
+    <option name="LAST_COMMIT_MESSAGE" value="add parseSelect" />
   </component>
   <component name="XDebuggerManager">
     <watches-manager>
diff --git a/sqlTranslate/src/main/java/Parser/OracleParser.java b/sqlTranslate/src/main/java/Parser/OracleParser.java
index f1f65796..b441c3c2 100644
--- a/sqlTranslate/src/main/java/Parser/OracleParser.java
+++ b/sqlTranslate/src/main/java/Parser/OracleParser.java
@@ -17,8 +17,7 @@ import Parser.AST.Insert.InsertDataNode;
 import Parser.AST.Insert.InsertEndNode;
 import Parser.AST.Insert.InsertNode;
 import Parser.AST.Insert.InsertObjNode;
-import Parser.AST.Select.SelectNode;
-import Parser.AST.Select.SelectObjNode;
+import Parser.AST.Select.*;
 
 import java.util.Stack;
 
@@ -381,8 +380,123 @@ public class OracleParser {
             if (i == 1 && parseTokens.get(i).hasType(Token.TokenType.KEYWORD) && parseTokens.get(i).getValue().equalsIgnoreCase("DISTINCT")) {
                 SelectObjNode childNode = new SelectObjNode();
                 childNode.setIsDistinct(parseTokens.get(i).getValue());
+                tokens = new ArrayList<>();
                 for (int j = i + 1; j < parseTokens.size(); j++) {
-                    //TODO: match select_obj and reindex i
+                    tokens.add(parseTokens.get(j));
+                    if (parseTokens.get(j).hasType(Token.TokenType.KEYWORD) && parseTokens.get(j).getValue().equalsIgnoreCase("FROM")) {
+                        i = j;
+                        break;
+                    }
+                }
+                childNode.setTokens(tokens);
+                currentNode.addChild(childNode);
+                currentNode = childNode;
+            }
+            else if (i == 1 && parseTokens.get(i).hasType(Token.TokenType.KEYWORD) && !parseTokens.get(i).getValue().equalsIgnoreCase("DISTINCT")) {
+                tokens = new ArrayList<>();
+                for (int j = i; j < parseTokens.size(); j++) {
+                    tokens.add(parseTokens.get(j));
+                    if (parseTokens.get(j).hasType(Token.TokenType.KEYWORD) && parseTokens.get(j).getValue().equalsIgnoreCase("FROM")) {
+                        i = j;
+                        break;
+                    }
+                }
+                ASTNode childNode = new SelectObjNode(tokens);
+                currentNode.addChild(childNode);
+                currentNode = childNode;
+            }
+            // match select_tab (possible last token: ; | GROUP | ORDER | HAVING | WHERE | UNION)
+            else if (currentNode instanceof SelectObjNode) {
+                tokens = new ArrayList<>();
+                for (int j = i; j < parseTokens.size(); j++) {
+                    tokens.add(parseTokens.get(j));
+                    if (
+                            (parseTokens.get(j).hasType(Token.TokenType.KEYWORD) && parseTokens.get(j).getValue().equalsIgnoreCase("WHERE"))
+                            || (parseTokens.get(j).hasType(Token.TokenType.KEYWORD) && parseTokens.get(j).getValue().equalsIgnoreCase("GROUP"))
+                            || (parseTokens.get(j).hasType(Token.TokenType.KEYWORD) && parseTokens.get(j).getValue().equalsIgnoreCase("ORDER"))
+                            || (parseTokens.get(j).hasType(Token.TokenType.KEYWORD) && parseTokens.get(j).getValue().equalsIgnoreCase("HAVING"))
+                            || (parseTokens.get(j).hasType(Token.TokenType.KEYWORD) && parseTokens.get(j).getValue().equalsIgnoreCase("UNION"))
+                            || (parseTokens.get(j).hasType(Token.TokenType.SYMBOL) && parseTokens.get(j).getValue().equals(";"))
+                    ) {
+                        i = j - 1;
+                        break;
+                    }
+                }
+                ASTNode childNode = new SelectTableNode(tokens);
+                currentNode.addChild(childNode);
+                currentNode = childNode;
+            }
+            // match where_clause (possible last token: ; | GROUP | ORDER | HAVING | UNION)
+            else if (parseTokens.get(i).hasType(Token.TokenType.KEYWORD) && parseTokens.get(i).getValue().equalsIgnoreCase("WHERE")) {
+                tokens = new ArrayList<>();
+                for (int j = i; j < parseTokens.size(); j++) {
+                    tokens.add(parseTokens.get(j));
+                    if (
+                            (parseTokens.get(j).hasType(Token.TokenType.KEYWORD) && parseTokens.get(j).getValue().equalsIgnoreCase("GROUP"))
+                            || (parseTokens.get(j).hasType(Token.TokenType.KEYWORD) && parseTokens.get(j).getValue().equalsIgnoreCase("ORDER"))
+                            || (parseTokens.get(j).hasType(Token.TokenType.KEYWORD) && parseTokens.get(j).getValue().equalsIgnoreCase("HAVING"))
+                            || (parseTokens.get(j).hasType(Token.TokenType.KEYWORD) && parseTokens.get(j).getValue().equalsIgnoreCase("UNION"))
+                            || (parseTokens.get(j).hasType(Token.TokenType.SYMBOL) && parseTokens.get(j).getValue().equals(";"))
+                    ) {
+                        i = j - 1;
+                        break;
+                    }
+                    ASTNode childNode = new SelectWhereClauseNode(tokens);
+                    currentNode.addChild(childNode);
+                    currentNode = childNode;
+                }
+            }
+            // match select_option
+            else if (
+                    (parseTokens.get(i).hasType(Token.TokenType.KEYWORD) && parseTokens.get(i).getValue().equalsIgnoreCase("GROUP"))
+                    || (parseTokens.get(i).hasType(Token.TokenType.KEYWORD) && parseTokens.get(i).getValue().equalsIgnoreCase("ORDER"))
+                    || (parseTokens.get(i).hasType(Token.TokenType.KEYWORD) && parseTokens.get(i).getValue().equalsIgnoreCase("HAVING"))
+            ) {
+                tokens = new ArrayList<>();
+                for (int j = i; j < parseTokens.size(); j++) {
+                    tokens.add(parseTokens.get(j));
+                    if (
+                          (parseTokens.get(j).hasType(Token.TokenType.KEYWORD) && parseTokens.get(j).getValue().equalsIgnoreCase("UNION"))
+                          || (parseTokens.get(j).hasType(Token.TokenType.SYMBOL) && parseTokens.get(j).getValue().equals(";"))
+                    ) {
+                        i = j - 1;
+                        break;
+                    }
+                    ASTNode childNode = new SelectOptionNode(tokens);
+                    currentNode.addChild(childNode);
+                    currentNode = childNode;
+                }
+            }
+            // match union_clause
+            else if (parseTokens.get(i).hasType(Token.TokenType.KEYWORD) && parseTokens.get(i).getValue().equalsIgnoreCase("UNION")) {
+                tokens = new ArrayList<>();
+                tokens.add(parseTokens.get(i));
+                ASTNode childNode = new SelectUnionNode(tokens);
+                currentNode.addChild(childNode);
+                currentNode = childNode;
+                List <Token> unionTokens = parseTokens.subList(i + 1, parseTokens.size());
+                ASTNode unionChildNode = parseSelect(unionTokens);
+                currentNode.addChild(unionChildNode);
+                currentNode = unionChildNode;
+                break;
+            }
+            // match ;
+            else if (parseTokens.get(i).hasType(Token.TokenType.SYMBOL) && parseTokens.get(i).getValue().equals(";")) {
+                tokens = new ArrayList<>();
+                tokens.add(parseTokens.get(i));
+                ASTNode childNode = new SelectEndNode(tokens);
+                currentNode.addChild(childNode);
+                currentNode = childNode;
+            }
+            else if (parseTokens.get(i).hasType(Token.TokenType.EOF)) {
+                break;
+            }
+            else {
+                try {
+                    throw new ParseFailedException("Parse failed!");
+                }
+                catch (ParseFailedException e) {
+                    e.printStackTrace();
                 }
             }
         }
@@ -393,7 +507,7 @@ public class OracleParser {
     /**
      * CASE WHEN expr THEN expr [ELSE expr] END
      */
-    private ASTNode parseCaseWhen(List<Token> parseTokens) {
+    public static ASTNode parseCaseWhen(List<Token> parseTokens) {
         List <Token> tokens = new ArrayList<>();
         tokens.add(parseTokens.get(0));