add view generate method

This commit is contained in:
XudongXie 2024-09-13 21:48:56 +08:00
parent c4cec7795c
commit aaa152a922
5 changed files with 51 additions and 33 deletions

View File

@ -4,12 +4,11 @@
<option name="autoReloadType" value="SELECTIVE" />
</component>
<component name="ChangeListManager">
<list default="true" id="d8d12963-4777-48e3-9669-309cf5e41d60" name="更改" comment="add ast class of view">
<list default="true" id="d8d12963-4777-48e3-9669-309cf5e41d60" name="更改" comment="add view">
<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/Document/Update statement in Oracle" beforeDir="false" afterPath="$PROJECT_DIR$/src/main/java/Document/Update statement in Oracle" afterDir="false" />
<change beforePath="$PROJECT_DIR$/src/main/java/Generator/OpenGaussGenerator.java" beforeDir="false" afterPath="$PROJECT_DIR$/src/main/java/Generator/OpenGaussGenerator.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/AST/View/ViewEndNode.java" beforeDir="false" afterPath="$PROJECT_DIR$/src/main/java/Parser/AST/View/ViewEndNode.java" afterDir="false" />
<change beforePath="$PROJECT_DIR$/src/main/java/Parser/AST/View/ViewNode.java" beforeDir="false" afterPath="$PROJECT_DIR$/src/main/java/Parser/AST/View/ViewCreateNode.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" />
@ -108,13 +107,6 @@
<option name="presentableId" value="Default" />
<updated>1723106789329</updated>
</task>
<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>
<task id="LOCAL-00013" summary="add function">
<created>1724423798948</created>
<option name="number" value="00013" />
@ -451,7 +443,14 @@
<option name="project" value="LOCAL" />
<updated>1726215657277</updated>
</task>
<option name="localTasksCounter" value="61" />
<task id="LOCAL-00061" summary="add view">
<created>1726234244501</created>
<option name="number" value="00061" />
<option name="presentableId" value="LOCAL-00061" />
<option name="project" value="LOCAL" />
<updated>1726234244501</updated>
</task>
<option name="localTasksCounter" value="62" />
<servers />
</component>
<component name="Vcs.Log.Tabs.Properties">
@ -466,7 +465,6 @@
</option>
</component>
<component name="VcsManagerConfiguration">
<MESSAGE value="add select" />
<MESSAGE value="add case when" />
<MESSAGE value="add pattern of functions in select" />
<MESSAGE value="add some comments" />
@ -491,7 +489,8 @@
<MESSAGE value="add view document" />
<MESSAGE value="add keywords of view" />
<MESSAGE value="add ast class of view" />
<option name="LAST_COMMIT_MESSAGE" value="add ast class of view" />
<MESSAGE value="add view" />
<option name="LAST_COMMIT_MESSAGE" value="add view" />
</component>
<component name="XDebuggerManager">
<watches-manager>

View File

@ -1 +1,4 @@
UPDATE table_name SET column1 = value1, column2 = value2, ... WHERE condition;
UPDATE table_name SET column1 = value1, column2 = value2, ... WHERE condition;
Example: UPDATE employees e JOIN departments d using e.department_id = d.department_id
SET e.salary = e.salary * 1.10, d.budget = d.budget * 1.10 WHERE d.department_name = 'Sales';

View File

@ -19,6 +19,7 @@ import Parser.AST.Select.SelectNode;
import Parser.AST.Select.SelectObjNode;
import Parser.AST.Update.UpdateNode;
import Parser.AST.Update.UpdateObjNode;
import Parser.AST.View.ViewCreateNode;
import Parser.OracleParser;
import java.util.ArrayList;
@ -59,6 +60,9 @@ public class OpenGaussGenerator {
else if (node instanceof AlterNode) {
return GenAlterSQL(node);
}
else if (node instanceof ViewCreateNode) {
return GenCreateViewSQL(node);
}
else {
try {
throw new GenerateFailedException("Root node:" + node.getClass() + "(Unsupported node type!)");
@ -88,7 +92,7 @@ public class OpenGaussGenerator {
}
private String GenSelectSQL(ASTNode node) {
visitSelect(node, null);
visitSelect(node);
// System.out.println(node.getASTString());
return node.toQueryString();
}
@ -114,6 +118,11 @@ public class OpenGaussGenerator {
return node.toQueryString();
}
private String GenCreateViewSQL(ASTNode node) {
visitCreateView(node);
return node.toQueryString();
}
private void visitCrt(ASTNode node) {
if (node instanceof ColumnNode) {
ColumnTypeConvert((ColumnNode) node);
@ -153,21 +162,12 @@ public class OpenGaussGenerator {
}
}
private void visitSelect(ASTNode node, ASTNode parentNode) {
if (node instanceof SelectObjNode) {
// check whether there exists a join statement in the select sql
if (node.getTokens().contains(new Token(Token.TokenType.KEYWORD, "JOIN"))) {
ASTNode joinRootNode = OracleParser.parseJoin(node.getTokens());
visitJoin(joinRootNode);
parentNode.replaceChild(node, joinRootNode);
ASTNode smallestChild = joinRootNode.getDeepestChild();
for (ASTNode child : node.getChildren()) {
smallestChild.addChild(child);
}
}
private void visitSelect(ASTNode node) {
if (node instanceof JoinSourceTabNode) {
visitJoin(node);
}
for (ASTNode child : node.getChildren()) {
visitSelect(child, node);
visitSelect(child);
}
}
@ -201,6 +201,15 @@ public class OpenGaussGenerator {
}
}
private void visitCreateView(ASTNode node) {
if (node instanceof SelectNode) {
visitSelect(node);
}
for (ASTNode child : node.getChildren()) {
visitCreateView(child);
}
}
private void ColumnTypeConvert(ColumnType node) {
// type convert
if (node.getType().getValue().equalsIgnoreCase("NUMBER")) {

View File

@ -32,7 +32,7 @@ public class Main {
// ");";
// String sql = "ALTER TABLE employees ADD email VARCHAR2(100) Check (email != '12813@163.com');";
// String sql = "CASE WHEN column2 > 0 THEN 'Positive' ELSE 'Non-positive' END";
String sql = "CREATE OR REPLACE VIEW emp_dept_info AS SELECT e.first_name, e.last_name, d.department_name FROM employees e JOIN departments d ON e.department_id = d.department_id;";
String sql = "CREATE OR REPLACE VIEW emp_dept_info AS SELECT e.first_name, e.last_name, d.department_name FROM employees e JOIN departments d Using e.department_id = d.department_id;";
OracleLexer lexer = new OracleLexer(sql);
lexer.printTokens();
OracleParser parser = new OracleParser(lexer);

View File

@ -450,9 +450,16 @@ public class OracleParser {
}
tokens.add(parseTokens.get(j));
}
ASTNode childNode = new SelectObjNode(tokens);
currentNode.addChild(childNode);
currentNode = childNode;
if (tokens.contains(new Token(Token.TokenType.KEYWORD, "JOIN"))) {
ASTNode joinRootNode = parseJoin(tokens);
currentNode.addChild(joinRootNode);
currentNode = joinRootNode.getDeepestChild();
}
else {
ASTNode childNode = new SelectObjNode(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")) {