Threading

src/main/java/org/ecorous/esnesnon/mojmap_patcher/MojMapPatcher.java

@@ -3,7 +3,10 @@ package org.ecorous.esnesnon.mojmap_patcher;
 import java.io.IOException;
 import java.nio.file.*;
 import java.nio.file.attribute.BasicFileAttributes;
+import java.util.ArrayList;
+import java.util.List;
 import java.util.Map;
+import java.util.concurrent.*;
 
 import org.ecorous.esnesnon.mojmap_patcher.api.Mapper;
 import org.ecorous.esnesnon.mojmap_patcher.api.data.MappingData;
@@ -15,19 +18,27 @@ import org.objectweb.asm.commons.ClassRemapper;
 
 public class MojMapPatcher {
 
-	public static void run(String minecraftVersion, Path inputJar, Path outputJar) throws IOException {
+    public static void run(String minecraftVersion, Path inputJar, Path outputJar) throws IOException, InterruptedException {
         MappingData data = ProguardParser.read(MojmapProvider.get(minecraftVersion).orElseThrow()).reverse();
 
         Mapper mapper = new Mapper(data);
 
         Files.deleteIfExists(outputJar);
 
-		try (FileSystem inFs = FileSystems.newFileSystem(inputJar);
+        System.out.println("Remapping...");
+
+        long startTime = System.currentTimeMillis();
+
+        try (ExecutorService exec = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());
+             FileSystem inFs = FileSystems.newFileSystem(inputJar);
              FileSystem outFs = FileSystems.newFileSystem(outputJar, Map.of("create", "true"))) {
+            List<Callable<Void>> tasks = new ArrayList<>();
+
             Files.walkFileTree(inFs.getPath("/"), new SimpleFileVisitor<>() {
                 @Override
                 public FileVisitResult visitFile(Path path, BasicFileAttributes basicFileAttributes) throws IOException {
-
+                    tasks.add(() -> {
+                        try {
                             if (path.getFileName().toString().endsWith(".class")) {
                                 String className = path.getFileName().toString().substring(0, path.getFileName().toString().lastIndexOf("."));
                                 Path result = outFs.getPath(path.toString()).resolveSibling(data.classes().getOrDefault(className, className) + ".class");
@@ -44,14 +55,24 @@ public class MojMapPatcher {
                                 Files.createDirectories(result.getParent());
                                 Files.copy(path, result);
                             }
+                        } catch (IOException ex) {
+                            throw new Error(ex);
+                        }
+
+                        return null;
+                    });
 
                     return FileVisitResult.CONTINUE;
                 }
             });
+
+            exec.invokeAll(tasks);
+
+            System.out.printf("Finished remapping (%.2fs)\n", (System.currentTimeMillis() - startTime) / 1000F);
         }
     }
 
-	public static void main(String[] args) throws IOException {
+    public static void main(String[] args) throws IOException, InterruptedException {
         // temporary
         if (args.length != 3) {
             System.err.println("Usage: [minecraft-version] [minecraft.jar] [out.jar]");