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.

Browse Source
This commit is contained in:
wenshao
2017-08-01 17:37:56 +08:00
parent 512b8dbabd
commit c2adc8bfd4
3 changed files with 38 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/sql/repository/Schema.java
+26
View File
@@ -182,6 +182,19 @@ public class Schema {
return false;
}
public boolean visit(SQLAlterTableStatement x) {
SQLName table = x.getName();
SchemaObject object = repository.findTable(table);
if (object != null) {
SQLCreateTableStatement stmt = (SQLCreateTableStatement) object.getStatement();
if (stmt != null) {
stmt.apply(x);
}
}
return false;
}
}
private class MySqlSchemaVisitor extends MySqlASTVisitorAdapter {
@@ -248,6 +261,19 @@ public class Schema {
return false;
}
public boolean visit(SQLAlterTableStatement x) {
SQLName table = x.getName();
SchemaObject object = repository.findTable(table);
if (object != null) {
SQLCreateTableStatement stmt = (SQLCreateTableStatement) object.getStatement();
if (stmt != null) {
stmt.apply(x);
}
}
return false;
}
}
public SchemaObject findTable(SQLTableSource tableSource, String alias) {
src/main/java/com/alibaba/druid/sql/repository/SchemaRepository.java
+5 -3
View File
@@ -213,15 +213,17 @@ public class SchemaRepository {
if (stmt instanceof MySqlShowColumnsStatement) {
MySqlShowColumnsStatement showColumns = ((MySqlShowColumnsStatement) stmt);
SQLName table = showColumns.getTable();
SchemaObject schemaObject = findSchemaObject(table);
SchemaObject schemaObject = findTable(table);
MySqlCreateTableStatement createTableStmt = (MySqlCreateTableStatement) schemaObject.getStatement();
createTableStmt.showCoumns(buf);
} else if (stmt instanceof MySqlShowCreateTableStatement) {
MySqlShowCreateTableStatement showCreateTableStmt = (MySqlShowCreateTableStatement) stmt;
SQLName table = showCreateTableStmt.getName();
SchemaObject schemaObject = findSchemaObject(table);
SchemaObject schemaObject = findTable(table);
MySqlCreateTableStatement createTableStmt = (MySqlCreateTableStatement) schemaObject.getStatement();
createTableStmt.output(buf);
} else {
this.getDefaultSchema().accept(stmt);
}
}
@@ -231,7 +233,7 @@ public class SchemaRepository {
}
}
private SchemaObject findSchemaObject(SQLName name) {
public SchemaObject findTable(SQLName name) {
if (name instanceof SQLIdentifierExpr) {
return findTable(((SQLIdentifierExpr) name).getName());
}