add function

This commit is contained in:
XudongXie 2024-08-23 22:36:37 +08:00
parent bae72996a4
commit 7eb3119503
3 changed files with 76 additions and 22 deletions

View File

@ -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">

View File

@ -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());
}
}

View File

@ -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 {