diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/main/java/org/apache/hadoop/yarn/state/Graph.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/main/java/org/apache/hadoop/yarn/state/Graph.java index a090194..f6d00e2 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/main/java/org/apache/hadoop/yarn/state/Graph.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/main/java/org/apache/hadoop/yarn/state/Graph.java @@ -190,8 +190,11 @@ public String generateGraphViz() { public void save(String filepath) throws IOException { OutputStreamWriter fout = new OutputStreamWriter( new FileOutputStream(filepath), Charset.forName("UTF-8")); - fout.write(generateGraphViz()); - fout.close(); + try { + fout.write(generateGraphViz()); + } finally { + fout.close(); + } } public static List combineEdges(List edges) {