#146: LengthOfInput closes Input

yegor256 · Jun 15, 2017 · da2b7e3 · da2b7e3
1 parent c952ae6
commit da2b7e3
Showing 1 changed file with 12 additions and 11 deletions.
diff --git a/src/main/java/org/cactoos/io/LengthOfInput.java b/src/main/java/org/cactoos/io/LengthOfInput.java
@@ -70,19 +70,20 @@ public LengthOfInput(final Input input, final int max) {
 
     @Override
     public Long asValue() throws IOException {
-        final InputStream stream = this.source.stream();
-        final byte[] buf = new byte[this.size];
-        long length = 0L;
-        while (true) {
-            final int len = stream.read(buf);
-            if (len > 0) {
-                length += (long) len;
-            }
-            if (len != buf.length) {
-                break;
+        try (final InputStream stream = this.source.stream()) {
+            final byte[] buf = new byte[this.size];
+            long length = 0L;
+            while (true) {
+                final int len = stream.read(buf);
+                if (len > 0) {
+                    length += (long) len;
+                }
+                if (len != buf.length) {
+                    break;
+                }
             }
+            return length;
         }
-        return length;
     }
 
 }