-
Notifications
You must be signed in to change notification settings - Fork 77
Expand file tree
/
Copy pathtestExplain.java
More file actions
51 lines (44 loc) · 2.71 KB
/
Copy pathtestExplain.java
File metadata and controls
51 lines (44 loc) · 2.71 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
package hive;
import gudusoft.gsqlparser.EDbVendor;
import gudusoft.gsqlparser.ESqlStatementType;
import gudusoft.gsqlparser.TGSqlParser;
import gudusoft.gsqlparser.nodes.TGroupBy;
import gudusoft.gsqlparser.nodes.EHiveInsertType;
import gudusoft.gsqlparser.stmt.TInsertSqlStatement;
import gudusoft.gsqlparser.stmt.TSelectSqlStatement;
import gudusoft.gsqlparser.stmt.hive.THiveExplain;
import junit.framework.TestCase;
public class testExplain extends TestCase {
public void test1(){
TGSqlParser sqlparser = new TGSqlParser(EDbVendor.dbvhive);
sqlparser.sqltext = "EXPLAIN\n" +
"FROM src INSERT OVERWRITE TABLE dest_g1 SELECT src.key, sum(substr(src.value,4)) GROUP BY src.key;";
assertTrue(sqlparser.parse() == 0);
THiveExplain explain = (THiveExplain)sqlparser.sqlstatements.get(0);
assertTrue(explain.getStmt().sqlstatementtype == ESqlStatementType.sstinsert);
TInsertSqlStatement insert = (TInsertSqlStatement)explain.getStmt();
// TInsertSqlStatement insert = (TInsertSqlStatement)fromQuery.getHiveBodyList().get(0);
assertTrue(insert.getHiveInsertType() == EHiveInsertType.overwriteTable);
assertTrue(insert.getTargetTable().getTableName().toString().equalsIgnoreCase("dest_g1"));
TSelectSqlStatement fromQuery = insert.getSubQuery();
assertTrue(fromQuery.tables.getTable(0).getTableName().toString().equalsIgnoreCase("src"));
// assertTrue(fromQuery.getHiveBodyList().size() == 1);
// TSelectSqlStatement select = (TSelectSqlStatement)insert.getSubQuery();
TGroupBy groupBy = fromQuery.getGroupByClause();
assertTrue(groupBy.getItems().getGroupByItem(0).getExpr().toString().equalsIgnoreCase("src.key"));
}
public void test2(){
TGSqlParser sqlparser = new TGSqlParser(EDbVendor.dbvhive);
sqlparser.sqltext = "EXPLAIN DEPENDENCY\n" +
" SELECT key, count(1) FROM srcpart WHERE ds IS NOT NULL GROUP BY key;";
assertTrue(sqlparser.parse() == 0);
THiveExplain explain = (THiveExplain)sqlparser.sqlstatements.get(0);
assertTrue(explain.getStmt().sqlstatementtype == ESqlStatementType.sstselect);
TSelectSqlStatement select = (TSelectSqlStatement)explain.getStmt();
assertTrue(select.getResultColumnList().size() == 2);
assertTrue(select.tables.getTable(0).getTableName().toString().equalsIgnoreCase("srcpart"));
assertTrue(select.getWhereClause().getCondition().toString().equalsIgnoreCase("ds IS NOT NULL"));
TGroupBy groupBy = select.getGroupByClause();
assertTrue(groupBy.getItems().getGroupByItem(0).getExpr().toString().equalsIgnoreCase("key"));
}
}