top.hendrixshen.magiclib.impl.dev.threadtweak.ThreadTweaker Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of magiclib-better-dev-1.20.6-fabric Show documentation
Development Environment runtime QOL.
The newest version!
/*
 * MIT License
 *
 * Copyright 2022 Steven Cao
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

package top.hendrixshen.magiclib.impl.dev.threadtweak;

import org.jetbrains.annotations.NotNull;
import top.hendrixshen.magiclib.MagicLib;
import top.hendrixshen.magiclib.MagicLibProperties;
import top.hendrixshen.magiclib.mixin.dev.accessor.UtilAccessor;

import java.util.concurrent.*;
import java.util.concurrent.atomic.AtomicInteger;
import net.minecraft.class_148;
import net.minecraft.class_2966;
import net.minecraft.class_3532;

//#if MC > 12006
//$$ import net.minecraft.ReportType;
//#endif

/**
 * Reference to SmoothBoot
 */
public class ThreadTweaker {
    public static class Default {
        public static final int bootstrapCount = 1;
        public static final int mainCount = MagicLibProperties.getMaxBackgroundThreads();
        public static final int bootstrapPriority = 1;
        public static final int gamePriority = 5;
        public static final int integratedServerPriority = 5;
        public static final int ioPriority = 1;
        public static final int mainPriority = 1;
    }

    public static @NotNull ExecutorService replaceForkJoinWorker(String name, int priority, int counter) {
        AtomicInteger atomicInteger = new AtomicInteger(1);
        return new ForkJoinPool(class_3532.method_15340(counter, 1, 0x7fff), (forkJoinPool) -> {
            //#if MC > 11502
            String workerName = "Worker-" + name + "-" + atomicInteger.getAndIncrement();
            //#else
            //$$ String workerName = "Server-Worker-" + name + "-" + atomicInteger.getAndIncrement();
            //#endif
            MagicLib.getLogger().debug("Initialized " + workerName);
            ForkJoinWorkerThread forkJoinWorkerThread = new LoggingForkJoinWorkerThread(forkJoinPool,
                    UtilAccessor.magiclib$getLogger());
            forkJoinWorkerThread.setPriority(priority);
            forkJoinWorkerThread.setName(workerName);
            return forkJoinWorkerThread;
        }, (thread, throwable) -> {
            if (throwable instanceof CompletionException) {
                throwable = throwable.getCause();
            }

            if (throwable instanceof class_148) {
                class_2966.method_12847(((class_148) throwable).method_631().method_568(
                        //#if MC > 12006
                        //$$ ReportType.CRASH
                        //#endif
                ));
                System.exit(-1);
            }

            UtilAccessor.magiclib$getLogger().error(String.format("Caught exception in thread %s", thread), throwable);
        }, true);
    }

    public static @NotNull ExecutorService replaceThreadWorker(
            String name, int priority, Thread.UncaughtExceptionHandler uncaughtExceptionHandler) {
        AtomicInteger atomicInteger = new AtomicInteger(1);
        return Executors.newCachedThreadPool((runnable) -> {
            String workerName = name + "-" + atomicInteger.getAndIncrement();
            MagicLib.getLogger().debug("Initialized " + workerName);
            Thread thread = new Thread(runnable);
            thread.setName(workerName);
            thread.setDaemon(true);
            thread.setPriority(priority);
            thread.setUncaughtExceptionHandler(uncaughtExceptionHandler);
            return thread;
        });
    }

    public static int getBootstrapCount() {
        try {
            int i = Integer.parseInt(MagicLibProperties.DEV_QOL_THREAD_TWEAK_COUNT_BOOTSTRAP.getStringValue());
            return Math.max(i, 1);
        } catch (NumberFormatException e) {
            return Default.bootstrapCount;
        }
    }

    public static int getMainCount() {
        try {
            int i = Integer.parseInt(MagicLibProperties.DEV_QOL_THREAD_TWEAK_COUNT_MAIN.getStringValue());
            return Math.max(i, 1);
        } catch (NumberFormatException e) {
            return Default.mainCount;
        }
    }

    public static int getBootstrapPriority() {
        try {
            int i = Integer.parseInt(MagicLibProperties.DEV_QOL_THREAD_TWEAK_PRIORITY_BOOTSTRAP.getStringValue());
            return class_3532.method_15340(i, 1, 10);
        } catch (NumberFormatException e) {
            return Default.bootstrapPriority;
        }
    }

    public static int getGamePriority() {
        try {
            int i = Integer.parseInt(MagicLibProperties.DEV_QOL_THREAD_TWEAK_PRIORITY_GAME.getStringValue());
            return class_3532.method_15340(i, 1, 10);
        } catch (NumberFormatException e) {
            return Default.gamePriority;
        }
    }

    public static int getIntegratedServerPriority() {
        try {
            int i = Integer.parseInt(MagicLibProperties.DEV_QOL_THREAD_TWEAK_PRIORITY_INTEGRATED_SERVER
                    .getStringValue());
            return class_3532.method_15340(i, 1, 10);
        } catch (NumberFormatException e) {
            return Default.integratedServerPriority;
        }
    }

    public static int getIOPriority() {
        try {
            int i = Integer.parseInt(MagicLibProperties.DEV_QOL_THREAD_TWEAK_PRIORITY_IO.getStringValue());
            return class_3532.method_15340(i, 1, 10);
        } catch (NumberFormatException e) {
            return Default.ioPriority;
        }
    }

    public static int getMainPriority() {
        try {
            int i = Integer.parseInt(MagicLibProperties.DEV_QOL_THREAD_TWEAK_PRIORITY_MAIN.getStringValue());
            return class_3532.method_15340(i, 1, 10);
        } catch (NumberFormatException e) {
            return Default.mainPriority;
        }
    }
}