using env HADOOP_HOME in heap dump script

mayoor · Mar 16, 2015 · a949576 · a949576
1 parent 006675e
commit a949576
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gobblin-utility/src/main/java/gobblin/util/HeapDumpForTaskUtils.java b/gobblin-utility/src/main/java/gobblin/util/HeapDumpForTaskUtils.java
@@ -65,7 +65,8 @@ public static void generateDumpScript(Path dumpScript, FileSystem fs, String hea
       BufferedWriter scriptWriter =
           closer.register(new BufferedWriter(new OutputStreamWriter(fs.create(dumpScript), Charset
               .forName(ConfigurationKeys.DEFAULT_CHARSET_ENCODING))));
-      scriptWriter.write("#!/bin/sh\n" + "hadoop dfs -put " + heapFileName + " " + dumpDir + "/${PWD//\\//_}.hprof");
+      scriptWriter.write("#!/bin/sh\n" + "${HADOOP_HOME}/bin/hadoop dfs -put " + heapFileName + " " + dumpDir
+          + "/${PWD//\\//_}.hprof");
     } catch (IOException ioe) {
       LOG.error("Heap dump script is not generated successfully.");
       if (fs.exists(dumpScript)) {