package com.glitchclient.performance; import com.glitchclient.GlitchClient; import com.glitchclient.config.GlitchConfigManager; import java.util.ArrayList; import java.util.Collection; import java.util.List; import java.util.concurrent.Callable; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import java.util.concurrent.Future; import java.util.concurrent.TimeUnit; import java.util.function.Consumer; public class MultiCoreTaskDistributor { private static ExecutorService pool; private static int threadCount; private static long tasksSubmitted = 0L; private static long tasksCompleted = 0L; public static void initialize() { if (!GlitchConfigManager.get().performance.multiCoreEntityDistributor) { GlitchClient.LOGGER.info("[GlitchClient] MultiCoreTaskDistributor: disabled by config."); return; } int cfg = GlitchConfigManager.get().performance.cpuCores; threadCount = Math.max(1, cfg > 0 ? cfg : (Runtime.getRuntime().availableProcessors() - 1)); pool = Executors.newFixedThreadPool(threadCount, r -> { Thread t = new Thread(r, "glitch-distributor-" + System.nanoTime()); t.setDaemon(true); t.setPriority(Thread.NORM_PRIORITY - 1); return t; }); GlitchClient.LOGGER.info("[GlitchClient] MultiCoreTaskDistributor: started with {} threads.", threadCount); } public static List> submitBatch(Collection> tasks) { if (pool == null || pool.isShutdown()) return List.of(); List> futures = new ArrayList<>(tasks.size()); for (Callable task : tasks) { futures.add(pool.submit(task)); tasksSubmitted++; } return futures; } public static void distributeWork(List items, Consumer worker) { if (pool == null || pool.isShutdown() || items.isEmpty()) { items.forEach(worker); return; } int chunkSize = Math.max(1, items.size() / threadCount); List> futures = new ArrayList<>(); for (int start = 0; start < items.size(); start += chunkSize) { int end = Math.min(start + chunkSize, items.size()); List slice = items.subList(start, end); futures.add(pool.submit(() -> { slice.forEach(worker); tasksCompleted += slice.size(); return null; })); } for (Future f : futures) { try { f.get(200, TimeUnit.MILLISECONDS); } catch (Exception e) { GlitchClient.LOGGER.warn("[GlitchClient] Distributor task timeout/error: {}", e.getMessage()); } } } public static void reconfigure(int cores) { shutdown(); threadCount = Math.max(1, cores); GlitchConfigManager.get().performance.cpuCores = threadCount; GlitchConfigManager.save(); initialize(); } public static void shutdown() { if (pool != null && !pool.isShutdown()) { pool.shutdown(); try { if (!pool.awaitTermination(1, TimeUnit.SECONDS)) { pool.shutdownNow(); } } catch (InterruptedException e) { pool.shutdownNow(); Thread.currentThread().interrupt(); } } } public static int getThreadCount() { return threadCount; } public static long getTasksSubmitted() { return tasksSubmitted; } public static long getTasksCompleted() { return tasksCompleted; } public static boolean isActive() { return pool != null && !pool.isShutdown(); } }