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

mysql parser, updateStatement support join tableSource

Browse Source
This commit is contained in:
wenshao
2011-11-17 09:43:14 +08:00
parent f353cb1794
commit f2efb27d86
5 changed files with 24 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/sql/ast/statement/SQLExprTableSource.java
+2
View File
@@ -27,6 +27,8 @@ public class SQLExprTableSource extends SQLTableSource {
public SQLExprTableSource(){
}
public SQLExpr getExpr() {
return this.expr;
src/main/java/com/alibaba/druid/sql/ast/statement/SQLUpdateStatement.java
+16 -8
View File
@@ -27,21 +27,29 @@ public class SQLUpdateStatement extends SQLStatementImpl {
private static final long serialVersionUID = 1L;
private SQLName tableName;
private final List<SQLUpdateSetItem> items = new ArrayList<SQLUpdateSetItem>();
private SQLExpr where;
private SQLTableSource tableSource;
public SQLUpdateStatement(){
}
public SQLName getTableName() {
return tableName;
public SQLTableSource getTableSource() {
return tableSource;
}
public void setTableName(SQLName tableName) {
this.tableName = tableName;
public void setTableSource(SQLTableSource tableSource) {
this.tableSource = tableSource;
}
public SQLName getTableName() {
if (tableSource instanceof SQLExprTableSource) {
SQLExprTableSource exprTableSource = (SQLExprTableSource) tableSource;
return (SQLName) exprTableSource.getExpr();
}
return null;
}
public SQLExpr getWhere() {
@@ -60,7 +68,7 @@ public class SQLUpdateStatement extends SQLStatementImpl {
public void output(StringBuffer buf) {
buf.append("UPDATE ");
this.tableName.output(buf);
this.tableSource.output(buf);
buf.append(" SET ");
for (int i = 0, size = items.size(); i < size; ++i) {
@@ -79,7 +87,7 @@ public class SQLUpdateStatement extends SQLStatementImpl {
@Override
protected void accept0(SQLASTVisitor visitor) {
if (visitor.visit(this)) {
acceptChild(visitor, tableName);
acceptChild(visitor, tableSource);
acceptChild(visitor, items);
acceptChild(visitor, where);
}