From cae3aa99fb4e9458482ea2e4a8465cc72032729c Mon Sep 17 00:00:00 2001
From: David Mollitor <dmollitor@apache.org>
Date: Sun, 24 Jan 2021 17:25:32 -0500
Subject: [PATCH] TEZ-4268: Buffer File Read of TEZ AM Local Resources PB File

---
 .../org/apache/tez/dag/app/DAGAppMaster.java     | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

diff --git a/tez-dag/src/main/java/org/apache/tez/dag/app/DAGAppMaster.java b/tez-dag/src/main/java/org/apache/tez/dag/app/DAGAppMaster.java
index cde77b3bf6..ba072a9b5e 100644
--- a/tez-dag/src/main/java/org/apache/tez/dag/app/DAGAppMaster.java
+++ b/tez-dag/src/main/java/org/apache/tez/dag/app/DAGAppMaster.java
@@ -20,6 +20,7 @@
 
 
 
+import java.io.BufferedInputStream;
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.FileNotFoundException;
@@ -604,17 +605,14 @@ public synchronized void serviceInit(final Configuration conf) throws Exception
 
     if (!versionMismatch) {
       if (isSession) {
-        FileInputStream sessionResourcesStream = null;
-        try {
-          sessionResourcesStream = new FileInputStream(
-              new File(workingDirectory, TezConstants.TEZ_AM_LOCAL_RESOURCES_PB_FILE_NAME));
+        try (BufferedInputStream sessionResourcesStream =
+            new BufferedInputStream(
+                new FileInputStream(new File(workingDirectory,
+                    TezConstants.TEZ_AM_LOCAL_RESOURCES_PB_FILE_NAME)))) {
           PlanLocalResourcesProto amLocalResourceProto = PlanLocalResourcesProto
               .parseDelimitedFrom(sessionResourcesStream);
-          amResources.putAll(DagTypeConverters.convertFromPlanLocalResources(amLocalResourceProto));
-        } finally {
-          if (sessionResourcesStream != null) {
-            sessionResourcesStream.close();
-          }
+          amResources.putAll(DagTypeConverters
+              .convertFromPlanLocalResources(amLocalResourceProto));
         }
       }
     }