add join convert rule

This commit is contained in:
XudongXie 2024-09-12 16:18:41 +08:00
parent 6fcb0541d0
commit d4dca20bb9
4 changed files with 49 additions and 12 deletions

View File

@ -4,13 +4,11 @@
<option name="autoReloadType" value="SELECTIVE" />
</component>
<component name="ChangeListManager">
<list default="true" id="d8d12963-4777-48e3-9669-309cf5e41d60" name="更改" comment="add join class">
<change afterPath="$PROJECT_DIR$/src/main/java/Parser/AST/Join/JoinEndNode.java" afterDir="false" />
<list default="true" id="d8d12963-4777-48e3-9669-309cf5e41d60" name="更改" comment="add join">
<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/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/Join/JoinColumnNode.java" beforeDir="false" afterPath="$PROJECT_DIR$/src/main/java/Parser/AST/Join/JoinConditionNode.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" />
<change beforePath="$PROJECT_DIR$/src/main/java/Parser/AST/Join/JoinConditionNode.java" beforeDir="false" afterPath="$PROJECT_DIR$/src/main/java/Parser/AST/Join/JoinConditionNode.java" afterDir="false" />
</list>
<option name="SHOW_DIALOG" value="false" />
<option name="HIGHLIGHT_CONFLICTS" value="true" />
@ -374,7 +372,14 @@
<option name="project" value="LOCAL" />
<updated>1726112198384</updated>
</task>
<option name="localTasksCounter" value="39" />
<task id="LOCAL-00039" summary="add join">
<created>1726127728384</created>
<option name="number" value="00039" />
<option name="presentableId" value="LOCAL-00039" />
<option name="project" value="LOCAL" />
<updated>1726127728384</updated>
</task>
<option name="localTasksCounter" value="40" />
<servers />
</component>
<component name="Vcs.Log.Tabs.Properties">
@ -408,7 +413,8 @@
<MESSAGE value="revise" />
<MESSAGE value="add join pattern" />
<MESSAGE value="add join class" />
<option name="LAST_COMMIT_MESSAGE" value="add join class" />
<MESSAGE value="add join" />
<option name="LAST_COMMIT_MESSAGE" value="add join" />
</component>
<component name="XDebuggerManager">
<watches-manager>

View File

@ -8,6 +8,9 @@ import Exception.GenerateFailedException;
import Parser.AST.DropTable.DropTableNode;
import Parser.AST.DropTable.DropTableOptionNode;
import Parser.AST.Insert.InsertNode;
import Parser.AST.Join.JoinConditionNode;
import Parser.AST.Join.JoinSourceTabNode;
import Parser.AST.Join.JoinTypeNode;
import Parser.AST.Select.SelectNode;
import java.util.ArrayList;
@ -18,6 +21,11 @@ public class OpenGaussGenerator {
public OpenGaussGenerator(ASTNode node) {
this.node = node;
}
public void setNode(ASTNode node) {
this.node = node;
}
public String generate() {
if (node instanceof CreateTabNode) {
return GenCreatTableSQL(node);
@ -31,6 +39,9 @@ public class OpenGaussGenerator {
else if (node instanceof SelectNode) {
return GenSelectSQL(node);
}
else if (node instanceof JoinSourceTabNode) {
return GenJoinSQL(node);
}
else {
try {
throw new GenerateFailedException("Root node:" + node.getClass() + "(Unsupported node type!)");
@ -63,6 +74,11 @@ public class OpenGaussGenerator {
return node.toQueryString();
}
private String GenJoinSQL(ASTNode node) {
visitJoin(node);
return node.toQueryString();
}
private void visitCrt(ASTNode node) {
if (node instanceof ColumnNode) {
ColumnTypeConvert((ColumnNode) node);
@ -91,6 +107,17 @@ public class OpenGaussGenerator {
}
}
private void visitJoin(ASTNode node) {
if (node instanceof JoinConditionNode) {
if (((JoinConditionNode) node).getKeyword().equalsIgnoreCase("USING")) {
((JoinConditionNode) node).setKeyword("ON");
}
}
for (ASTNode child : node.getChildren()) {
visitJoin(child);
}
}
private void ColumnTypeConvert(ColumnNode node) {
// type convert
if (node.getType().getValue().equalsIgnoreCase("NUMBER")) {

View File

@ -16,14 +16,14 @@ public class Main {
// ");";
// String sql = "INSERT INTO employees (first_name, last_name, email) VALUES ('Jane', 'Smith', 'janesm@example.com');";
// String sql = "DROP TABLE employees CASCADE CONSTRAINTS;";
String sql = "SELECT e.first_name, d.department_name FROM employees e JOIN departments d ON e.department_id = d.department_id;";
// String sql = "Customers\n" +
// "LEFT JOIN Orders ON Customers.CustomerID = Orders.CustomerID where";
// String sql = "SELECT e.first_name, d.department_name FROM employees e JOIN departments d ON e.department_id = d.department_id;";
String sql = "Customers\n" +
"LEFT JOIN Orders USING Customers.CustomerID = Orders.CustomerID where";
OracleLexer lexer = new OracleLexer(sql);
lexer.printTokens();
OracleParser parser = new OracleParser(lexer);
ASTNode root = parser.parse();
// ASTNode root = OracleParser.parseJoin(lexer.getTokens());
// ASTNode root = parser.parse();
ASTNode root = OracleParser.parseJoin(lexer.getTokens());
System.out.println(root.toQueryString());
System.out.println(root.getASTString());
OpenGaussGenerator generator = new OpenGaussGenerator(root);

View File

@ -22,6 +22,10 @@ public class JoinConditionNode extends ASTNode {
super();
}
public String getKeyword() {
return keyword.replace(" ", "");
}
public void setKeyword(String keyword) {
this.keyword = keyword + " ";
}