diff --git a/src/test/java/com/alibaba/druid/demo/sql/CreateTableSetSchemaDemo.java b/src/test/java/com/alibaba/druid/demo/sql/CreateTableSetSchemaDemo.java new file mode 100644 index 0000000000..1ca4a3dce3 --- /dev/null +++ b/src/test/java/com/alibaba/druid/demo/sql/CreateTableSetSchemaDemo.java @@ -0,0 +1,36 @@ +package com.alibaba.druid.demo.sql; + +import com.alibaba.druid.sql.SQLUtils; +import com.alibaba.druid.sql.ast.SQLStatement; +import com.alibaba.druid.sql.ast.statement.SQLCreateTableStatement; +import com.alibaba.druid.sql.parser.SQLParserUtils; +import com.alibaba.druid.sql.parser.SQLStatementParser; +import com.alibaba.druid.sql.visitor.SchemaStatVisitor; +import com.alibaba.druid.stat.TableStat; +import com.alibaba.druid.util.JdbcConstants; +import junit.framework.TestCase; +import org.apache.commons.io.FileUtils; + +import java.io.File; +import java.util.List; +import java.util.Set; + +public class CreateTableSetSchemaDemo extends TestCase { + + public void test_schemaStat() throws Exception { +String sql = "create table t(fid varchar(20))"; + +String dbType = JdbcConstants.ORACLE; +SQLStatementParser parser = SQLParserUtils.createSQLStatementParser(sql, dbType); +List stmtList = parser.parseStatementList(); + +SchemaStatVisitor statVisitor = SQLUtils.createSchemaStatVisitor(dbType); +for (SQLStatement stmt : stmtList) { + SQLCreateTableStatement createTable = ((SQLCreateTableStatement) stmt); + createTable.setSchema("sc001"); +} + +String sql2 = SQLUtils.toSQLString(stmtList, JdbcConstants.ORACLE); +System.out.println(sql2); + } +} diff --git a/src/test/java/com/alibaba/druid/demo/sql/PGVisitorDemo.java b/src/test/java/com/alibaba/druid/demo/sql/PGVisitorDemo.java index fa6ac35dcb..8cc611d6f1 100644 --- a/src/test/java/com/alibaba/druid/demo/sql/PGVisitorDemo.java +++ b/src/test/java/com/alibaba/druid/demo/sql/PGVisitorDemo.java @@ -27,16 +27,16 @@ public class PGVisitorDemo extends TestCase { System.out.println(tableSource); } -public static class ExportTableAliasVisitor extends PGASTVisitorAdapter { - private Map aliasMap = new HashMap(); - public boolean visit(SQLExprTableSource x) { - String alias = x.getAlias(); - aliasMap.put(alias, x); - return true; - } + public static class ExportTableAliasVisitor extends PGASTVisitorAdapter { + private Map aliasMap = new HashMap(); + public boolean visit(SQLExprTableSource x) { + String alias = x.getAlias(); + aliasMap.put(alias, x); + return true; + } - public Map getAliasMap() { - return aliasMap; + public Map getAliasMap() { + return aliasMap; + } } } -}