优化性能

Glavo · Jun 17, 2021 · 380d0e8 · 380d0e8
1 parent fb6df48
commit 380d0e8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/src/main/java/org/glavo/checksum/Hasher.java b/src/main/java/org/glavo/checksum/Hasher.java
@@ -101,7 +101,7 @@ public int getHashStringLength() {
     public final String hashFile(Path file) throws IOException {
         MessageDigest md = localMessageDigest.get();
         byte[] buffer = localBuffer.get();
-        int read = 0;
+        int read;
         try (InputStream input = Files.newInputStream(file)) {
             do {
                 read = input.read(buffer);

diff --git a/src/main/java/org/glavo/checksum/Main.java b/src/main/java/org/glavo/checksum/Main.java
@@ -7,6 +7,7 @@
 import java.util.TreeMap;
 import java.util.concurrent.*;
 import java.util.concurrent.atomic.LongAdder;
+import java.util.stream.Collectors;
 
 public final class Main {
 
@@ -294,13 +295,14 @@ private static void verify(Path basePath, BufferedReader reader, Hasher hasher,
                     });
                 }
                 final Hasher finalHasher = hasher;
-                pool.submit(() -> reader.lines().parallel()
-                        .filter(line -> !line.isEmpty())
+                pool.submit(() -> reader.lines().collect(Collectors.toList()).stream().parallel()
                         .forEach(line -> {
-                            if (verifyFile(basePath, line, finalHasher)) {
-                                successCount.add(1);
-                            } else {
-                                failureCount.add(1);
+                            if (!line.isEmpty()) {
+                                if (verifyFile(basePath, line, finalHasher)) {
+                                    successCount.add(1);
+                                } else {
+                                    failureCount.add(1);
+                                }
                             }
                         })).get();