Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

improved oracle pl/sql parser.

Browse Source
This commit is contained in:
wenshao
2017-05-13 05:03:10 +08:00
parent b6a15a3c23
commit 33914b35df
10 changed files with 364 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/sql/SQLUtils.java
+12 -8
View File
@@ -292,11 +292,15 @@ public class SQLUtils {
visitor.setTableMapping(tableMapping);
}
for (int i = 0; i < statementList.size(); i++) {
boolean printStmtSeperator = !JdbcConstants.ORACLE.equals(dbType);
for (int i = 0, size = statementList.size(); i < size; i++) {
SQLStatement stmt = statementList.get(i);
if (i > 0) {
visitor.print(";");
if (printStmtSeperator) {
visitor.print(";");
}
SQLStatement preStmt = statementList.get(i - 1);
List<String> comments = preStmt.getAfterCommentsDirect();
@@ -309,7 +313,10 @@ public class SQLUtils {
visitor.print(comment);
}
}
visitor.println();
if (printStmtSeperator) {
visitor.println();
}
if (!(stmt instanceof SQLSetStatement)) {
visitor.println();
@@ -325,12 +332,9 @@ public class SQLUtils {
}
stmt.accept(visitor);
if (i == statementList.size() - 1) {
if (i == size - 1) {
Boolean semi = (Boolean) stmt.getAttribute("format.semi");
if (semi != null && semi.booleanValue()) {
// if (stmt.hasAfterComment()) {
// visitor.println();
// }
if (semi != null && semi.booleanValue() && printStmtSeperator) {
visitor.print(";");
}
src/main/java/com/alibaba/druid/sql/dialect/oracle/ast/stmt/OracleForStatement.java
+9
View File
@@ -31,6 +31,8 @@ public class OracleForStatement extends OracleStatementImpl {
private List<SQLStatement> statements = new ArrayList<SQLStatement>();
private boolean all;
@Override
public void accept0(OracleASTVisitor visitor) {
if (visitor.visit(this)) {
@@ -65,4 +67,11 @@ public class OracleForStatement extends OracleStatementImpl {
this.statements = statements;
}
public boolean isAll() {
return all;
}
public void setAll(boolean all) {
this.all = all;
}
}
src/main/java/com/alibaba/druid/sql/dialect/oracle/parser/OracleStatementParser.java
+41 -8
View File
@@ -210,11 +210,18 @@ public class OracleStatementParser extends SQLStatementParser {
}
if (lexer.token() == Token.IDENTIFIER) {
if (lexer.stringVal().equals("RAISE")) {
String strVal = lexer.stringVal();
if (strVal.equalsIgnoreCase("RAISE")) {
statementList.add(this.parseRaise());
continue;
}
if (strVal.equalsIgnoreCase("FORALL")) {
SQLStatement forAll = this.parseFor();
statementList.add(forAll);
continue;
}
SQLExpr expr = exprParser.expr();
OracleExprStatement stmt = new OracleExprStatement(expr);
statementList.add(stmt);
@@ -547,18 +554,30 @@ public class OracleStatementParser extends SQLStatementParser {
}
public OracleForStatement parseFor() {
accept(Token.FOR);
OracleForStatement stmt = new OracleForStatement();
if (lexer.token() == Token.FOR) {
lexer.nextToken();
} else {
acceptIdentifier("FORALL");
stmt.setAll(true);
}
stmt.setIndex(this.exprParser.name());
accept(Token.IN);
stmt.setRange(this.exprParser.expr());
accept(Token.LOOP);
this.parseStatementList(stmt.getStatements());
accept(Token.END);
accept(Token.LOOP);
if (stmt.isAll()) {
SQLStatement itemStmt = this.parseStatement();
itemStmt.setParent(stmt);
stmt.getStatements().add(itemStmt);
} else {
accept(Token.LOOP);
this.parseStatementList(stmt.getStatements());
accept(Token.END);
accept(Token.LOOP);
}
return stmt;
}
@@ -1122,8 +1141,22 @@ public class OracleStatementParser extends SQLStatementParser {
acceptIdentifier("TYPE");
}
String typeName = name.toString() + "%TYPE";
String typeName = "TABLE OF " + name.toString() + "%TYPE";
dataType = new SQLDataTypeImpl(typeName);
} else if (identifierEquals("VARRAY")) {
lexer.nextToken();
accept(Token.LPAREN);
int len = this.exprParser.acceptInteger();
accept(Token.RPAREN);
accept(Token.OF);
if (identifierEquals("NUMBER")) {
lexer.nextToken();
String typeName = "VARRAY(" + len + ") OF NUMBER";
dataType = new SQLDataTypeImpl(typeName);
} else {
throw new ParserException("TODO : " + lexer.info());
}
} else {
throw new ParserException("TODO : " + lexer.info());
}
src/main/java/com/alibaba/druid/sql/dialect/oracle/visitor/OracleOutputVisitor.java
+31 -11
View File
@@ -112,12 +112,21 @@ public class OracleOutputVisitor extends SQLASTOutputVisitor implements OracleAS
if (x instanceof OraclePLSQLCommitStatement) {
return;
}
if (x.getParent() instanceof SQLCreateProcedureStatement) {
SQLObject parent = x.getParent();
if (parent instanceof SQLCreateProcedureStatement) {
return;
}
if (isPrettyFormat()) {
if (x.getParent() != null) {
if (parent instanceof OracleForStatement) {
OracleForStatement forStatement = (OracleForStatement) parent;
if (forStatement.isAll()) {
return;
}
}
if (parent != null) {
print(';');
} else {
println(";");
@@ -1857,28 +1866,39 @@ public class OracleOutputVisitor extends SQLASTOutputVisitor implements OracleAS
@Override
public boolean visit(OracleForStatement x) {
print0(ucase ? "FOR " : "for ");
boolean all = x.isAll();
if (all) {
print0(ucase ? "FORALL " : "forall ");
} else {
print0(ucase ? "FOR " : "for ");
}
x.getIndex().accept(this);
print0(ucase ? " IN " : " in ");
x.getRange().accept(this);
println();
print0(ucase ? "LOOP" : "loop");
if (!all) {
println();
print0(ucase ? "LOOP" : "loop");
}
incrementIndent();
println();
for (int i = 0, size = x.getStatements().size(); i < size; ++i) {
SQLStatement stmt = x.getStatements().get(i);
stmt.accept(this);
if (i != size - 1) {
println(";");
} else {
print(';');
if (!all) {
if (i != size - 1) {
println(";");
} else {
print(';');
}
}
}
decrementIndent();
println();
print0(ucase ? "END LOOP" : "end loop");
if (!all) {
println();
print0(ucase ? "END LOOP" : "end loop");
}
return false;
}
src/main/java/com/alibaba/druid/sql/parser/SQLParser.java
+11
View File
@@ -270,6 +270,17 @@ public class SQLParser {
}
}
public int acceptInteger() {
if (lexer.token() == Token.LITERAL_INT) {
int intVal = ((Integer) lexer.integerValue()).intValue();
lexer.nextToken();
return intVal;
} else {
throw new ParserException("syntax error, expect int, actual " + lexer.token() + " "
+ lexer.stringVal());
}
}
public void match(Token token) {
if (lexer.token() != token) {
throw new ParserException("syntax error, expect " + token + ", actual " + lexer.token() + " "
src/main/java/com/alibaba/druid/sql/visitor/SQLASTOutputVisitor.java
+8 -2
View File
@@ -3353,8 +3353,10 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Paramet
if (JdbcConstants.ORACLE.equals(dbType)) {
String dataTypeName = dataType.getName();
if ((dataTypeName.endsWith("%TYPE") && x.getDefaultValue() == null)
|| dataTypeName.equalsIgnoreCase("REF CURSOR")) {
boolean printType = (dataTypeName.startsWith("TABLE OF") && x.getDefaultValue() == null)
|| dataTypeName.equalsIgnoreCase("REF CURSOR")
|| dataTypeName.startsWith("VARRAY(");
if (printType) {
print0(ucase ? "TYPE " : "type ");
}
@@ -3372,6 +3374,10 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Paramet
if (x.isNoCopy()) {
print0(ucase ? "NOCOPY " : "nocopy ");
}
if (printType) {
print0(ucase ? "IS " : "is ");
}
} else {
if (x.getParamType() == SQLParameter.ParameterType.IN) {
print0(ucase ? "IN " : "in ");
src/main/java/com/alibaba/druid/support/http/StatViewFilter.java
+24
View File
@@ -0,0 +1,24 @@
package com.alibaba.druid.support.http;
import javax.servlet.*;
import java.io.IOException;
/**
* Created by wenshao on 12/05/2017.
*/
public class StatViewFilter implements Filter {
@Override
public void init(FilterConfig filterConfig) throws ServletException {
}
@Override
public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
}
@Override
public void destroy() {
}
}