diff --git ql/src/java/org/apache/hadoop/hive/ql/exec/tez/TezJobMonitor.java ql/src/java/org/apache/hadoop/hive/ql/exec/tez/TezJobMonitor.java index 731d58e..1cabca6 100644 --- ql/src/java/org/apache/hadoop/hive/ql/exec/tez/TezJobMonitor.java +++ ql/src/java/org/apache/hadoop/hive/ql/exec/tez/TezJobMonitor.java @@ -35,6 +35,7 @@ import org.apache.tez.dag.api.client.DAGClient; import org.apache.tez.dag.api.client.DAGStatus; import org.apache.tez.dag.api.client.Progress; +import org.apache.tez.dag.api.client.StatusGetOpts; /** * TezJobMonitor keeps track of a tez job while it's being executed. It will @@ -76,6 +77,7 @@ public int monitorExecution(DAGClient dagClient) throws InterruptedException { int rc = 0; DAGStatus.State lastState = null; String lastReport = null; + Set opts = new HashSet(); console.printInfo("\n"); perfLogger.PerfLogBegin(CLASS_NAME, PerfLogger.TEZ_RUN_DAG); @@ -84,7 +86,7 @@ public int monitorExecution(DAGClient dagClient) throws InterruptedException { while(true) { try { - status = dagClient.getDAGStatus(); + status = dagClient.getDAGStatus(opts); Map progressMap = status.getVertexProgress(); DAGStatus.State state = status.getState();