From 7eb3119503c7663b74f7e6ed5401fd9938bcf284 Mon Sep 17 00:00:00 2001
From: XudongXie <xiexudong23@otcaix.iscas.ac.cn>
Date: Fri, 23 Aug 2024 22:36:37 +0800
Subject: [PATCH] add function

---
 sqlTranslate/.idea/workspace.xml              | 30 +++++++----
 sqlTranslate/src/main/java/Main.java          | 16 +++---
 .../src/main/java/Parser/OracleParser.java    | 52 +++++++++++++++++--
 3 files changed, 76 insertions(+), 22 deletions(-)

diff --git a/sqlTranslate/.idea/workspace.xml b/sqlTranslate/.idea/workspace.xml
index 0e43459d..c3f62bc4 100644
--- a/sqlTranslate/.idea/workspace.xml
+++ b/sqlTranslate/.idea/workspace.xml
@@ -6,7 +6,7 @@
   <component name="ChangeListManager">
     <list default="true" id="d8d12963-4777-48e3-9669-309cf5e41d60" name="更改" comment="add function">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/src/main/java/Parser/AST/CreateTable/ColumnNode.java" beforeDir="false" afterPath="$PROJECT_DIR$/src/main/java/Parser/AST/CreateTable/ColumnNode.java" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/src/main/java/Main.java" beforeDir="false" afterPath="$PROJECT_DIR$/src/main/java/Main.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" />
@@ -40,15 +40,18 @@
     <option name="hideEmptyMiddlePackages" value="true" />
     <option name="showLibraryContents" value="true" />
   </component>
-  <component name="PropertiesComponent">{
-  &quot;keyToString&quot;: {
-    &quot;RunOnceActivity.OpenProjectViewOnStart&quot;: &quot;true&quot;,
-    &quot;RunOnceActivity.ShowReadmeOnStart&quot;: &quot;true&quot;,
-    &quot;SHARE_PROJECT_CONFIGURATION_FILES&quot;: &quot;true&quot;,
-    &quot;last_opened_file_path&quot;: &quot;F:/oracle2opengauss/examples/sqlTranslate&quot;,
-    &quot;settings.editor.selected.configurable&quot;: &quot;preferences.pluginManager&quot;
+  <component name="PropertiesComponent"><![CDATA[{
+  "keyToString": {
+    "RunOnceActivity.OpenProjectViewOnStart": "true",
+    "RunOnceActivity.ShowReadmeOnStart": "true",
+    "SHARE_PROJECT_CONFIGURATION_FILES": "true",
+    "last_opened_file_path": "F:/oracle2opengauss/examples/sqlTranslate",
+    "project.structure.last.edited": "模块",
+    "project.structure.proportion": "0.0",
+    "project.structure.side.proportion": "0.0",
+    "settings.editor.selected.configurable": "build.tools"
   }
-}</component>
+}]]></component>
   <component name="RunManager" selected="应用程序.Main">
     <configuration name="App" type="Application" factoryName="Application" temporary="true" nameIsGenerated="true">
       <option name="MAIN_CLASS_NAME" value="org.example.App" />
@@ -180,7 +183,14 @@
       <option name="project" value="LOCAL" />
       <updated>1724338301831</updated>
     </task>
-    <option name="localTasksCounter" value="12" />
+    <task id="LOCAL-00012" summary="add function">
+      <created>1724419179050</created>
+      <option name="number" value="00012" />
+      <option name="presentableId" value="LOCAL-00012" />
+      <option name="project" value="LOCAL" />
+      <updated>1724419179050</updated>
+    </task>
+    <option name="localTasksCounter" value="13" />
     <servers />
   </component>
   <component name="Vcs.Log.Tabs.Properties">
diff --git a/sqlTranslate/src/main/java/Main.java b/sqlTranslate/src/main/java/Main.java
index ff19ccf5..fc8e3791 100644
--- a/sqlTranslate/src/main/java/Main.java
+++ b/sqlTranslate/src/main/java/Main.java
@@ -1,11 +1,11 @@
-import Lexer.OracleLexer;
-import Lexer.Token;
 
-import java.util.List;
+import Lexer.OracleLexer;
+import Parser.AST.ASTNode;
+import Parser.OracleParser;
+
 
 public class Main {
     public static void main(String[] args) {
-//        String sql = "SELECT NuMBER(1,3) Not NULL FROM table WHERE name = 'John Doe';";
         String sql = "CREATE TABLE employees (\n" +
                 "    employee_id NUMBER(6) PRIMARY KEY,\n" +
                 "    first_name VARCHAR2(20),\n" +
@@ -15,10 +15,8 @@ public class Main {
                 ");";
         OracleLexer lexer = new OracleLexer(sql);
         lexer.printTokens();
-//        List<Token> tokens = lexer.tokenize();
-//
-//        for (Token token : tokens) {
-//            System.out.println(token);
-//        }
+        OracleParser parser = new OracleParser(lexer);
+        ASTNode root = parser.parse();
+        System.out.println(root.toQueryString());
     }
 }
diff --git a/sqlTranslate/src/main/java/Parser/OracleParser.java b/sqlTranslate/src/main/java/Parser/OracleParser.java
index b514da91..caff544f 100644
--- a/sqlTranslate/src/main/java/Parser/OracleParser.java
+++ b/sqlTranslate/src/main/java/Parser/OracleParser.java
@@ -95,7 +95,8 @@ public class OracleParser {
                     tokens.add(lexer.getTokens().get(j));
                     // Check () or REFERENCES other_table(other_column)
                     if (lexer.getTokens().get(j).hasType(Token.TokenType.KEYWORD) &&
-                            (lexer.getTokens().get(j).getValue().equalsIgnoreCase("REFERENCES") || lexer.getTokens().get(j).getValue().equalsIgnoreCase("CHECK"))) {
+                            (lexer.getTokens().get(j).getValue().equalsIgnoreCase("REFERENCES")
+                                    || lexer.getTokens().get(j).getValue().equalsIgnoreCase("CHECK"))) {
                         Stack<String> stack = new Stack<>();
                         for (int k = j + 1; k < lexer.getTokens().size(); k++) {
                             tokens.add(lexer.getTokens().get(k));
@@ -124,18 +125,63 @@ public class OracleParser {
                     if ((lexer.getTokens().get(j).hasType(Token.TokenType.SYMBOL) && lexer.getTokens().get(j).getValue().equals(",")) ||
                             (lexer.getTokens().get(j).hasType(Token.TokenType.SYMBOL) && lexer.getTokens().get(j).getValue().equals(")")) ) {
                         i = j;
-                        child.setConstraint(constraint);
                         break;
                     }
                 }
                 child.setTokens(tokens);
+                child.setConstraint(constraint);
                 currentNode.addChild(child);
                 currentNode = child;
             }
-            else if (lexer.getTokens().get(i).hasType(Token.TokenType.KEYWORD) && lexer.getTokens().get(i).getValue().equalsIgnoreCase("CONSTRAINT")) {
+            else if (lexer.getTokens().get(i).hasType(Token.TokenType.KEYWORD)
+                    && lexer.getTokens().get(i).getValue().equalsIgnoreCase("CONSTRAINT")) {
                 // Table constraint
                 tokens.clear();
                 tokens.add(lexer.getTokens().get(i));
+                for (int j = i + 1; j < lexer.getTokens().size(); j++) {
+                    tokens.add(lexer.getTokens().get(j));
+                    /**
+                     * CONSTRAINT pk_example PRIMARY KEY (column1, column2)
+                     * CONSTRAINT uk_example UNIQUE (column1)
+                     * CONSTRAINT fk_example FOREIGN KEY (column1) REFERENCES other_table(column2)
+                     * CONSTRAINT chk_example CHECK (column1 > 0)
+                     */
+                    if (lexer.getTokens().get(j).hasType(Token.TokenType.KEYWORD) &&
+                            (lexer.getTokens().get(j).getValue().equalsIgnoreCase("UNIQUE")
+                                    || lexer.getTokens().get(j).getValue().equalsIgnoreCase("CHECK")
+                                    || lexer.getTokens().get(j).getValue().equalsIgnoreCase("PRIMARY KEY")
+                                    || lexer.getTokens().get(j).getValue().equalsIgnoreCase("FOREIGN KEY")
+                                    || lexer.getTokens().get(j).getValue().equalsIgnoreCase("REFERENCES"))) {
+                        Stack<String> stack = new Stack<>();
+                        for (int k = j + 1; k < lexer.getTokens().size(); k++) {
+                            tokens.add(lexer.getTokens().get(k));
+                            if (lexer.getTokens().get(k).getValue().equals("(")) {
+                                stack.push("(");
+                                for (int t = k + 1; t < lexer.getTokens().size(); t++) {
+                                    tokens.add(lexer.getTokens().get(t));
+                                    if (lexer.getTokens().get(t).getValue().equals("(")) {
+                                        stack.push("(");
+                                    }
+                                    else if (lexer.getTokens().get(t).getValue().equals(")")) {
+                                        stack.pop();
+                                        if (stack.empty()) {
+                                            i = t;
+                                            j = t;
+                                            break;
+                                        }
+                                    }
+                                }
+                                break;
+                            }
+                        }
+                        continue;
+                    }
+                    if ((lexer.getTokens().get(j).hasType(Token.TokenType.SYMBOL) && lexer.getTokens().get(j).getValue().equals(",")) ||
+                            (lexer.getTokens().get(j).hasType(Token.TokenType.SYMBOL) && lexer.getTokens().get(j).getValue().equals(")")) ) {
+                        i = j;
+                        break;
+                    }
+                }
 
             }
             else {