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: OracleSelectTest23.java, v 0.1 2012-5-17 ����10:20:41 xiaoqing.zhouxq Exp $ */ public class OracleSelectTest23 extends OracleTest { public void test_0() throws Exception { String sql = // "select (select decode(count(*),1,'YES','NO') FROM sys.obj$ o, sys.user$ u WHERE u.name = 'PERFSTAT' AND o.owner# = u.user# AND o.name = 'STATSPACK' AND o.type# = 11 AND o.status = 1) is_installed , (select nvl(INTERVAL,'') from dba_jobs where what like 'statspack.snap%' and SCHEMA_USER='PERFSTAT' and rownum =1) freq from dual"; // 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(3, visitor.getTables().size()); Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("sys.obj$"))); Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("sys.user$"))); Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("dba_jobs"))); Assert.assertEquals(10, 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"))); } }