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 parser performance.

Browse Source
This commit is contained in:
wenshao
2017-08-21 15:34:10 +08:00
parent 17772cde69
commit 5804eea0c9
20 changed files with 484 additions and 151 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/sql/dialect/mysql/visitor/MySqlOutputVisitor.java
+73 -43
View File
@@ -53,6 +53,7 @@ import com.alibaba.druid.sql.visitor.ParameterizedOutputVisitorUtils;
import com.alibaba.druid.sql.visitor.SQLASTOutputVisitor;
import com.alibaba.druid.util.JdbcConstants;
import java.io.IOException;
import java.security.AccessControlException;
import java.util.List;
import java.util.Map;
@@ -567,44 +568,59 @@ public class MySqlOutputVisitor extends SQLASTOutputVisitor implements MySqlASTV
}
public boolean visit(SQLCharExpr x) {
if (this.parameterized
&& ParameterizedOutputVisitorUtils.checkParameterize(x)) {
print('?');
incrementReplaceCunt();
if (this instanceof ExportParameterVisitor || this.parameters != null) {
ExportParameterVisitorUtils.exportParameter(this.parameters, x);
}
if (this.appender == null) {
return false;
}
print('\'');
String text = x.getText();
StringBuilder buf = new StringBuilder(text.length());
for (int i = 0; i < text.length(); ++i) {
char ch = text.charAt(i);
if (ch == '\'') {
buf.append('\'');
buf.append('\'');
} else if (ch == '\\') {
buf.append('\\');
buf.append('\\');
} else if (ch == '\0') {
buf.append('\\');
buf.append('0');
} else {
buf.append(ch);
try {
if (this.parameterized
&& ParameterizedOutputVisitorUtils.checkParameterize(x)) {
this.appender.append('?');
incrementReplaceCunt();
if (this instanceof ExportParameterVisitor || this.parameters != null) {
ExportParameterVisitorUtils.exportParameter(this.parameters, x);
}
return false;
}
}
if (buf.length() != text.length()) {
text = buf.toString();
}
print0(text);
this.appender.append('\'');
print('\'');
String text = x.getText();
boolean hasSpecial = false;
for (int i = 0; i < text.length(); ++i) {
char ch = text.charAt(i);
if (ch == '\'' || ch == '\\' || ch == '\0') {
hasSpecial = true;
break;
}
}
if (hasSpecial) {
for (int i = 0; i < text.length(); ++i) {
char ch = text.charAt(i);
if (ch == '\'') {
appender.append('\'');
appender.append('\'');
} else if (ch == '\\') {
appender.append('\\');
appender.append('\\');
} else if (ch == '\0') {
appender.append('\\');
appender.append('0');
} else {
appender.append(ch);
}
}
} else {
appender.append(text);
}
appender.append('\'');
return false;
} catch (IOException e) {
throw new RuntimeException("println error", e);
}
}
public boolean visit(SQLVariantRefExpr x) {
@@ -924,12 +940,18 @@ public class MySqlOutputVisitor extends SQLASTOutputVisitor implements MySqlASTV
print0(ucase ? "INTO " : "into ");
x.getTableSource().accept(this);
SQLExprTableSource tableSource = x.getTableSource();
if (tableSource.getClass() == SQLExprTableSource.class) {
visit(tableSource);
} else {
tableSource.accept(this);
}
if (x.getColumns().size() > 0) {
List<SQLExpr> columns = x.getColumns();
if (columns.size() > 0) {
incrementIndent();
print0(" (");
for (int i = 0, size = x.getColumns().size(); i < size; ++i) {
for (int i = 0, size = columns.size(); i < size; ++i) {
if (i != 0) {
if (i % 5 == 0) {
println();
@@ -937,15 +959,22 @@ public class MySqlOutputVisitor extends SQLASTOutputVisitor implements MySqlASTV
print0(", ");
}
x.getColumns().get(i).accept(this);
SQLExpr columnn = columns.get(i);
if (columnn instanceof SQLIdentifierExpr) {
visit((SQLIdentifierExpr) columnn);
}
else {
columnn.accept(this);
}
}
print(')');
decrementIndent();
}
if (!x.getValuesList().isEmpty()) {
List<SQLInsertStatement.ValuesClause> valuesList = x.getValuesList();
if (!valuesList.isEmpty()) {
println();
printValuesList(x);
printValuesList(valuesList);
}
if (x.getQuery() != null) {
@@ -970,8 +999,7 @@ public class MySqlOutputVisitor extends SQLASTOutputVisitor implements MySqlASTV
return false;
}
protected void printValuesList(MySqlInsertStatement x) {
List<SQLInsertStatement.ValuesClause> valuesList = x.getValuesList();
protected void printValuesList(List<SQLInsertStatement.ValuesClause> valuesList) {
if (parameterized) {
print0(ucase ? "VALUES " : "values ");
@@ -985,18 +1013,20 @@ public class MySqlOutputVisitor extends SQLASTOutputVisitor implements MySqlASTV
}
print0(ucase ? "VALUES " : "values ");
if (x.getValuesList().size() > 1) {
incrementIndent();
if (valuesList.size() > 1) {
this.indentCount++;
}
for (int i = 0, size = valuesList.size(); i < size; ++i) {
if (i != 0) {
print(',');
println();
}
valuesList.get(i).accept(this);
SQLInsertStatement.ValuesClause item = valuesList.get(i);
visit(item);
}
if (valuesList.size() > 1) {
decrementIndent();
this.indentCount--;
}
}
src/main/java/com/alibaba/druid/sql/dialect/oracle/ast/stmt/OracleAlterStatement.java
+15
View File
@@ -1,3 +1,18 @@
/*
* Copyright 1999-2017 Alibaba Group Holding Ltd.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.alibaba.druid.sql.dialect.oracle.ast.stmt;
import com.alibaba.druid.sql.ast.SQLStatement;
src/main/java/com/alibaba/druid/sql/dialect/oracle/ast/stmt/OracleCreateSynonymStatement.java
+15
View File
@@ -1,3 +1,18 @@
/*
* Copyright 1999-2017 Alibaba Group Holding Ltd.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.alibaba.druid.sql.dialect.oracle.ast.stmt;
import com.alibaba.druid.sql.ast.SQLName;
src/main/java/com/alibaba/druid/sql/dialect/oracle/parser/OracleExprParser.java
+2 -2
View File
@@ -330,7 +330,7 @@ public class OracleExprParser extends SQLExprParser {
lexer.nextToken();
break;
case LITERAL_FLOAT:
sqlExpr = new SQLNumberExpr(lexer.decimalValue());
sqlExpr = lexer.numberExpr();
lexer.nextToken();
break;
case BINARY_FLOAT:
@@ -382,7 +382,7 @@ public class OracleExprParser extends SQLExprParser {
lexer.nextToken();
break;
case LITERAL_FLOAT:
sqlExpr = new SQLNumberExpr(lexer.decimalValue().negate());
sqlExpr = lexer.numberExpr();
lexer.nextToken();
break;
case BINARY_FLOAT: