package com.alipay.zdal.parser.druid.bvt.sql.oracle; import java.util.List; import junit.framework.Assert; import com.alipay.zdal.parser.druid.sql.OracleTest; import com.alipay.zdal.parser.sql.ast.SQLStatement; import com.alipay.zdal.parser.sql.dialect.oracle.parser.OracleStatementParser; import com.alipay.zdal.parser.sql.dialect.oracle.visitor.OracleSchemaStatVisitor; import com.alipay.zdal.parser.sql.stat.TableStat; /** * * @author xiaoqing.zhouxq * @version $Id: OracleSelectTest19.java, v 0.1 2012-5-17 ����10:20:23 xiaoqing.zhouxq Exp $ */ public class OracleSelectTest19 extends OracleTest { public void test_0() throws Exception { String sql = // "SELECT ddf.file_name file_name, vdf.status status, ddf.tablespace_name tablespace_name" + // ", '', ddf.autoextensible autoextensible" + // " , ddf.increment_by increment_by, ddf.maxbytes max_file_size, vdf.create_bytes " + // "FROM sys.dba_data_files ddf, v$datafile vdf /*+ all_rows use_concat */ " + // "WHERE (ddf.file_name = vdf.name) " + // "UNION ALL " + // "SELECT dtf.file_name file_name, vtf.status status, dtf.tablespace_name tablespace_name" + // " , '', dtf.autoextensible autoextensible, dtf.increment_by increment_by" + // " , dtf.maxbytes max_file_size, vtf.create_bytes " + // "FROM sys.dba_temp_files dtf, v$tempfile vtf " + // "WHERE (dtf.file_id = vtf.file#) "; // OracleStatementParser parser = new OracleStatementParser(sql); List<SQLStatement> statementList = parser.parseStatementList(); SQLStatement statemen = statementList.get(0); print(statementList); Assert.assertEquals(1, statementList.size()); OracleSchemaStatVisitor visitor = new OracleSchemaStatVisitor(); statemen.accept(visitor); System.out.println("Tables : " + visitor.getTables()); System.out.println("fields : " + visitor.getColumns()); System.out.println("alias : " + visitor.getAliasMap()); System.out.println("conditions : " + visitor.getConditions()); System.out.println("orderBy : " + visitor.getOrderByColumns()); System.out.println("groupBy : " + visitor.getGroupByColumns()); System.out.println("variant : " + visitor.getVariants()); System.out.println("relationShip : " + visitor.getRelationships()); Assert.assertEquals(4, visitor.getTables().size()); Assert .assertTrue(visitor.getTables().containsKey(new TableStat.Name("sys.dba_data_files"))); Assert.assertEquals(17, visitor.getColumns().size()); // Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("pivot_table", "*"))); // Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("pivot_table", "YEAR"))); // Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("pivot_table", "order_mode"))); } }