You are viewing a plain text version of this content. The canonical link for it is here.
Posted to notifications@shardingsphere.apache.org by GitBox <gi...@apache.org> on 2020/02/01 10:13:26 UTC
[GitHub] [incubator-shardingsphere] SteNicholas commented on a change in
pull request #4123: Add MySQL DDLStatement antlr visitor
SteNicholas commented on a change in pull request #4123: Add MySQL DDLStatement antlr visitor
URL: https://github.com/apache/incubator-shardingsphere/pull/4123#discussion_r373771777
##########
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());
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:
users@infra.apache.org
With regards,
Apache Git Services