diff --git a/ql/src/java/org/apache/hadoop/hive/ql/exec/tez/TezTask.java b/ql/src/java/org/apache/hadoop/hive/ql/exec/tez/TezTask.java index 2356345..a1c5193 100644 --- a/ql/src/java/org/apache/hadoop/hive/ql/exec/tez/TezTask.java +++ b/ql/src/java/org/apache/hadoop/hive/ql/exec/tez/TezTask.java @@ -64,6 +64,7 @@ import org.apache.tez.dag.api.VertexGroup; import org.apache.tez.dag.api.client.DAGClient; import org.apache.tez.dag.api.client.StatusGetOpts; +import org.json.JSONObject; /** * @@ -279,6 +280,14 @@ DAG build(JobConf conf, TezWork work, Path scratchDir, // the name of the dag is what is displayed in the AM/Job UI DAG dag = DAG.create(work.getName()); + + // set some info for the query + JSONObject json = new JSONObject().put("context", "Hive").put("description", ctx.getCmd()); + String dagInfo = json.toString(); + + LOG.debug("DagInfo: " + dagInfo); + dag.setDAGInfo(dagInfo); + dag.setCredentials(conf.getCredentials()); setAccessControlsForCurrentUser(dag);