SteNicholas commented on a change in pull request #4123: Add MySQL DDLStatement
antlr visitor
URL:
https://github.com/apache/incubator-shardingsphere/pull/4123#discussion_r373771791
##########
File path:
shardingsphere-sql-parser/shardingsphere-sql-parser-mysql/src/main/java/org/apache/shardingsphere/sql/parser/MySQLVisitor.java
##########
@@ -190,6 +227,135 @@ public ASTNode visitCreateUser(final CreateUserContext
ctx) {
}
// DDLStatement.g4
+ @Override
+ public ASTNode visitCreateTable(final CreateTableContext ctx) {
+ CreateTableStatement result = new CreateTableStatement();
+ TableSegment table = (TableSegment) visit(ctx.tableName());
+ result.setTable(table);
+ result.getAllSQLSegments().add(table);
+ CreateDefinitionClause_Context createDefinitionClause =
ctx.createDefinitionClause_();
+ if (null != createDefinitionClause) {
+ for (CreateDefinition_Context createDefinition :
createDefinitionClause.createDefinitions_().createDefinition_()) {
+ ColumnDefinitionContext columnDefinition =
createDefinition.columnDefinition();
+ if (null != columnDefinition) {
+ ColumnDefinitionSegment columnDefinitionSegment =
createColumnDefinitionSegment(columnDefinition, result);
+ result.getColumnDefinitions().add(columnDefinitionSegment);
+ result.getAllSQLSegments().add(columnDefinitionSegment);
+ }
+ ConstraintDefinition_Context constraintDefinition =
createDefinition.constraintDefinition_();
+ ForeignKeyOption_Context foreignKeyOption = null ==
constraintDefinition ? null : constraintDefinition.foreignKeyOption_();
+ if (null != foreignKeyOption) {
+ result.getAllSQLSegments().add((TableSegment)
visit(foreignKeyOption.referenceDefinition_().tableName()));
+ }
+ }
+ }
+ CreateLikeClause_Context createLikeClause = ctx.createLikeClause_();
+ if (null != createLikeClause) {
+ result.getAllSQLSegments().add((TableSegment)
visit(createLikeClause));
+ }
+ return result;
+ }
+
+ @Override
+ public ASTNode visitAlterTable(final AlterTableContext ctx) {
+ AlterTableStatement result = new AlterTableStatement();
+ TableSegment table = (TableSegment) visit(ctx.tableName());
+ result.setTable(table);
+ result.getAllSQLSegments().add(table);
+ if (null != ctx.alterDefinitionClause_()) {
+ for (AlterSpecification_Context alterSpecification :
ctx.alterDefinitionClause_().alterSpecification_()) {
+ AddColumnSpecificationContext addColumnSpecification =
alterSpecification.addColumnSpecification();
+ if (null != addColumnSpecification) {
+ List<ColumnDefinitionContext> columnDefinitions =
addColumnSpecification.columnDefinition();
+ ColumnDefinitionSegment columnDefinitionSegment = null;
+ for (ColumnDefinitionContext columnDefinition :
columnDefinitions) {
+ columnDefinitionSegment =
createColumnDefinitionSegment(columnDefinition, result);
+
result.getAddedColumnDefinitions().add(columnDefinitionSegment);
+
result.getAllSQLSegments().add(columnDefinitionSegment);
+ }
+
createColumnPositionSegment(addColumnSpecification.firstOrAfterColumn(),
columnDefinitionSegment, result);
+ }
+ AddConstraintSpecificationContext addConstraintSpecification =
alterSpecification.addConstraintSpecification();
+ ForeignKeyOption_Context foreignKeyOption = null ==
addConstraintSpecification
+ ? null :
addConstraintSpecification.constraintDefinition_().foreignKeyOption_();
+ if (null != foreignKeyOption) {
+ result.getAllSQLSegments().add((TableSegment)
visit(foreignKeyOption.referenceDefinition_().tableName()));
+ }
+ ChangeColumnSpecificationContext changeColumnSpecification =
alterSpecification.changeColumnSpecification();
+ if (null != changeColumnSpecification) {
+
createColumnPositionSegment(changeColumnSpecification.firstOrAfterColumn(),
+
createColumnDefinitionSegment(changeColumnSpecification.columnDefinition(),
result), result);
+ }
+ DropColumnSpecificationContext dropColumnSpecification =
alterSpecification.dropColumnSpecification();
+ if (null != dropColumnSpecification) {
+ result.getDroppedColumnNames().add(((ColumnSegment)
visit(dropColumnSpecification)).getName());
+ }
+ ModifyColumnSpecificationContext modifyColumnSpecification =
alterSpecification.modifyColumnSpecification();
+ if (null != modifyColumnSpecification) {
+
createColumnPositionSegment(modifyColumnSpecification.firstOrAfterColumn(),
+
createColumnDefinitionSegment(modifyColumnSpecification.columnDefinition(),
result), result);
+ }
+ }
+ }
+ return result;
+ }
+
+ @Override
+ public ASTNode visitDropTable(final DropTableContext ctx) {
+ DropTableStatement result = new DropTableStatement();
+ ListValue<TableSegment> tables = (ListValue<TableSegment>)
visit(ctx.tableNames());
+ result.getTables().addAll(tables.getValues());
+ result.getAllSQLSegments().addAll(tables.getValues());
+ return result;
+ }
+
+ @Override
+ public ASTNode visitTruncateTable(final TruncateTableContext ctx) {
+ DDLStatement result = new DDLStatement();
+ TableSegment table = (TableSegment) visit(ctx.tableName());
+ result.getAllSQLSegments().add(table);
+ return result;
+ }
+
+ @Override
+ public ASTNode visitCreateIndex(final CreateIndexContext ctx) {
+ CreateIndexStatement result = new CreateIndexStatement();
+ TableSegment table = (TableSegment) visit(ctx.tableName());
+ result.setTable(table);
+ result.getAllSQLSegments().add(table);
+ IndexSegment index = (IndexSegment) visit(ctx.indexName());
+ result.setIndex(index);
+ result.getAllSQLSegments().add(index);
+ return result;
+ }
+
+ @Override
+ public ASTNode visitDropIndex(final DropIndexContext ctx) {
+ DropIndexStatement result = new DropIndexStatement();
+ TableSegment table = (TableSegment) visit(ctx.tableName());
+ result.setTable(table);
+ result.getAllSQLSegments().add(table);
+ IndexSegment index = (IndexSegment) visit(ctx.indexName());
Review comment:
@tristaZero Okay, I would like to remove `IndexSegment`.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
