From dc0e636f630f8ab96d7cc71ede0a5a3ea5ce24a4 Mon Sep 17 00:00:00 2001 From: Alexander Pivovarov Date: Fri, 9 Sep 2016 20:32:28 -0700 Subject: [PATCH] [FLINK-4607] Close FileInputStream in ParameterTool and other --- .../flink/core/fs/local/LocalFileSystemTest.java | 6 +++--- .../apache/flink/api/java/utils/ParameterTool.java | 6 +++--- .../flink/api/java/utils/ParameterToolTest.java | 4 +++- .../flink/runtime/util/JarFileCreatorLambdaTest.java | 12 ++++++------ .../flink/runtime/util/JarFileCreatorTest.java | 12 ++++++------ 5 files changed, 21 insertions(+), 19 deletions(-) diff --git a/flink-core/src/test/java/org/apache/flink/core/fs/local/LocalFileSystemTest.java b/flink-core/src/test/java/org/apache/flink/core/fs/local/LocalFileSystemTest.java index 299524d50b5d5..d21e0f14be202 100644 --- a/flink-core/src/test/java/org/apache/flink/core/fs/local/LocalFileSystemTest.java +++ b/flink-core/src/test/java/org/apache/flink/core/fs/local/LocalFileSystemTest.java @@ -116,10 +116,10 @@ public void testLocalFilesystem() { assertEquals(testfile1.length(), 5L); - final FileInputStream fisfile1 = new FileInputStream(testfile1); byte[] testbytestest = new byte[5]; - assertEquals(testbytestest.length, fisfile1.read(testbytestest)); - fisfile1.close(); + try (FileInputStream fisfile1 = new FileInputStream(testfile1)) { + assertEquals(testbytestest.length, fisfile1.read(testbytestest)); + } assertArrayEquals(testbytes, testbytestest); diff --git a/flink-java/src/main/java/org/apache/flink/api/java/utils/ParameterTool.java b/flink-java/src/main/java/org/apache/flink/api/java/utils/ParameterTool.java index 38ae6df8ee73e..a9389a50daf20 100644 --- a/flink-java/src/main/java/org/apache/flink/api/java/utils/ParameterTool.java +++ b/flink-java/src/main/java/org/apache/flink/api/java/utils/ParameterTool.java @@ -156,9 +156,9 @@ public static ParameterTool fromPropertiesFile(String path) throws IOException { throw new FileNotFoundException("Properties file " + propertiesFile.getAbsolutePath() + " does not exist"); } Properties props = new Properties(); - FileInputStream fis = new FileInputStream(propertiesFile); - props.load(fis); - fis.close(); + try (FileInputStream fis = new FileInputStream(propertiesFile)) { + props.load(fis); + } return fromMap((Map)props); } diff --git a/flink-java/src/test/java/org/apache/flink/api/java/utils/ParameterToolTest.java b/flink-java/src/test/java/org/apache/flink/api/java/utils/ParameterToolTest.java index 09a778130b5d6..605f0333b13a8 100644 --- a/flink-java/src/test/java/org/apache/flink/api/java/utils/ParameterToolTest.java +++ b/flink-java/src/test/java/org/apache/flink/api/java/utils/ParameterToolTest.java @@ -185,7 +185,9 @@ private void validate(ParameterTool parameter) { String pathToFile = tmp.newFile().getAbsolutePath(); parameter.createPropertiesFile(pathToFile); Properties defaultProps = new Properties(); - defaultProps.load(new FileInputStream(pathToFile)); + try (FileInputStream fis = new FileInputStream(pathToFile)) { + defaultProps.load(fis); + } Assert.assertEquals("myDefaultValue", defaultProps.get("output")); Assert.assertEquals("-1", defaultProps.get("expectedCount")); diff --git a/flink-java8/src/test/java/org/apache/flink/runtime/util/JarFileCreatorLambdaTest.java b/flink-java8/src/test/java/org/apache/flink/runtime/util/JarFileCreatorLambdaTest.java index 249e082fa9f43..d90f096373db8 100644 --- a/flink-java8/src/test/java/org/apache/flink/runtime/util/JarFileCreatorLambdaTest.java +++ b/flink-java8/src/test/java/org/apache/flink/runtime/util/JarFileCreatorLambdaTest.java @@ -98,13 +98,13 @@ public void TestFilterFunctionOnLambda4() throws Exception { } public boolean validate(Set expected, File out) throws Exception { - - JarInputStream jis = new JarInputStream(new FileInputStream(out)); - ZipEntry ze; int count = expected.size(); - while ((ze = jis.getNextEntry()) != null) { - count--; - expected.remove(ze.getName()); + try (JarInputStream jis = new JarInputStream(new FileInputStream(out))) { + ZipEntry ze; + while ((ze = jis.getNextEntry()) != null) { + count--; + expected.remove(ze.getName()); + } } return count == 0 && expected.size() == 0; } diff --git a/flink-runtime/src/test/java/org/apache/flink/runtime/util/JarFileCreatorTest.java b/flink-runtime/src/test/java/org/apache/flink/runtime/util/JarFileCreatorTest.java index ba207eccf374b..8f8016e14c6fa 100644 --- a/flink-runtime/src/test/java/org/apache/flink/runtime/util/JarFileCreatorTest.java +++ b/flink-runtime/src/test/java/org/apache/flink/runtime/util/JarFileCreatorTest.java @@ -205,13 +205,13 @@ public void TestUDFPackage() throws IOException { } private boolean validate(Set expected, File out) throws IOException { - - JarInputStream jis = new JarInputStream(new FileInputStream(out)); - ZipEntry ze; int count = expected.size(); - while ((ze = jis.getNextEntry()) != null) { - count--; - expected.remove(ze.getName()); + try (JarInputStream jis = new JarInputStream(new FileInputStream(out))) { + ZipEntry ze; + while ((ze = jis.getNextEntry()) != null) { + count--; + expected.remove(ze.getName()); + } } return count == 0 && expected.size() == 0; }