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 insert into direct support multi-values

Browse Source
This commit is contained in:
高铁
2016-05-15 00:32:42 +08:00
parent d93411e8a7
commit 186cfb6a80
7 changed files with 106 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/sql/ast/statement/SQLInsertInto.java
+19 -3
View File
@@ -28,8 +28,9 @@ public abstract class SQLInsertInto extends SQLObjectImpl {
protected SQLExprTableSource tableSource;
protected final List<SQLExpr> columns = new ArrayList<SQLExpr>();
protected ValuesClause values;
protected SQLSelect query;
protected List<ValuesClause> valuesList = new ArrayList<ValuesClause>();
public SQLInsertInto(){
@@ -86,10 +87,25 @@ public abstract class SQLInsertInto extends SQLObjectImpl {
}
public ValuesClause getValues() {
return values;
if (valuesList.size() == 0) {
return null;
}
return valuesList.get(0);
}
public void setValues(ValuesClause values) {
this.values = values;
if (valuesList.size() == 0) {
valuesList.add(values);
} else {
valuesList.set(0, values);
}
}
public List<ValuesClause> getValuesList() {
return valuesList;
}
public void setValuesList(List<ValuesClause> valuesList) {
this.valuesList = valuesList;
}
}
src/main/java/com/alibaba/druid/sql/ast/statement/SQLInsertStatement.java
+1 -1
View File
@@ -35,7 +35,7 @@ public class SQLInsertStatement extends SQLInsertInto implements SQLStatement {
if (visitor.visit(this)) {
this.acceptChild(visitor, tableSource);
this.acceptChild(visitor, columns);
this.acceptChild(visitor, values);
this.acceptChild(visitor, valuesList);
this.acceptChild(visitor, query);
}
src/main/java/com/alibaba/druid/sql/dialect/mysql/ast/statement/MySqlInsertStatement.java
+1 -26
View File
@@ -31,38 +31,13 @@ public class MySqlInsertStatement extends SQLInsertStatement {
private boolean highPriority = false;
private boolean ignore = false;
private List<ValuesClause> valuesList = new ArrayList<ValuesClause>();
private final List<SQLExpr> duplicateKeyUpdate = new ArrayList<SQLExpr>();
public List<SQLExpr> getDuplicateKeyUpdate() {
return duplicateKeyUpdate;
}
public ValuesClause getValues() {
if (valuesList.size() == 0) {
return null;
}
return valuesList.get(0);
}
public void setValues(ValuesClause values) {
if (valuesList.size() == 0) {
valuesList.add(values);
} else {
valuesList.set(0, values);
}
}
public List<ValuesClause> getValuesList() {
return valuesList;
}
public void setValuesList(List<ValuesClause> valuesList) {
this.valuesList = valuesList;
}
public boolean isLowPriority() {
public boolean isLowPriority() {
return lowPriority;
}
src/main/java/com/alibaba/druid/sql/dialect/oracle/ast/stmt/OracleMultiInsertStatement.java
+1 -1
View File
@@ -190,7 +190,7 @@ public class OracleMultiInsertStatement extends OracleStatementImpl {
if (visitor.visit(this)) {
this.acceptChild(visitor, tableSource);
this.acceptChild(visitor, columns);
this.acceptChild(visitor, values);
this.acceptChild(visitor, valuesList);
this.acceptChild(visitor, query);
this.acceptChild(visitor, returning);
this.acceptChild(visitor, errorLogging);
src/main/java/com/alibaba/druid/sql/parser/SQLStatementParser.java
+14 -5
View File
@@ -1474,11 +1474,20 @@ public class SQLStatementParser extends SQLParser {
if (lexer.token() == Token.VALUES) {
lexer.nextToken();
accept(Token.LPAREN);
SQLInsertStatement.ValuesClause values = new SQLInsertStatement.ValuesClause();
this.exprParser.exprList(values.getValues(), values);
insertStatement.setValues(values);
accept(Token.RPAREN);
for (;;) {
accept(Token.LPAREN);
SQLInsertStatement.ValuesClause values = new SQLInsertStatement.ValuesClause();
this.exprParser.exprList(values.getValues(), values);
insertStatement.getValuesList().add(values);
accept(Token.RPAREN);
if (lexer.token() == Token.COMMA) {
lexer.nextToken();
continue;
} else {
break;
}
}
} else if (acceptSubQuery && (lexer.token() == Token.SELECT || lexer.token() == Token.LPAREN)) {
SQLQueryExpr queryExpr = (SQLQueryExpr) this.exprParser.expr();
insertStatement.setQuery(queryExpr.getSubQuery());
src/main/java/com/alibaba/druid/sql/visitor/SQLASTOutputVisitor.java
+2 -2
View File
@@ -1067,11 +1067,11 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Printab
decrementIndent();
}
if (x.getValues() != null) {
if (!x.getValuesList().isEmpty()) {
println();
print0(ucase ? "VALUES" : "values");
println();
x.getValues().accept(this);
printAndAccept(x.getValuesList(), ", ");
} else {
if (x.getQuery() != null) {
println();