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 sql parser schema repository.

Browse Source
This commit is contained in:
wenshao
2017-08-07 16:47:49 +08:00
parent ea8017382f
commit 9f50882488
5 changed files with 69 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/sql/ast/statement/SQLCreateTableStatement.java
+17 -2
View File
@@ -257,6 +257,10 @@ public class SQLCreateTableStatement extends SQLStatementImpl implements SQLDDLS
if (element instanceof MySqlUnique) {
MySqlUnique unique = (MySqlUnique) element;
if (unique.getColumns().size() == 0) {
continue;
}
SQLExpr column = unique.getColumns().get(0).getExpr();
if (column instanceof SQLIdentifierExpr
&& SQLUtils.nameEquals(columnName, ((SQLIdentifierExpr) column).getName())) {
@@ -738,9 +742,20 @@ public class SQLCreateTableStatement extends SQLStatementImpl implements SQLDDLS
}
}
for (SQLTableElement e : tableElementList) {
for (int i = tableElementList.size() - 1; i >= 0; --i) {
SQLTableElement e = tableElementList.get(i);
if (e instanceof SQLUnique) {
((SQLUnique) e).applyDropColumn(column);
SQLUnique unique = (SQLUnique) e;
unique.applyDropColumn(column);
if (unique.getColumns().size() == 0) {
tableElementList.remove(i);
}
} else if (e instanceof MySqlTableIndex) {
MySqlTableIndex index = (MySqlTableIndex) e;
index.applyDropColumn(column);
if (index.getColumns().size() == 0) {
tableElementList.remove(i);
}
}
}
}
src/main/java/com/alibaba/druid/sql/dialect/mysql/ast/statement/MySqlCreateTableStatement.java
+6 -4
View File
@@ -314,11 +314,13 @@ public class MySqlCreateTableStatement extends SQLCreateTableStatement implement
}
}
for (SQLTableElement e : tableElementList) {
for (int i = 0; i < tableElementList.size(); i++) {
SQLTableElement e = tableElementList.get(i);
if(e instanceof MySqlTableIndex) {
((MySqlTableIndex) e).apply(item);
} else if (e instanceof MySqlKey) {
((MySqlKey) e).applyColumnRename(columnName, column.getName());
((MySqlTableIndex) e).applyColumnRename(columnName, column.getName());
} else if (e instanceof SQLUnique) {
SQLUnique unique = (SQLUnique) e;
unique.applyColumnRename(columnName, column.getName());
}
}
src/main/java/com/alibaba/druid/sql/dialect/mysql/ast/statement/MySqlTableIndex.java
+22 -1
View File
@@ -18,6 +18,7 @@ package com.alibaba.druid.sql.dialect.mysql.ast.statement;
import java.util.ArrayList;
import java.util.List;
import com.alibaba.druid.sql.SQLUtils;
import com.alibaba.druid.sql.ast.SQLExpr;
import com.alibaba.druid.sql.ast.SQLName;
import com.alibaba.druid.sql.ast.statement.SQLColumnDefinition;
@@ -85,7 +86,27 @@ public class MySqlTableIndex extends MySqlObjectImpl implements SQLTableElement
return x;
}
public boolean apply(MySqlAlterTableChangeColumn x) {
public boolean applyColumnRename(SQLName columnName, SQLName to) {
for (SQLSelectOrderByItem orderByItem : columns) {
SQLExpr expr = orderByItem.getExpr();
if (expr instanceof SQLName
&& SQLUtils.nameEquals((SQLName) expr, columnName)) {
orderByItem.setExpr(to.clone());
return true;
}
}
return false;
}
public boolean applyDropColumn(SQLName columnName) {
for (int i = columns.size() - 1; i >= 0; i--) {
SQLExpr expr = columns.get(i).getExpr();
if (expr instanceof SQLName
&& SQLUtils.nameEquals((SQLName) expr, columnName)) {
columns.remove(i);
return true;
}
}
return false;
}
}