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

sql parser parameterized insert_stmt_null output changed.

Browse Source
This commit is contained in:
wenshao
2016-12-10 20:16:44 +08:00
parent e5b92293b2
commit 2659474225
3 changed files with 109 additions and 82 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
-59
View File
@@ -194,65 +194,6 @@ public class MySqlOutputVisitor extends SQLASTOutputVisitor implements MySqlASTV
this.shardingSupport = shardingSupport;
}
public boolean visit(SQLIdentifierExpr x) {
if (!parameterized) {
return super.visit(x);
}
final String name = x.getName();
if (shardingSupport) {
SQLObject parent = x.getParent();
shardingSupport = parent instanceof SQLExprTableSource || parent instanceof SQLPropertyExpr;
}
if (shardingSupport) {
int pos = name.lastIndexOf('_');
if (pos != -1 && pos != name.length() - 1) {
boolean quote = name.charAt(0) == '`' && name.charAt(name.length() - 1) == '`';
boolean isNumber = true;
int end = name.length();
if (quote) {
end--;
}
for (int i = pos + 1; i < end; ++i) {
char ch = name.charAt(i);
if (ch < '0' || ch > '9') {
isNumber = false;
break;
}
}
if (isNumber) {
int start = quote ? 1 : 0;
String realName = name.substring(start, pos);
print0(realName);
incrementReplaceCunt();
return false;
}
}
int numberCount = 0;
for (int i = name.length() - 1; i >= 0; --i) {
char ch = name.charAt(i);
if (ch < '0' || ch > '9') {
break;
} else {
numberCount++;
}
}
if (numberCount > 1) {
int numPos = name.length() - numberCount;
String realName = name.substring(0, numPos);
print0(realName);
incrementReplaceCunt();
return false;
}
}
print0(name);
return false;
}
@Override
public boolean visit(SQLSelectQueryBlock select) {
if (select instanceof MySqlSelectQueryBlock) {
src/main/java/com/alibaba/druid/sql/visitor/SQLASTOutputVisitor.java
+77 -23
View File
@@ -522,7 +522,67 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Paramet
}
public boolean visit(SQLIdentifierExpr x) {
print0(x.getName());
final String name = x.getName();
if (!parameterized) {
print0(x.getName());
return false;
}
return printName(x, name);
}
private boolean printName(SQLName x, String name) {
boolean shardingSupport = this.shardingSupport;
if (shardingSupport) {
SQLObject parent = x.getParent();
shardingSupport = parent instanceof SQLExprTableSource || parent instanceof SQLPropertyExpr;
}
if (shardingSupport) {
int pos = name.lastIndexOf('_');
if (pos != -1 && pos != name.length() - 1) {
boolean quote = name.charAt(0) == '`' && name.charAt(name.length() - 1) == '`';
boolean isNumber = true;
int end = name.length();
if (quote) {
end--;
}
for (int i = pos + 1; i < end; ++i) {
char ch = name.charAt(i);
if (ch < '0' || ch > '9') {
isNumber = false;
break;
}
}
if (isNumber) {
int start = quote ? 1 : 0;
String realName = name.substring(start, pos);
print0(realName);
incrementReplaceCunt();
return false;
}
}
int numberCount = 0;
for (int i = name.length() - 1; i >= 0; --i) {
char ch = name.charAt(i);
if (ch < '0' || ch > '9') {
break;
} else {
numberCount++;
}
}
if (numberCount > 1) {
int numPos = name.length() - numberCount;
String realName = name.substring(0, numPos);
print0(realName);
incrementReplaceCunt();
return false;
}
}
print0(name);
return false;
}
@@ -735,26 +795,15 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Paramet
}
public boolean visit(SQLNullExpr x) {
// if (this.parameterized
// && ParameterizedOutputVisitorUtils.checkParameterize(x)) {
// SQLObject parent = x.getParent();
// if (parent instanceof SQLBinaryOpExpr) {
// SQLBinaryOpExpr binaryOpExpr = (SQLBinaryOpExpr) parent;
// if (binaryOpExpr.getOperator() == SQLBinaryOperator.IsNot
// || binaryOpExpr.getOperator() == SQLBinaryOperator.Is) {
// print("NULL");
// return false;
// }
// }
//
// print('?');
// incrementReplaceCunt();
//
// if(this instanceof ExportParameterVisitor || this.parameters != null){
// ExportParameterVisitorUtils.exportParameter(this.parameters, x);
// }
// return false;
// }
if (this.parameterized
&& ParameterizedOutputVisitorUtils.checkParameterize(x)) {
SQLObject parent = x.getParent();
if (parent instanceof ValuesClause) {
print('?');
incrementReplaceCunt();
return false;
}
}
print0(ucase ? "NULL" : "null");
return false;
@@ -776,9 +825,14 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Paramet
}
public boolean visit(SQLPropertyExpr x) {
x.getOwner().accept(this);
SQLExpr owner = x.getOwner();
owner.accept(this);
print('.');
print0(x.getName());
String name = x.getName();
printName(x, name);
return false;
}