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 a2060da..e2d7fd4 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 @@ -236,7 +236,7 @@ public int execute(DriverContext driverContext) { ctx.clear(); } catch (Exception e) { /*best effort*/ - LOG.warn("Failed to clean up after tez job"); + LOG.warn("Failed to clean up after tez job", e); } } // need to either move tmp files or remove them @@ -498,6 +498,12 @@ int close(TezWork work, int rc) { console.printError(mesg, "\n" + StringUtils.stringifyException(e)); } } + try { + dagClient.close(); + dagClient = null; + } catch (Exception e) { + LOG.warn("Failed to close DagClient", e); + } return rc; } @@ -564,7 +570,13 @@ public void shutdown() { LOG.info("Waiting for Tez task to shut down: " + this); dagClient.waitForCompletion(); } catch (Exception ex) { - LOG.info("Failed to shut down TezTask" + this, ex); + LOG.warn("Failed to shut down TezTask" + this, ex); + } + try { + dagClient.close(); + dagClient = null; + } catch (Exception e) { + LOG.warn("Failed to close DagClient", e); } } }