Taoyouce
/
examples
forked from opengaussexamples/examples
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add parseSelect

This commit is contained in:
XudongXie 2024-09-11 15:55:54 +08:00
parent 995516a90d
commit e6cc96a21f
2 changed files with 146 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
sqlTranslate/.idea/workspace.xml
View File

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

122
sqlTranslate/src/main/java/Parser/OracleParser.java
View File

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