From dc2d7bd2af180a33d2dd6c6ac7c9cae86b4af13f Mon Sep 17 00:00:00 2001 From: wenshao Date: Sat, 8 Oct 2016 17:59:31 +0800 Subject: [PATCH] improved SchemaStatVisitor. --- .../mysql/visitor/MySqlSchemaStatVisitor.java | 31 +++++++++- .../druid/sql/visitor/SchemaStatVisitor.java | 4 ++ .../com/alibaba/druid/stat/TableStat.java | 3 +- .../druid/bvt/sql/mysql/MySqlReplaceTest.java | 61 +++++++++++++++++++ 4 files changed, 97 insertions(+), 2 deletions(-) create mode 100644 src/test/java/com/alibaba/druid/bvt/sql/mysql/MySqlReplaceTest.java diff --git a/src/main/java/com/alibaba/druid/sql/dialect/mysql/visitor/MySqlSchemaStatVisitor.java b/src/main/java/com/alibaba/druid/sql/dialect/mysql/visitor/MySqlSchemaStatVisitor.java index e81fc42f91..d90e2d19a3 100644 --- a/src/main/java/com/alibaba/druid/sql/dialect/mysql/visitor/MySqlSchemaStatVisitor.java +++ b/src/main/java/com/alibaba/druid/sql/dialect/mysql/visitor/MySqlSchemaStatVisitor.java @@ -365,8 +365,37 @@ public class MySqlSchemaStatVisitor extends SchemaStatVisitor implements MySqlAS @Override public boolean visit(MySqlReplaceStatement x) { + setMode(x, Mode.Replace); - return true; + setAliasMap(); + + SQLName tableName = x.getTableName(); + + String ident = null; + if (tableName instanceof SQLIdentifierExpr) { + ident = ((SQLIdentifierExpr) tableName).getName(); + } else if (tableName instanceof SQLPropertyExpr) { + SQLPropertyExpr propertyExpr = (SQLPropertyExpr) tableName; + if (propertyExpr.getOwner() instanceof SQLIdentifierExpr) { + ident = propertyExpr.toString(); + } + } + + if (ident != null) { + setCurrentTable(x, ident); + + TableStat stat = getTableStat(ident); + stat.incrementInsertCount(); + + Map aliasMap = getAliasMap(); + putAliasMap(aliasMap, ident, ident); + } + + accept(x.getColumns()); + accept(x.getValuesList()); + accept(x.getQuery()); + + return false; } @Override diff --git a/src/main/java/com/alibaba/druid/sql/visitor/SchemaStatVisitor.java b/src/main/java/com/alibaba/druid/sql/visitor/SchemaStatVisitor.java index 3fc7ba4d3f..066eb441b9 100644 --- a/src/main/java/com/alibaba/druid/sql/visitor/SchemaStatVisitor.java +++ b/src/main/java/com/alibaba/druid/sql/visitor/SchemaStatVisitor.java @@ -929,6 +929,10 @@ public class SchemaStatVisitor extends SQLASTVisitorAdapter { return tableStats.containsKey(new TableStat.Name(tableName)); } + public boolean containsColumn(String tableName, String columnName) { + return columns.containsKey(new Column(tableName, columnName)); + } + public Collection getColumns() { return columns.values(); } diff --git a/src/main/java/com/alibaba/druid/stat/TableStat.java b/src/main/java/com/alibaba/druid/stat/TableStat.java index 210c71f328..9ed81a614f 100644 --- a/src/main/java/com/alibaba/druid/stat/TableStat.java +++ b/src/main/java/com/alibaba/druid/stat/TableStat.java @@ -572,7 +572,8 @@ public class TableStat { Alter(64), // Drop(128), // DropIndex(256), // - CreateIndex(512)// + CreateIndex(512), // + Replace(1024), ; // public final int mark; diff --git a/src/test/java/com/alibaba/druid/bvt/sql/mysql/MySqlReplaceTest.java b/src/test/java/com/alibaba/druid/bvt/sql/mysql/MySqlReplaceTest.java new file mode 100644 index 0000000000..ce3799d96b --- /dev/null +++ b/src/test/java/com/alibaba/druid/bvt/sql/mysql/MySqlReplaceTest.java @@ -0,0 +1,61 @@ +/* + * Copyright 1999-2101 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; + +import com.alibaba.druid.sql.MysqlTest; +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.dialect.mysql.visitor.MySqlSchemaStatVisitor; +import com.alibaba.druid.stat.TableStat; +import com.alibaba.druid.stat.TableStat.Column; +import org.junit.Assert; + +import java.util.List; + +public class MySqlReplaceTest extends MysqlTest { + + public void test_0() throws Exception { + String sql = "replace into t1 (id,name) values('1','aa'),('2','bb')"; + + MySqlStatementParser parser = new MySqlStatementParser(sql); + List statementList = parser.parseStatementList(); + SQLStatement stmt = statementList.get(0); + assertEquals("REPLACE INTO t1 (id, name)\n" + + "VALUES ('1', 'aa'), ('2', 'bb')", SQLUtils.toMySqlString(stmt)); + assertEquals("replace into t1 (id, name)\n" + + "values ('1', 'aa'), ('2', 'bb')", SQLUtils.toMySqlString(stmt, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION)); + + assertEquals(1, statementList.size()); + + MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor(); + stmt.accept(visitor); + +// System.out.println("Tables : " + visitor.getTables()); +// System.out.println("fields : " + visitor.getColumns()); +// System.out.println("coditions : " + visitor.getConditions()); +// System.out.println("orderBy : " + visitor.getOrderByColumns()); + + assertEquals(1, visitor.getTables().size()); + assertEquals(2, visitor.getColumns().size()); + assertEquals(0, visitor.getConditions().size()); + + assertTrue(visitor.containsTable("t1")); + + assertTrue(visitor.containsColumn("t1", "id")); + assertTrue(visitor.containsColumn("t1", "name")); + } +}