Parallelize background file scan

neoforged · Feb 4, 2024 · 3c2df4d · 3c2df4d
1 parent 36f618c
commit 3c2df4d
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/loader/src/main/java/net/minecraftforge/fml/loading/moddiscovery/BackgroundScanHandler.java b/loader/src/main/java/net/minecraftforge/fml/loading/moddiscovery/BackgroundScanHandler.java
@@ -6,6 +6,7 @@
 package net.minecraftforge.fml.loading.moddiscovery;
 
 import com.mojang.logging.LogUtils;
+import net.minecraftforge.fml.loading.FMLConfig;
 import net.minecraftforge.fml.loading.ImmediateWindowHandler;
 import net.minecraftforge.fml.loading.LoadingModList;
 import net.minecraftforge.fml.loading.LogMarkers;
@@ -18,6 +19,7 @@
 import java.util.ArrayList;
 import java.util.List;
 import java.util.concurrent.*;
+import java.util.concurrent.atomic.AtomicInteger;
 
 public class BackgroundScanHandler
 {
@@ -41,10 +43,14 @@ private enum ScanStatus {
 
     public BackgroundScanHandler(final List<ModFile> modFiles) {
         this.modFiles = modFiles;
-        modContentScanner = Executors.newSingleThreadExecutor(r -> {
+        int maxThreads = FMLConfig.getIntConfigValue(FMLConfig.ConfigValue.MAX_THREADS);
+        // Leave 1 thread for Minecraft's own bootstrap
+        int poolSize = Math.max(1, maxThreads - 1);
+        AtomicInteger threadCount = new AtomicInteger();
+        modContentScanner = Executors.newFixedThreadPool(poolSize, r -> {
             final Thread thread = Executors.defaultThreadFactory().newThread(r);
             thread.setDaemon(true);
-            thread.setName("Background Scan Handler");
+            thread.setName("background-scan-handler-" + threadCount.getAndIncrement());
             return thread;
         });
         scannedFiles = new ArrayList<>();
@@ -72,7 +78,7 @@ public void submitForScanning(final ModFile file) {
         file.setFutureScanResult(future);
     }
 
-    private void addCompletedFile(final ModFile file, final ModFileScanData modFileScanData, final Throwable throwable) {
+    private synchronized void addCompletedFile(final ModFile file, final ModFileScanData modFileScanData, final Throwable throwable) {
         if (throwable != null) {
             status = ScanStatus.ERRORED;
             LOGGER.error(LogMarkers.SCAN,"An error occurred scanning file {}", file, throwable);