Index: ql/src/java/org/apache/hadoop/hive/ql/hooks/WriteEntity.java =================================================================== --- ql/src/java/org/apache/hadoop/hive/ql/hooks/WriteEntity.java (revision 772022) +++ ql/src/java/org/apache/hadoop/hive/ql/hooks/WriteEntity.java (working copy) @@ -118,8 +118,22 @@ return null; } - + /** + * Get the partition associated with the entity. + */ + public Partition getPartition() { + return p; + } + + /** + * Get the table associated with the entity. + */ + public Table getTable() { + return t; + } + + /** * toString function. */ public String toString() { Index: ql/src/java/org/apache/hadoop/hive/ql/hooks/ReadEntity.java =================================================================== --- ql/src/java/org/apache/hadoop/hive/ql/hooks/ReadEntity.java (revision 772022) +++ ql/src/java/org/apache/hadoop/hive/ql/hooks/ReadEntity.java (working copy) @@ -73,7 +73,7 @@ /** * Get the parameter map of the Entity. */ - public Map getParameter() { + public Map getParameters() { if (p != null) { return p.getTPartition().getParameters(); } @@ -93,6 +93,20 @@ return t.getDataLocation(); } } + + /** + * Get partition entity. + */ + public Partition getPartition() { + return p; + } + + /** + * Get table entity. + */ + public Table getTable() { + return t; + } /** * toString function.