From c39f787c102a061b46a222429637145fa1721a61 Mon Sep 17 00:00:00 2001 From: wenshao Date: Fri, 12 Jan 2018 19:32:37 +0800 Subject: [PATCH] improved mysql sql parser. support alter table rename index. --- .../statement/SQLAlterTableRenameIndex.java | 65 +++++++++++++++++++ .../mysql/parser/MySqlStatementParser.java | 11 ++++ .../sql/visitor/SQLASTOutputVisitor.java | 8 +++ .../druid/sql/visitor/SQLASTVisitor.java | 3 + .../sql/visitor/SQLASTVisitorAdapter.java | 10 +++ .../MySqlAlterTableTest31_renameIndex.java | 37 +++++++++++ 6 files changed, 134 insertions(+) create mode 100644 src/main/java/com/alibaba/druid/sql/ast/statement/SQLAlterTableRenameIndex.java create mode 100644 src/test/java/com/alibaba/druid/bvt/sql/mysql/alter/MySqlAlterTableTest31_renameIndex.java diff --git a/src/main/java/com/alibaba/druid/sql/ast/statement/SQLAlterTableRenameIndex.java b/src/main/java/com/alibaba/druid/sql/ast/statement/SQLAlterTableRenameIndex.java new file mode 100644 index 0000000000..daff17ca2a --- /dev/null +++ b/src/main/java/com/alibaba/druid/sql/ast/statement/SQLAlterTableRenameIndex.java @@ -0,0 +1,65 @@ +/* + * Copyright 1999-2017 Alibaba Group Holding Ltd. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.alibaba.druid.sql.ast.statement; + +import com.alibaba.druid.sql.ast.SQLName; +import com.alibaba.druid.sql.ast.SQLObjectImpl; +import com.alibaba.druid.sql.visitor.SQLASTVisitor; + +public class SQLAlterTableRenameIndex extends SQLObjectImpl implements SQLAlterTableItem { + + private SQLName name; + private SQLName to; + + public SQLAlterTableRenameIndex(SQLName name, SQLName to){ + this.setName(name); + this.setTo(to); + } + + public SQLAlterTableRenameIndex(){ + } + + + public void accept0(SQLASTVisitor visitor) { + if (visitor.visit(this)) { + acceptChild(visitor, name); + acceptChild(visitor, to); + } + visitor.endVisit(this); + } + + public SQLName getName() { + return name; + } + + public void setName(SQLName x) { + if (x != null) { + x.setParent(this); + } + this.name = x; + } + + public SQLName getTo() { + return to; + } + + public void setTo(SQLName x) { + if (x != null) { + x.setParent(this); + } + this.to = x; + } +} diff --git a/src/main/java/com/alibaba/druid/sql/dialect/mysql/parser/MySqlStatementParser.java b/src/main/java/com/alibaba/druid/sql/dialect/mysql/parser/MySqlStatementParser.java index 4aa8901a60..622753bd9e 100644 --- a/src/main/java/com/alibaba/druid/sql/dialect/mysql/parser/MySqlStatementParser.java +++ b/src/main/java/com/alibaba/druid/sql/dialect/mysql/parser/MySqlStatementParser.java @@ -3651,6 +3651,17 @@ public class MySqlStatementParser extends SQLStatementParser { } } else if (lexer.identifierEquals("RENAME")) { lexer.nextToken(); + + if (lexer.token() == Token.INDEX) { + lexer.nextToken(); + SQLName name = this.exprParser.name(); + accept(Token.TO); + SQLName to = this.exprParser.name(); + SQLAlterTableRenameIndex item = new SQLAlterTableRenameIndex(name, to); + stmt.addItem(item); + continue; + } + if (lexer.token() == Token.TO || lexer.token() == Token.AS) { lexer.nextToken(); } diff --git a/src/main/java/com/alibaba/druid/sql/visitor/SQLASTOutputVisitor.java b/src/main/java/com/alibaba/druid/sql/visitor/SQLASTOutputVisitor.java index 1e24a65556..5dd13a7327 100644 --- a/src/main/java/com/alibaba/druid/sql/visitor/SQLASTOutputVisitor.java +++ b/src/main/java/com/alibaba/druid/sql/visitor/SQLASTOutputVisitor.java @@ -6130,4 +6130,12 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Paramet print('>'); return false; } + + public boolean visit(SQLAlterTableRenameIndex x) { + print0(ucase ? "RENAME INDEX " : "rename index "); + x.getName().accept(this); + print0(ucase ? " TO " : " to "); + x.getTo().accept(this); + return false; + } } diff --git a/src/main/java/com/alibaba/druid/sql/visitor/SQLASTVisitor.java b/src/main/java/com/alibaba/druid/sql/visitor/SQLASTVisitor.java index a4d53f3fb6..9e624da1be 100644 --- a/src/main/java/com/alibaba/druid/sql/visitor/SQLASTVisitor.java +++ b/src/main/java/com/alibaba/druid/sql/visitor/SQLASTVisitor.java @@ -817,4 +817,7 @@ public interface SQLASTVisitor { boolean visit(SQLDropMaterializedViewStatement x); void endVisit(SQLDropMaterializedViewStatement x); + + boolean visit(SQLAlterTableRenameIndex x); + void endVisit(SQLAlterTableRenameIndex x); } diff --git a/src/main/java/com/alibaba/druid/sql/visitor/SQLASTVisitorAdapter.java b/src/main/java/com/alibaba/druid/sql/visitor/SQLASTVisitorAdapter.java index 5107e3bf24..21fd8eae5f 100644 --- a/src/main/java/com/alibaba/druid/sql/visitor/SQLASTVisitorAdapter.java +++ b/src/main/java/com/alibaba/druid/sql/visitor/SQLASTVisitorAdapter.java @@ -1924,6 +1924,16 @@ public class SQLASTVisitorAdapter implements SQLASTVisitor { } + @Override + public boolean visit(SQLAlterTableRenameIndex x) { + return true; + } + + @Override + public void endVisit(SQLAlterTableRenameIndex x) { + + } + public final boolean isEnabled(VisitorFeature feature) { return VisitorFeature.isEnabled(this.features, feature); } diff --git a/src/test/java/com/alibaba/druid/bvt/sql/mysql/alter/MySqlAlterTableTest31_renameIndex.java b/src/test/java/com/alibaba/druid/bvt/sql/mysql/alter/MySqlAlterTableTest31_renameIndex.java new file mode 100644 index 0000000000..7012b7b61a --- /dev/null +++ b/src/test/java/com/alibaba/druid/bvt/sql/mysql/alter/MySqlAlterTableTest31_renameIndex.java @@ -0,0 +1,37 @@ +/* + * Copyright 1999-2017 Alibaba Group Holding Ltd. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.alibaba.druid.bvt.sql.mysql.alter; + +import com.alibaba.druid.sql.SQLUtils; +import com.alibaba.druid.sql.ast.SQLStatement; +import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser; +import com.alibaba.druid.sql.parser.Token; +import junit.framework.TestCase; +import org.junit.Assert; + +public class MySqlAlterTableTest31_renameIndex extends TestCase { + + public void test_alter_add_key() throws Exception { + String sql = "ALTER TABLE test_table_normal\n" + + "RENAME INDEX old_index_name TO new_index_name"; + MySqlStatementParser parser = new MySqlStatementParser(sql); + SQLStatement stmt = parser.parseStatementList().get(0); + parser.match(Token.EOF); + String output = SQLUtils.toMySqlString(stmt); + Assert.assertEquals("ALTER TABLE test_table_normal\n" + + "\tRENAME INDEX old_index_name TO new_index_name", output); + } +}