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

mysql parser support create user

Browse Source
This commit is contained in:
温少
2012-02-22 21:04:05 +08:00
parent 5b6b8b8b76
commit 3a5583c16f
15 changed files with 2635 additions and 2253 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/sql/dialect/mysql/ast/MySqlObject.java
+9
View File
@@ -0,0 +1,9 @@
package com.alibaba.druid.sql.dialect.mysql.ast;
import com.alibaba.druid.sql.ast.SQLObject;
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlASTVisitor;
public interface MySqlObject extends SQLObject {
void accept0(MySqlASTVisitor visitor);
}
src/main/java/com/alibaba/druid/sql/dialect/mysql/ast/MySqlObjectImpl.java
+36
View File
@@ -0,0 +1,36 @@
/*
* Copyright 1999-2011 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.dialect.mysql.ast;
import com.alibaba.druid.sql.ast.SQLObjectImpl;
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlASTVisitor;
import com.alibaba.druid.sql.visitor.SQLASTVisitor;
public abstract class MySqlObjectImpl extends SQLObjectImpl implements MySqlObject {
private static final long serialVersionUID = 1L;
@Override
protected void accept0(SQLASTVisitor visitor) {
if (visitor instanceof MySqlASTVisitor) {
accept0((MySqlASTVisitor) visitor);
} else {
throw new IllegalArgumentException("not support visitor type : " + visitor.getClass().getName());
}
}
public abstract void accept0(MySqlASTVisitor visitor);
}
src/main/java/com/alibaba/druid/sql/dialect/mysql/ast/statement/MySqlCreateUserStatement.java
+73
View File
@@ -0,0 +1,73 @@
package com.alibaba.druid.sql.dialect.mysql.ast.statement;
import java.util.ArrayList;
import java.util.List;
import com.alibaba.druid.sql.ast.SQLExpr;
import com.alibaba.druid.sql.dialect.mysql.ast.MySqlObjectImpl;
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlASTVisitor;
public class MySqlCreateUserStatement extends MySqlStatementImpl {
private static final long serialVersionUID = 1L;
private List<UserSpecification> users = new ArrayList<UserSpecification>(2);
public List<UserSpecification> getUsers() {
return users;
}
public void setUsers(List<UserSpecification> users) {
this.users = users;
}
@Override
public void accept0(MySqlASTVisitor visitor) {
if (visitor.visit(this)) {
acceptChild(visitor, users);
}
}
public static class UserSpecification extends MySqlObjectImpl {
private static final long serialVersionUID = 1L;
private SQLExpr user;
private SQLExpr password;
private SQLExpr authPlugin;
public SQLExpr getUser() {
return user;
}
public void setUser(SQLExpr user) {
this.user = user;
}
public SQLExpr getPassword() {
return password;
}
public void setPassword(SQLExpr password) {
this.password = password;
}
public SQLExpr getAuthPlugin() {
return authPlugin;
}
public void setAuthPlugin(SQLExpr authPlugin) {
this.authPlugin = authPlugin;
}
@Override
public void accept0(MySqlASTVisitor visitor) {
if (visitor.visit(this)) {
acceptChild(visitor, user);
acceptChild(visitor, password);
acceptChild(visitor, authPlugin);
}
}
}
}
src/main/java/com/alibaba/druid/sql/dialect/mysql/parser/MySqlLexer.java
+3
View File
@@ -104,11 +104,14 @@ public class MySqlLexer extends Lexer {
map.put("SOME", Token.SOME);
map.put("ANY", Token.ANY);
map.put("TRUNCATE", Token.TRUNCATE);
map.put("USER", Token.USER);
map.put("TRUE", Token.TRUE);
map.put("FALSE", Token.FALSE);
map.put("LIMIT", Token.LIMIT);
map.put("KILL", Token.KILL);
map.put("IDENTIFIED", Token.IDENTIFIED);
map.put("PASSWORD", Token.PASSWORD);
DEFAULT_MYSQL_KEYWORDS = new Keywords(map);
}
src/main/java/com/alibaba/druid/sql/dialect/mysql/parser/MySqlStatementParser.java
+868 -816
View File
File diff suppressed because it is too large Load Diff
src/main/java/com/alibaba/druid/sql/dialect/mysql/visitor/MySqlASTVisitor.java
+9
View File
@@ -25,6 +25,7 @@ import com.alibaba.druid.sql.dialect.mysql.ast.expr.MySqlMatchAgainstExpr;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.CobarShowStatus;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlBinlogStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlCommitStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlCreateUserStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlDeleteStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlExecuteStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlInsertStatement;
@@ -163,4 +164,12 @@ public interface MySqlASTVisitor extends SQLASTVisitor {
void endVisit(MySqlResetStatement x);
boolean visit(MySqlResetStatement x);
void endVisit(MySqlCreateUserStatement x);
boolean visit(MySqlCreateUserStatement x);
void endVisit(MySqlCreateUserStatement.UserSpecification x);
boolean visit(MySqlCreateUserStatement.UserSpecification x);
}
src/main/java/com/alibaba/druid/sql/dialect/mysql/visitor/MySqlASTVisitorAdapter.java
+22
View File
@@ -10,6 +10,8 @@ import com.alibaba.druid.sql.dialect.mysql.ast.expr.MySqlMatchAgainstExpr;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.CobarShowStatus;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlBinlogStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlCommitStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlCreateUserStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlCreateUserStatement.UserSpecification;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlDeleteStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlExecuteStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlInsertStatement;
@@ -345,4 +347,24 @@ public class MySqlASTVisitorAdapter extends SQLASTVisitorAdapter implements
return true;
}
@Override
public void endVisit(MySqlCreateUserStatement x) {
}
@Override
public boolean visit(MySqlCreateUserStatement x) {
return true;
}
@Override
public void endVisit(UserSpecification x) {
}
@Override
public boolean visit(UserSpecification x) {
return true;
}
}
src/main/java/com/alibaba/druid/sql/dialect/mysql/visitor/MySqlOutputVisitor.java
+1205 -1181
View File
File diff suppressed because it is too large Load Diff
src/main/java/com/alibaba/druid/sql/dialect/mysql/visitor/MySqlSchemaStatVisitor.java
+275 -256
View File
@@ -15,6 +15,7 @@ import com.alibaba.druid.sql.dialect.mysql.ast.expr.MySqlMatchAgainstExpr;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.CobarShowStatus;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlBinlogStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlCommitStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlCreateUserStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlDeleteStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlExecuteStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlInsertStatement;
@@ -26,6 +27,7 @@ import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlReplicateStatement
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlResetStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlRollbackStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlSelectGroupBy;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlCreateUserStatement.UserSpecification;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlSelectQueryBlock.Limit;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlShowColumnsStatement;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlShowDatabasesStatement;
@@ -38,376 +40,393 @@ import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
import com.alibaba.druid.stat.TableStat.Mode;
public class MySqlSchemaStatVisitor extends SchemaStatVisitor implements
MySqlASTVisitor {
public boolean visit(MySqlDeleteStatement x) {
aliasLocal.set(new HashMap<String, String>());
public class MySqlSchemaStatVisitor extends SchemaStatVisitor implements MySqlASTVisitor {
x.putAttribute("_original_use_mode", modeLocal.get());
modeLocal.set(Mode.Delete);
public boolean visit(MySqlDeleteStatement x) {
aliasLocal.set(new HashMap<String, String>());
aliasLocal.set(new HashMap<String, String>());
x.putAttribute("_original_use_mode", modeLocal.get());
modeLocal.set(Mode.Delete);
if (x.getTableNames().size() == 1) {
String ident = ((SQLIdentifierExpr) x.getTableNames().get(0))
.getName();
currentTableLocal.set(ident);
}
aliasLocal.set(new HashMap<String, String>());
for (SQLName tableName : x.getTableNames()) {
String ident = tableName.toString();
TableStat stat = tableStats.get(ident);
if (stat == null) {
stat = new TableStat();
tableStats.put(new TableStat.Name(ident), stat);
}
stat.incrementDeleteCount();
}
if (x.getTableNames().size() == 1) {
String ident = ((SQLIdentifierExpr) x.getTableNames().get(0)).getName();
currentTableLocal.set(ident);
}
accept(x.getWhere());
accept(x.getFrom());
accept(x.getUsing());
accept(x.getOrderBy());
accept(x.getLimit());
for (SQLName tableName : x.getTableNames()) {
String ident = tableName.toString();
TableStat stat = tableStats.get(ident);
if (stat == null) {
stat = new TableStat();
tableStats.put(new TableStat.Name(ident), stat);
}
stat.incrementDeleteCount();
}
return false;
}
accept(x.getWhere());
accept(x.getFrom());
accept(x.getUsing());
accept(x.getOrderBy());
accept(x.getLimit());
public void endVisit(MySqlDeleteStatement x) {
aliasLocal.set(null);
}
return false;
}
@Override
public void endVisit(MySqlInsertStatement x) {
Mode originalMode = (Mode) x.getAttribute("_original_use_mode");
modeLocal.set(originalMode);
}
public void endVisit(MySqlDeleteStatement x) {
aliasLocal.set(null);
}
@Override
public boolean visit(MySqlInsertStatement x) {
x.putAttribute("_original_use_mode", modeLocal.get());
modeLocal.set(Mode.Insert);
@Override
public void endVisit(MySqlInsertStatement x) {
Mode originalMode = (Mode) x.getAttribute("_original_use_mode");
modeLocal.set(originalMode);
}
aliasLocal.set(new HashMap<String, String>());
@Override
public boolean visit(MySqlInsertStatement x) {
x.putAttribute("_original_use_mode", modeLocal.get());
modeLocal.set(Mode.Insert);
String originalTable = currentTableLocal.get();
aliasLocal.set(new HashMap<String, String>());
if (x.getTableName() instanceof SQLIdentifierExpr) {
String ident = ((SQLIdentifierExpr) x.getTableName()).getName();
currentTableLocal.set(ident);
x.putAttribute("_old_local_", originalTable);
String originalTable = currentTableLocal.get();
TableStat stat = tableStats.get(ident);
if (stat == null) {
stat = new TableStat();
tableStats.put(new TableStat.Name(ident), stat);
}
stat.incrementInsertCount();
if (x.getTableName() instanceof SQLIdentifierExpr) {
String ident = ((SQLIdentifierExpr) x.getTableName()).getName();
currentTableLocal.set(ident);
x.putAttribute("_old_local_", originalTable);
Map<String, String> aliasMap = aliasLocal.get();
if (aliasMap != null) {
if (x.getAlias() != null) {
aliasMap.put(x.getAlias(), ident);
}
aliasMap.put(ident, ident);
}
}
TableStat stat = tableStats.get(ident);
if (stat == null) {
stat = new TableStat();
tableStats.put(new TableStat.Name(ident), stat);
}
stat.incrementInsertCount();
accept(x.getColumns());
accept(x.getValuesList());
accept(x.getQuery());
accept(x.getDuplicateKeyUpdate());
Map<String, String> aliasMap = aliasLocal.get();
if (aliasMap != null) {
if (x.getAlias() != null) {
aliasMap.put(x.getAlias(), ident);
}
aliasMap.put(ident, ident);
}
}
return false;
}
accept(x.getColumns());
accept(x.getValuesList());
accept(x.getQuery());
accept(x.getDuplicateKeyUpdate());
@Override
public boolean visit(MySqlBooleanExpr x) {
return false;
}
return true;
}
@Override
public boolean visit(MySqlBooleanExpr x) {
@Override
public void endVisit(MySqlBooleanExpr x) {
return true;
}
}
@Override
public void endVisit(MySqlBooleanExpr x) {
@Override
public boolean visit(Limit x) {
}
return true;
}
@Override
public boolean visit(Limit x) {
@Override
public void endVisit(Limit x) {
return true;
}
}
@Override
public void endVisit(Limit x) {
@Override
public boolean visit(MySqlTableIndex x) {
}
return true;
}
@Override
public boolean visit(MySqlTableIndex x) {
@Override
public void endVisit(MySqlTableIndex x) {
return true;
}
}
@Override
public void endVisit(MySqlTableIndex x) {
@Override
public boolean visit(MySqlKey x) {
}
return true;
}
@Override
public boolean visit(MySqlKey x) {
@Override
public void endVisit(MySqlKey x) {
return true;
}
}
@Override
public void endVisit(MySqlKey x) {
@Override
public boolean visit(MySqlPrimaryKey x) {
}
return true;
}
@Override
public boolean visit(MySqlPrimaryKey x) {
@Override
public void endVisit(MySqlPrimaryKey x) {
return true;
}
}
@Override
public void endVisit(MySqlPrimaryKey x) {
@Override
public void endVisit(MySqlIntervalExpr x) {
}
}
@Override
public void endVisit(MySqlIntervalExpr x) {
@Override
public boolean visit(MySqlIntervalExpr x) {
}
return true;
}
@Override
public boolean visit(MySqlIntervalExpr x) {
@Override
public void endVisit(MySqlExtractExpr x) {
return true;
}
}
@Override
public void endVisit(MySqlExtractExpr x) {
@Override
public boolean visit(MySqlExtractExpr x) {
}
return true;
}
@Override
public boolean visit(MySqlExtractExpr x) {
@Override
public void endVisit(MySqlMatchAgainstExpr x) {
return true;
}
}
@Override
public void endVisit(MySqlMatchAgainstExpr x) {
@Override
public boolean visit(MySqlMatchAgainstExpr x) {
}
return true;
}
@Override
public boolean visit(MySqlMatchAgainstExpr x) {
@Override
public void endVisit(MySqlBinaryExpr x) {
return true;
}
}
@Override
public void endVisit(MySqlBinaryExpr x) {
@Override
public boolean visit(MySqlBinaryExpr x) {
}
return true;
}
@Override
public boolean visit(MySqlBinaryExpr x) {
@Override
public void endVisit(MySqlPrepareStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlPrepareStatement x) {
@Override
public boolean visit(MySqlPrepareStatement x) {
}
return true;
}
@Override
public boolean visit(MySqlPrepareStatement x) {
@Override
public void endVisit(MySqlExecuteStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlExecuteStatement x) {
@Override
public boolean visit(MySqlExecuteStatement x) {
}
return true;
}
@Override
public boolean visit(MySqlExecuteStatement x) {
@Override
public void endVisit(MySqlLoadDataInFileStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlLoadDataInFileStatement x) {
@Override
public boolean visit(MySqlLoadDataInFileStatement x) {
}
return true;
}
@Override
public boolean visit(MySqlLoadDataInFileStatement x) {
@Override
public void endVisit(MySqlLoadXmlStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlLoadXmlStatement x) {
@Override
public boolean visit(MySqlLoadXmlStatement x) {
}
return true;
}
@Override
public boolean visit(MySqlLoadXmlStatement x) {
@Override
public void endVisit(MySqlReplicateStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlReplicateStatement x) {
@Override
public boolean visit(MySqlReplicateStatement x) {
}
return true;
}
@Override
public boolean visit(MySqlReplicateStatement x) {
@Override
public void endVisit(MySqlSelectGroupBy x) {
return true;
}
}
@Override
public void endVisit(MySqlSelectGroupBy x) {
@Override
public boolean visit(MySqlSelectGroupBy x) {
}
return true;
}
@Override
public boolean visit(MySqlSelectGroupBy x) {
@Override
public void endVisit(MySqlStartTransactionStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlStartTransactionStatement x) {
@Override
public boolean visit(MySqlStartTransactionStatement x) {
}
return true;
}
@Override
public boolean visit(MySqlStartTransactionStatement x) {
@Override
public void endVisit(MySqlCommitStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlCommitStatement x) {
@Override
public boolean visit(MySqlCommitStatement x) {
}
return true;
}
@Override
public boolean visit(MySqlCommitStatement x) {
@Override
public void endVisit(MySqlRollbackStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlRollbackStatement x) {
@Override
public boolean visit(MySqlRollbackStatement x) {
}
return true;
}
@Override
public boolean visit(MySqlRollbackStatement x) {
@Override
public void endVisit(MySqlShowColumnsStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlShowColumnsStatement x) {
@Override
public boolean visit(MySqlShowColumnsStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlShowTablesStatement x) {
@Override
public boolean visit(MySqlShowColumnsStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlShowTablesStatement x) {
@Override
public boolean visit(MySqlShowTablesStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlShowDatabasesStatement x) {
@Override
public boolean visit(MySqlShowTablesStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlShowDatabasesStatement x) {
@Override
public boolean visit(MySqlShowDatabasesStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlShowWarningsStatement x) {
@Override
public boolean visit(MySqlShowDatabasesStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlShowWarningsStatement x) {
@Override
public boolean visit(MySqlShowWarningsStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlShowStatusStatement x) {
@Override
public boolean visit(MySqlShowWarningsStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlShowStatusStatement x) {
@Override
public boolean visit(MySqlShowStatusStatement x) {
return true;
}
}
@Override
public void endVisit(CobarShowStatus x) {
@Override
public boolean visit(MySqlShowStatusStatement x) {
return true;
}
}
@Override
public void endVisit(CobarShowStatus x) {
@Override
public boolean visit(CobarShowStatus x) {
return true;
}
}
@Override
public void endVisit(MySqlKillStatement x) {
@Override
public boolean visit(CobarShowStatus x) {
return true;
}
}
@Override
public void endVisit(MySqlKillStatement x) {
@Override
public boolean visit(MySqlKillStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlBinlogStatement x) {
@Override
public boolean visit(MySqlKillStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlBinlogStatement x) {
@Override
public boolean visit(MySqlBinlogStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlResetStatement x) {
@Override
public boolean visit(MySqlBinlogStatement x) {
return true;
}
}
@Override
public void endVisit(MySqlResetStatement x) {
@Override
public boolean visit(MySqlResetStatement x) {
return true;
}
}
@Override
public boolean visit(MySqlResetStatement x) {
return true;
}
@Override
public void endVisit(MySqlCreateUserStatement x) {
}
@Override
public boolean visit(MySqlCreateUserStatement x) {
return true;
}
@Override
public void endVisit(UserSpecification x) {
}
@Override
public boolean visit(UserSpecification x) {
return true;
}
}
src/main/java/com/alibaba/druid/sql/dialect/postgresql/parser/PGLexer.java
+1
View File
@@ -83,6 +83,7 @@ public class PGLexer extends Lexer {
map.put("SOME", Token.SOME);
map.put("ANY", Token.ANY);
map.put("TRUNCATE", Token.TRUNCATE);
map.put("USER", Token.USER);
map.put("LIMIT", Token.LIMIT);
map.put("OFFSET", Token.OFFSET);
src/main/java/com/alibaba/druid/sql/parser/Token.java
+3
View File
@@ -67,6 +67,7 @@ public enum Token {
ON("ON"),
SCHEMA("SCHEMA"),
CAST("CAST"),
USER("USER"),
AND("AND"),
OR("OR"),
@@ -99,6 +100,8 @@ public enum Token {
FALSE("FALSE"),
LIMIT("LIMIT"),
KILL("KILL"),
IDENTIFIED("IDENTIFIED"),
PASSWORD("PASSWORD"),
//postgresql
WINDOW("WINDOW"),
src/test/java/com/alibaba/druid/bvt/sql/mysql/CommitTest.java
+34
View File
@@ -0,0 +1,34 @@
package com.alibaba.druid.bvt.sql.mysql;
import java.util.List;
import junit.framework.Assert;
import com.alibaba.druid.sql.MysqlTest;
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;
public class CommitTest extends MysqlTest {
public void test_0() throws Exception {
String sql = "COMMIT;";
MySqlStatementParser parser = new MySqlStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
SQLStatement statemen = statementList.get(0);
System.out.println(output(statementList));
Assert.assertEquals(1, statementList.size());
MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
statemen.accept(visitor);
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getFields());
Assert.assertTrue(visitor.getTables().size() == 0);
}
}
src/test/java/com/alibaba/druid/bvt/sql/mysql/CreateUserTest.java
+34
View File
@@ -0,0 +1,34 @@
package com.alibaba.druid.bvt.sql.mysql;
import java.util.List;
import junit.framework.Assert;
import com.alibaba.druid.sql.MysqlTest;
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;
public class CreateUserTest extends MysqlTest {
public void test_0() throws Exception {
String sql = "CREATE USER 'jeffrey'@'localhost' IDENTIFIED BY 'mypass';";
MySqlStatementParser parser = new MySqlStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
SQLStatement statemen = statementList.get(0);
System.out.println(output(statementList));
Assert.assertEquals(1, statementList.size());
MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
statemen.accept(visitor);
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getFields());
Assert.assertTrue(visitor.getTables().size() == 0);
}
}
src/test/java/com/alibaba/druid/bvt/sql/mysql/DropTableTest.java
+41
View File
@@ -0,0 +1,41 @@
package com.alibaba.druid.bvt.sql.mysql;
import java.util.List;
import junit.framework.Assert;
import com.alibaba.druid.sql.MysqlTest;
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;
public class DropTableTest extends MysqlTest {
public void test_0() throws Exception {
String sql = "DROP TABLE films, distributors;";
MySqlStatementParser parser = new MySqlStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
SQLStatement statemen = statementList.get(0);
System.out.println(output(statementList));
Assert.assertEquals(1, statementList.size());
MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
statemen.accept(visitor);
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getFields());
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("films")));
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("distributors")));
Assert.assertTrue(visitor.getTables().get(new TableStat.Name("films")).getDropCount() == 1);
Assert.assertTrue(visitor.getTables().get(new TableStat.Name("distributors")).getDropCount() == 1);
Assert.assertTrue(visitor.getFields().size() == 0);
}
}
src/test/java/com/alibaba/druid/sql/MysqlTest.java
+22
View File
@@ -0,0 +1,22 @@
package com.alibaba.druid.sql;
import java.util.List;
import junit.framework.TestCase;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlOutputVisitor;
import com.alibaba.druid.sql.dialect.postgresql.visitor.PGOutputVisitor;
public class MysqlTest extends TestCase {
protected String output(List<SQLStatement> stmtList) {
StringBuilder out = new StringBuilder();
MySqlOutputVisitor visitor = new MySqlOutputVisitor(out);
for (SQLStatement stmt : stmtList) {
stmt.accept(visitor);
}
return out.toString();
}
}