add hive sql parser

src/main/java/com/alibaba/druid/sql/dialect/hive/ast/HiveSQLObject.java

@@ -0,0 +1,9 @@
+package com.alibaba.druid.sql.dialect.hive.ast;
+
+import com.alibaba.druid.sql.ast.SQLObject;
+import com.alibaba.druid.sql.dialect.hive.visitor.HiveASTVisitor;
+
+
+public interface HiveSQLObject extends SQLObject {
+    void accept0(HiveASTVisitor visitor);
+}

src/main/java/com/alibaba/druid/sql/dialect/hive/ast/HiveSQLObjectImpl.java

@@ -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.hive.ast;
+
+import com.alibaba.druid.sql.ast.SQLObjectImpl;
+import com.alibaba.druid.sql.dialect.hive.visitor.HiveASTVisitor;
+import com.alibaba.druid.sql.visitor.SQLASTVisitor;
+
+public abstract class HiveSQLObjectImpl extends SQLObjectImpl implements HiveSQLObject {
+
+    private static final long serialVersionUID = 1L;
+
+    @Override
+    public void accept0(SQLASTVisitor visitor) {
+        if (visitor instanceof HiveASTVisitor) {
+            accept0((HiveASTVisitor) visitor);
+        } else {
+            throw new IllegalArgumentException("not support visitor type : " + visitor.getClass().getName());
+        }
+    }
+
+    public abstract void accept0(HiveASTVisitor visitor);
+}

src/main/java/com/alibaba/druid/sql/dialect/hive/ast/HiveStatement.java

@@ -0,0 +1,7 @@
+package com.alibaba.druid.sql.dialect.hive.ast;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+
+public interface HiveStatement extends SQLStatement, HiveSQLObject {
+
+}

src/main/java/com/alibaba/druid/sql/dialect/hive/ast/stmt/HiveCreateTableStatement.java

@@ -0,0 +1,71 @@
+package com.alibaba.druid.sql.dialect.hive.ast.stmt;
+
+import com.alibaba.druid.sql.ast.SQLDataType;
+import com.alibaba.druid.sql.ast.statement.SQLCreateTableStatement;
+import com.alibaba.druid.sql.dialect.hive.ast.HiveSQLObjectImpl;
+import com.alibaba.druid.sql.dialect.hive.ast.HiveStatement;
+import com.alibaba.druid.sql.dialect.hive.visitor.HiveASTVisitor;
+import com.alibaba.druid.sql.visitor.SQLASTVisitor;
+
+public class HiveCreateTableStatement extends SQLCreateTableStatement implements HiveStatement {
+
+    private static final long serialVersionUID = 1L;
+
+    private PartitionedBy     partitionedBy;
+
+    public PartitionedBy getPartitionedBy() {
+        return partitionedBy;
+    }
+
+    public void setPartitionedBy(PartitionedBy partitionedBy) {
+        this.partitionedBy = partitionedBy;
+    }
+
+    @Override
+    protected void accept0(SQLASTVisitor visitor) {
+        this.accept0((HiveASTVisitor) visitor);
+    }
+
+    @Override
+    public void accept0(HiveASTVisitor visitor) {
+        if (visitor.visit(this)) {
+            this.acceptChild(visitor, name);
+            this.acceptChild(visitor, tableElementList);
+            this.acceptChild(visitor, partitionedBy);
+        }
+        visitor.endVisit(this);
+    }
+
+    public static class PartitionedBy extends HiveSQLObjectImpl {
+
+        private static final long serialVersionUID = 1L;
+        private String            name;
+        private SQLDataType       type;
+
+        public String getName() {
+            return name;
+        }
+
+        public void setName(String name) {
+            this.name = name;
+        }
+
+        public SQLDataType getType() {
+            return type;
+        }
+
+        public void setType(SQLDataType type) {
+            this.type = type;
+        }
+
+        @Override
+        public void accept0(HiveASTVisitor visitor) {
+            if (visitor.visit(this)) {
+                acceptChild(visitor, type);
+            }
+            visitor.endVisit(this);
+        }
+
+    }
+
+}

src/main/java/com/alibaba/druid/sql/dialect/hive/parser/HiveCreateTableParser.java

@@ -0,0 +1,44 @@
+package com.alibaba.druid.sql.dialect.hive.parser;
+
+import com.alibaba.druid.sql.ast.statement.SQLCreateTableStatement;
+import com.alibaba.druid.sql.dialect.hive.ast.stmt.HiveCreateTableStatement;
+import com.alibaba.druid.sql.dialect.hive.ast.stmt.HiveCreateTableStatement.PartitionedBy;
+import com.alibaba.druid.sql.parser.Lexer;
+import com.alibaba.druid.sql.parser.SQLCreateTableParser;
+import com.alibaba.druid.sql.parser.Token;
+
+public class HiveCreateTableParser extends SQLCreateTableParser {
+
+    public HiveCreateTableParser(String sql){
+        super(new HiveLexer(sql));
+        this.lexer.nextToken();
+    }
+
+    public HiveCreateTableParser(Lexer lexer){
+        super(lexer);
+    }
+
+    protected HiveCreateTableStatement newCreateStatement() {
+        return new HiveCreateTableStatement();
+    }
+
+    public SQLCreateTableStatement parseCrateTable(boolean acceptCreate) {
+        HiveCreateTableStatement stmt = (HiveCreateTableStatement) super.parseCrateTable(acceptCreate);
+
+        if (lexer.token() == Token.PARTITIONED) {
+            lexer.nextToken();
+            accept(Token.BY);
+            accept(Token.LPAREN);
+
+            PartitionedBy partitionedBy = new PartitionedBy();
+            partitionedBy.setName(exprParser.name().toString());
+            partitionedBy.setType(exprParser.parseDataType());
+
+            accept(Token.RPAREN);
+            
+            stmt.setPartitionedBy(partitionedBy);
+        }
+
+        return stmt;
+    }
+}

src/main/java/com/alibaba/druid/sql/dialect/hive/parser/HiveLexer.java

@@ -1,14 +1,102 @@
 package com.alibaba.druid.sql.dialect.hive.parser;
 
+import java.util.HashMap;
+import java.util.Map;
+
+import com.alibaba.druid.sql.parser.Keywords;
 import com.alibaba.druid.sql.parser.Lexer;
 import com.alibaba.druid.sql.parser.Token;
 
 public class HiveLexer extends Lexer {
+    public final static Keywords DEFAULT_SQL_SERVER_KEYWORDS;
 
-	public HiveLexer(char[] input, int inputLength, boolean skipComment) {
-		super(input, inputLength, skipComment);
-	}
+    static {
+        Map<String, Token> map = new HashMap<String, Token>();
+        map.put("EXISTS", Token.EXISTS);
+        map.put("THEN", Token.THEN);
+        map.put("AS", Token.AS);
+        map.put("GROUP", Token.GROUP);
+        map.put("BY", Token.BY);
+        map.put("HAVING", Token.HAVING);
+        map.put("DELETE", Token.DELETE);
+        map.put("ORDER", Token.ORDER);
+        map.put("INDEX", Token.INDEX);
+        map.put("FOR", Token.FOR);
+        map.put("SCHEMA", Token.SCHEMA);
+        map.put("FOREIGN", Token.FOREIGN);
+        map.put("REFERENCE", Token.REFERENCE);
+        map.put("REFERENCES", Token.REFERENCES);
+        map.put("CHECK", Token.CHECK);
+        map.put("PRIMARY", Token.PRIMARY);
+        map.put("KEY", Token.KEY);
+        map.put("CONSTRAINT", Token.CONSTRAINT);
+        map.put("DEFAULT", Token.DEFAULT);
+        map.put("VIEW", Token.VIEW);
+        map.put("CREATE", Token.CREATE);
+        map.put("VALUES", Token.VALUES);
+        map.put("ALTER", Token.ALTER);
+        map.put("TABLE", Token.TABLE);
+        map.put("DROP", Token.DROP);
+        map.put("SET", Token.SET);
+        map.put("INTO", Token.INTO);
+        map.put("UPDATE", Token.UPDATE);
+        map.put("NULL", Token.NULL);
+        map.put("IS", Token.IS);
+        map.put("NOT", Token.NOT);
+        map.put("SELECT", Token.SELECT);
+        map.put("INSERT", Token.INSERT);
+        map.put("FROM", Token.FROM);
+        map.put("WHERE", Token.WHERE);
+        map.put("AND", Token.AND);
+        map.put("OR", Token.OR);
+        map.put("XOR", Token.XOR);
+        map.put("DISTINCT", Token.DISTINCT);
+        map.put("UNIQUE", Token.UNIQUE);
+        map.put("ALL", Token.ALL);
+        map.put("UNION", Token.UNION);
+        map.put("INTERSECT", Token.INTERSECT);
+        map.put("MINUS", Token.MINUS);
+        map.put("INNER", Token.INNER);
+        map.put("LEFT", Token.LEFT);
+        map.put("RIGHT", Token.RIGHT);
+        map.put("FULL", Token.FULL);
+        map.put("ON", Token.ON);
+        map.put("OUTER", Token.OUTER);
+        map.put("JOIN", Token.JOIN);
+        map.put("NEW", Token.NEW);
+        map.put("CASE", Token.CASE);
+        map.put("WHEN", Token.WHEN);
+        map.put("END", Token.END);
+        map.put("WHEN", Token.WHEN);
+        map.put("ELSE", Token.ELSE);
+        map.put("EXISTS", Token.EXISTS);
+        map.put("CAST", Token.CAST);
+        map.put("IN", Token.IN);
+        map.put("ASC", Token.ASC);
+        map.put("DESC", Token.DESC);
+        map.put("LIKE", Token.LIKE);
+        map.put("ESCAPE", Token.ESCAPE);
+        map.put("BETWEEN", Token.BETWEEN);
+        map.put("INTERVAL", Token.INTERVAL);
+        map.put("LOCK", Token.LOCK);
+        map.put("SOME", Token.SOME);
+        map.put("ANY", Token.ANY);
 
+        map.put("WITH", Token.WITH);
+        map.put("PARTITIONED", Token.PARTITIONED);
+
+        DEFAULT_SQL_SERVER_KEYWORDS = new Keywords(map);
+    }
+    
+    public HiveLexer(char[] input, int inputLength, boolean skipComment){
+        super(input, inputLength, skipComment);
+        super.keywods = DEFAULT_SQL_SERVER_KEYWORDS;
+    }
+
+    public HiveLexer(String input) {
+        super(input);
+        super.keywods = DEFAULT_SQL_SERVER_KEYWORDS;
+    }
 
     public void scanComment() {
         if (ch != '/') {

src/main/java/com/alibaba/druid/sql/dialect/hive/parser/HiveStatementParser.java

@@ -0,0 +1,22 @@
+package com.alibaba.druid.sql.dialect.hive.parser;
+
+import com.alibaba.druid.sql.parser.Lexer;
+import com.alibaba.druid.sql.parser.SQLStatementParser;
+
+
+public class HiveStatementParser extends SQLStatementParser {
+
+    public HiveStatementParser(Lexer lexer){
+        super(lexer);
+    }
+
+    public HiveStatementParser(String sql){
+        super(new HiveLexer(sql));
+        this.lexer.nextToken();
+    }
+
+    public HiveCreateTableParser getSQLCreateTableParser() {
+        return new HiveCreateTableParser(lexer);
+    }
+
+}

src/main/java/com/alibaba/druid/sql/dialect/hive/visitor/HiveASTVisitor.java

@@ -0,0 +1,15 @@
+package com.alibaba.druid.sql.dialect.hive.visitor;
+
+import com.alibaba.druid.sql.dialect.hive.ast.stmt.HiveCreateTableStatement;
+import com.alibaba.druid.sql.visitor.SQLASTVisitor;
+
+public interface HiveASTVisitor extends SQLASTVisitor {
+
+    void endVisit(HiveCreateTableStatement x);
+
+    boolean visit(HiveCreateTableStatement x);
+    
+    void endVisit(HiveCreateTableStatement.PartitionedBy x);
+    
+    boolean visit(HiveCreateTableStatement.PartitionedBy x);
+}

src/main/java/com/alibaba/druid/sql/dialect/hive/visitor/HiveASTVisitorAdapter.java

@@ -0,0 +1,28 @@
+package com.alibaba.druid.sql.dialect.hive.visitor;
+
+import com.alibaba.druid.sql.dialect.hive.ast.stmt.HiveCreateTableStatement;
+import com.alibaba.druid.sql.visitor.SQLASTVisitorAdapter;
+
+public class HiveASTVisitorAdapter extends SQLASTVisitorAdapter implements HiveASTVisitor {
+
+    @Override
+    public void endVisit(HiveCreateTableStatement x) {
+
+    }
+
+    @Override
+    public boolean visit(HiveCreateTableStatement x) {
+        return true;
+    }
+
+    @Override
+    public void endVisit(HiveCreateTableStatement.PartitionedBy x) {
+
+    }
+
+    @Override
+    public boolean visit(HiveCreateTableStatement.PartitionedBy x) {
+        return true;
+    }
+
+}

src/main/java/com/alibaba/druid/sql/dialect/hive/visitor/HiveOutputVisitor.java

@@ -0,0 +1,45 @@
+package com.alibaba.druid.sql.dialect.hive.visitor;
+
+import com.alibaba.druid.sql.ast.statement.SQLCreateTableStatement;
+import com.alibaba.druid.sql.dialect.hive.ast.stmt.HiveCreateTableStatement;
+import com.alibaba.druid.sql.dialect.hive.ast.stmt.HiveCreateTableStatement.PartitionedBy;
+import com.alibaba.druid.sql.visitor.SQLASTOutputVisitor;
+
+public class HiveOutputVisitor extends SQLASTOutputVisitor implements HiveASTVisitor {
+
+    public HiveOutputVisitor(Appendable appender){
+        super(appender);
+    }
+
+    @Override
+    public void endVisit(HiveCreateTableStatement x) {
+        
+    }
+
+    @Override
+    public boolean visit(HiveCreateTableStatement x) {
+        visit((SQLCreateTableStatement) x);
+        
+        if (x.getPartitionedBy() != null) {
+            println();
+            x.getPartitionedBy().accept(this);
+        }
+        return false;
+    }
+
+    @Override
+    public void endVisit(PartitionedBy x) {
+        
+    }
+
+    @Override
+    public boolean visit(PartitionedBy x) {
+        print("PARTITIONED BY (");
+        print(x.getName());
+        print(" ");
+        x.getType().accept(this);
+        print(")");
+        return false;
+    }
+
+}

src/main/java/com/alibaba/druid/sql/dialect/hive/visitor/HiveSchemaStatVisitor.java

@@ -0,0 +1,28 @@
+package com.alibaba.druid.sql.dialect.hive.visitor;
+
+import com.alibaba.druid.sql.ast.statement.SQLCreateTableStatement;
+import com.alibaba.druid.sql.dialect.hive.ast.stmt.HiveCreateTableStatement;
+import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
+
+public class HiveSchemaStatVisitor extends SchemaStatVisitor implements HiveASTVisitor {
+
+    @Override
+    public void endVisit(HiveCreateTableStatement x) {
+        
+    }
+
+    @Override
+    public boolean visit(HiveCreateTableStatement x) {
+        return visit((SQLCreateTableStatement) x);
+    }
+
+    @Override
+    public void endVisit(HiveCreateTableStatement.PartitionedBy x) {
+
+    }
+
+    @Override
+    public boolean visit(HiveCreateTableStatement.PartitionedBy x) {
+        return false;
+    }
+}

src/main/java/com/alibaba/druid/sql/dialect/sqlserver/parser/SQLServerStatementParser.java

@@ -15,17 +15,8 @@
  */
 package com.alibaba.druid.sql.dialect.sqlserver.parser;
 
-import java.util.List;
-
-import com.alibaba.druid.sql.ast.SQLStatement;
-import com.alibaba.druid.sql.ast.statement.SQLSelectStatement;
-import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OraclePLSQLCommitStatement;
-import com.alibaba.druid.sql.dialect.oracle.parser.OracleDeleteParser;
-import com.alibaba.druid.sql.dialect.oracle.parser.OracleSelectParser;
 import com.alibaba.druid.sql.parser.Lexer;
-import com.alibaba.druid.sql.parser.ParserException;
 import com.alibaba.druid.sql.parser.SQLStatementParser;
-import com.alibaba.druid.sql.parser.Token;
 
 public class SQLServerStatementParser extends SQLStatementParser {
 
@@ -38,78 +29,4 @@ public class SQLServerStatementParser extends SQLStatementParser {
         super(lexer);
     }
 
-    @Override
-    public void parseStatementList(List<SQLStatement> statementList) throws ParserException {
-        for (;;) {
-            if (lexer.token() == Token.EOF) {
-                return;
-            }
-
-            if (lexer.token() == (Token.SEMI)) {
-                lexer.nextToken();
-                continue;
-            }
-
-            if (lexer.token() == (Token.SELECT)) {
-                statementList.add(new SQLSelectStatement(new SQLServerSelectParser(this.lexer).select()));
-                continue;
-            }
-
-            if (lexer.token() == (Token.UPDATE)) {
-                statementList.add(parseUpdateStatement());
-                continue;
-            }
-
-            if (lexer.token() == (Token.CREATE)) {
-                lexer.nextToken();
-
-                // if ((this.tokenList.lookup(1).equals(Token.ViewToken)) ||
-                // (this.tokenList.lookup(3).equals(Token.ViewToken))
-                // || (this.tokenList.lookup(4).equals(Token.ViewToken)) ||
-                // (this.tokenList.lookup(5).equals(Token.ViewToken))) {
-                // statementList.add(parseCreateView());
-                // }
-                //
-                // if ((this.tokenList.lookup(1).equals(Token.TableToken)) ||
-                // (this.tokenList.lookup(3).equals(Token.TableToken))) {
-                // statementList.add(parseOracleCreateTable());
-                // }
-
-                throw new ParserException("TODO");
-            }
-
-            if (lexer.token() == Token.INSERT) {
-                statementList.add(parseInsert());
-                continue;
-            }
-
-            if (lexer.token() == (Token.DELETE)) {
-                statementList.add(new OracleDeleteParser(this.lexer).parseDelete());
-                continue;
-            }
-
-            if (lexer.token() == (Token.SLASH)) {
-                lexer.nextToken();
-                statementList.add(new OraclePLSQLCommitStatement());
-                continue;
-            }
-
-            if (lexer.token() == Token.ALTER) {
-                throw new ParserException("TODO");
-            }
-
-            if (lexer.token() == Token.WITH) {
-                statementList.add(new SQLSelectStatement(new OracleSelectParser(this.lexer).select()));
-                continue;
-            }
-
-            if (identifierEquals("CALL")) {
-                statementList.add(this.parseCall());
-                continue;
-            }
-
-            throw new ParserException("TODO : " + lexer.token() + " " + lexer.stringVal());
-        }
-    }
-
 }

src/main/java/com/alibaba/druid/sql/parser/SQLCreateTableParser.java

@@ -37,7 +37,7 @@ public class SQLCreateTableParser extends SQLDDLParser {
             accept(Token.CREATE);
         }
 
-        SQLCreateTableStatement createTable = new SQLCreateTableStatement();
+        SQLCreateTableStatement createTable = newCreateStatement();
 
         if (lexer.token() == Token.IDENTIFIER && lexer.stringVal().equalsIgnoreCase("GLOBAL")) {
             lexer.nextToken();
@@ -97,4 +97,8 @@ public class SQLCreateTableParser extends SQLDDLParser {
 
         return createTable;
     }
+
+    protected SQLCreateTableStatement newCreateStatement() {
+        return new SQLCreateTableStatement();
+    }
 }

src/main/java/com/alibaba/druid/sql/parser/Token.java

@@ -154,6 +154,9 @@ public enum Token {
     
     // transact-sql
     TOP("TOP"),
+    
+    // hive
+    PARTITIONED("PARTITIONED"),
 
     EOF, 
     ERROR,

src/main/java/com/alibaba/druid/sql/visitor/SchemaStatVisitor.java

@@ -20,6 +20,8 @@ import com.alibaba.druid.sql.ast.expr.SQLInSubQueryExpr;
 import com.alibaba.druid.sql.ast.expr.SQLIntegerExpr;
 import com.alibaba.druid.sql.ast.expr.SQLMethodInvokeExpr;
 import com.alibaba.druid.sql.ast.expr.SQLPropertyExpr;
+import com.alibaba.druid.sql.ast.statement.SQLColumnDefinition;
+import com.alibaba.druid.sql.ast.statement.SQLCreateTableStatement;
 import com.alibaba.druid.sql.ast.statement.SQLDeleteStatement;
 import com.alibaba.druid.sql.ast.statement.SQLDropTableStatement;
 import com.alibaba.druid.sql.ast.statement.SQLExprTableSource;
@@ -32,6 +34,7 @@ import com.alibaba.druid.sql.ast.statement.SQLSelectQuery;
 import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
 import com.alibaba.druid.sql.ast.statement.SQLSelectStatement;
 import com.alibaba.druid.sql.ast.statement.SQLSubqueryTableSource;
+import com.alibaba.druid.sql.ast.statement.SQLTableElement;
 import com.alibaba.druid.sql.ast.statement.SQLTruncateStatement;
 import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement;
 import com.alibaba.druid.stat.TableStat;
@@ -149,7 +152,7 @@ public class SchemaStatVisitor extends SQLASTVisitorAdapter {
             case IsNot:
                 handleCondition(x.getLeft(), x.getOperator().name);
                 handleCondition(x.getRight(), x.getOperator().name);
-                
+
                 handleRelationship(x.getLeft(), x.getOperator().name, x.getRight());
                 break;
             default:
@@ -163,17 +166,17 @@ public class SchemaStatVisitor extends SQLASTVisitorAdapter {
         if (leftColumn == null) {
             return;
         }
-        
+
         Column rightColumn = getColumn(right);
         if (rightColumn == null) {
             return;
         }
-        
+
         Relationship relationship = new Relationship();
         relationship.setLeft(leftColumn);
         relationship.setRight(rightColumn);
         relationship.setOperator(operator);
-        
+
         this.relationships.add(relationship);
     }
 
@@ -182,7 +185,7 @@ public class SchemaStatVisitor extends SQLASTVisitorAdapter {
         if (column == null) {
             return;
         }
-        
+
         Condition condition = new Condition();
         condition.setColumn(column);
         condition.setOperator(operator);
@@ -209,7 +212,7 @@ public class SchemaStatVisitor extends SQLASTVisitorAdapter {
                 if (table != null) {
                     return new Column(table, column);
                 }
-                
+
                 return handleSubQueryColumn(tableName, column);
             }
 
@@ -229,7 +232,7 @@ public class SchemaStatVisitor extends SQLASTVisitorAdapter {
                 return new Column(table, column);
             }
         }
-        
+
         return null;
     }
 
@@ -660,4 +663,35 @@ public class SchemaStatVisitor extends SQLASTVisitorAdapter {
     public void endVisit(SQLUpdateStatement x) {
         aliasLocal.set(null);
     }
+
+    public boolean visit(SQLCreateTableStatement x) {
+        for (SQLTableElement e : x.getTableElementList()) {
+            e.setParent(x);
+        }
+
+        String tableName = x.getName().toString();
+        TableStat stat = new TableStat();
+        stat.incrementCreateCount();
+        tableStats.put(new TableStat.Name(tableName), stat);
+        return true;
+    }
+
+    public boolean visit(SQLColumnDefinition x) {
+        String tableName = null;
+        {
+            SQLObject parent = x.getParent();
+            if (parent instanceof SQLCreateTableStatement) {
+                tableName = ((SQLCreateTableStatement) parent).getName().toString();
+            }
+        }
+        
+        if (tableName == null) {
+            return true;
+        }
+        
+        String columnName = x.getName().toString();
+        columns.add(new Column(tableName, columnName));
+        
+        return true;
+    }
 }

src/main/java/com/alibaba/druid/stat/TableStat.java

@@ -23,6 +23,15 @@ public class TableStat {
     int insertCount = 0;
     int dropCount   = 0;
     int mergeCount  = 0;
+    int createCount = 0;
+
+    public int getCreateCount() {
+        return createCount;
+    }
+
+    public void incrementCreateCount() {
+        this.createCount++;
+    }
 
     public int getMergeCount() {
         return mergeCount;
@@ -112,6 +121,9 @@ public class TableStat {
         if (dropCount > 0) {
             buf.append("Drop");
         }
+        if (createCount > 0) {
+            buf.append("Create");
+        }
 
         return buf.toString();
     }

src/test/java/com/alibaba/druid/bvt/sql/hive/HiveCreateTableTest.java

@@ -0,0 +1,40 @@
+package com.alibaba.druid.bvt.sql.hive;
+
+import java.util.List;
+
+import junit.framework.Assert;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.hive.parser.HiveStatementParser;
+import com.alibaba.druid.sql.dialect.hive.visitor.HiveSchemaStatVisitor;
+import com.alibaba.druid.sql.hive.HiveTest;
+import com.alibaba.druid.stat.TableStat;
+
+public class HiveCreateTableTest extends HiveTest {
+
+    public void test_hive() throws Exception {
+        String sql = "CREATE TABLE pokes (foo INT, bar STRING);";
+
+        HiveStatementParser parser = new HiveStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        System.out.println(output(statementList));
+
+        Assert.assertEquals(1, statementList.size());
+
+        HiveSchemaStatVisitor visitor = new HiveSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+        System.out.println("coditions : " + visitor.getConditions());
+        System.out.println("relationships : " + visitor.getRelationships());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("pokes")));
+
+        Assert.assertEquals(2, visitor.getColumns().size());
+        
+        Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("pokes", "foo")));
+        Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("pokes", "bar")));
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/hive/HiveCreateTableTest2.java

@@ -0,0 +1,40 @@
+package com.alibaba.druid.bvt.sql.hive;
+
+import java.util.List;
+
+import junit.framework.Assert;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.hive.parser.HiveStatementParser;
+import com.alibaba.druid.sql.dialect.hive.visitor.HiveSchemaStatVisitor;
+import com.alibaba.druid.sql.hive.HiveTest;
+import com.alibaba.druid.stat.TableStat;
+
+public class HiveCreateTableTest2 extends HiveTest {
+
+    public void test_hive() throws Exception {
+        String sql = "CREATE TABLE invites (foo INT, bar STRING) PARTITIONED BY (ds STRING);";
+
+        HiveStatementParser parser = new HiveStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        System.out.println(output(statementList));
+
+        Assert.assertEquals(1, statementList.size());
+
+        HiveSchemaStatVisitor visitor = new HiveSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+        System.out.println("coditions : " + visitor.getConditions());
+        System.out.println("relationships : " + visitor.getRelationships());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("invites")));
+
+        Assert.assertEquals(2, visitor.getColumns().size());
+        
+        Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("invites", "foo")));
+        Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("invites", "bar")));
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/hive/HiveSelectTest.java

@@ -0,0 +1,38 @@
+package com.alibaba.druid.bvt.sql.hive;
+
+import java.util.List;
+
+import junit.framework.Assert;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.hive.parser.HiveStatementParser;
+import com.alibaba.druid.sql.dialect.hive.visitor.HiveSchemaStatVisitor;
+import com.alibaba.druid.sql.hive.HiveTest;
+import com.alibaba.druid.stat.TableStat;
+
+public class HiveSelectTest extends HiveTest {
+
+    public void test_hive() throws Exception {
+        String sql = "SELECT COUNT(*) FROM u_data;";
+
+        HiveStatementParser parser = new HiveStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        System.out.println(output(statementList));
+
+        Assert.assertEquals(1, statementList.size());
+
+        HiveSchemaStatVisitor visitor = new HiveSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+        System.out.println("coditions : " + visitor.getConditions());
+        System.out.println("relationships : " + visitor.getRelationships());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("u_data")));
+
+        Assert.assertEquals(1, visitor.getColumns().size());
+        Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("u_data", "*")));
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/hive/HiveSelectTest2.java

@@ -0,0 +1,40 @@
+package com.alibaba.druid.bvt.sql.hive;
+
+import java.util.List;
+
+import junit.framework.Assert;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.hive.parser.HiveStatementParser;
+import com.alibaba.druid.sql.dialect.hive.visitor.HiveSchemaStatVisitor;
+import com.alibaba.druid.sql.hive.HiveTest;
+import com.alibaba.druid.stat.TableStat;
+
+public class HiveSelectTest2 extends HiveTest {
+
+    public void test_hive() throws Exception {
+        String sql = "SELECT a.foo FROM invites a WHERE a.ds='2008-08-15';";
+
+        HiveStatementParser parser = new HiveStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        System.out.println(output(statementList));
+
+        Assert.assertEquals(1, statementList.size());
+
+        HiveSchemaStatVisitor visitor = new HiveSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+        System.out.println("coditions : " + visitor.getConditions());
+        System.out.println("relationships : " + visitor.getRelationships());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("invites")));
+
+        Assert.assertEquals(2, visitor.getColumns().size());
+        
+        Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("invites", "foo")));
+        Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("invites", "ds")));
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/oracle/OracleSelectTest2.java

@@ -49,7 +49,7 @@ public class OracleSelectTest2 extends OracleTest {
 
         Assert.assertEquals(4, visitor.getColumns().size());
 
-        // Assert.assertTrue(visitor.getFields().contains(new TableStat.Column("films", "producer_id")));
+        // Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("films", "producer_id")));
     }
 
 }

src/test/java/com/alibaba/druid/sql/hive/HiveTest.java

@@ -0,0 +1,22 @@
+package com.alibaba.druid.sql.hive;
+
+import java.util.List;
+
+import junit.framework.TestCase;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.hive.visitor.HiveOutputVisitor;
+
+public class HiveTest extends TestCase {
+
+    protected String output(List<SQLStatement> stmtList) {
+        StringBuilder out = new StringBuilder();
+        HiveOutputVisitor visitor = new HiveOutputVisitor(out);
+
+        for (SQLStatement stmt : stmtList) {
+            stmt.accept(visitor);
+        }
+
+        return out.toString();
+    }
+}