Toggleable lag generation.

Author: noobanidus

common/src/main/java/noobanidus/mods/laggenerator/common/mixins/AccessorMixinGameRules$BooleanValue.java

@@ -0,0 +1,17 @@
+package noobanidus.mods.laggenerator.common.mixins;
+
+import net.minecraft.server.MinecraftServer;
+import net.minecraft.world.level.GameRules;
+import org.apache.commons.lang3.NotImplementedException;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.gen.Invoker;
+
+import java.util.function.BiConsumer;
+
+@Mixin(GameRules.BooleanValue.class)
+public interface AccessorMixinGameRules$BooleanValue {
+  @Invoker("create")
+  static GameRules.Type<GameRules.BooleanValue> LagGenerator$invokeCreate(boolean defaultValue, BiConsumer<MinecraftServer, GameRules.BooleanValue> changeListener) {
+    throw new NotImplementedException();
+  }
+}

common/src/main/java/noobanidus/mods/laggenerator/common/mixins/MixinEntity.java

@@ -18,6 +18,9 @@ public class MixinEntity {
     if (level == null || level.isClientSide() || level.getServer() == null) {
       return;
     }
+    if (!LagGameRules.enabled(level.getServer())) {
+      return;
+    }
     int entityTickLag = LagGameRules.getValue(level.getServer(), LagGameRules.ENTITY_TICK_LAG);
     if (entityTickLag > 0) {
       try {

common/src/main/java/noobanidus/mods/laggenerator/common/mixins/MixinMinecraftServer.java

@@ -14,6 +14,9 @@ public class MixinMinecraftServer {
   @WrapOperation(method = "tickServer", at = @At(value = "INVOKE", target = "Lnet/minecraft/server/ServerTickRateManager;tick()V"))
   private void LagGenerator$PreServerTickLag(ServerTickRateManager instance, Operation<Void> original) {
     original.call(instance);
+    if (!LagGameRules.enabled((MinecraftServer) (Object) this)) {
+      return;
+    }
     int preTickLag = LagGameRules.getValue((MinecraftServer) (Object) this, LagGameRules.SERVER_PRE_TICK_LAG);
     if (preTickLag > 0) {
       try {
@@ -26,6 +29,9 @@ public class MixinMinecraftServer {
   @WrapOperation(method = "tickServer", at = @At(value = "INVOKE", target = "Lnet/minecraft/util/profiling/ProfilerFiller;pop()V", ordinal = 1))
   private void LagGenerator$PostServerTickLag(ProfilerFiller instance, Operation<Void> original) {
     original.call(instance);
+    if (!LagGameRules.enabled((MinecraftServer) (Object) this)) {
+      return;
+    }
     int postTickLag = LagGameRules.getValue((MinecraftServer) (Object) this, LagGameRules.SERVER_POST_TICK_LAG);
     if (postTickLag > 0) {
       try {

common/src/main/java/noobanidus/mods/laggenerator/common/rules/LagGameRules.java

@@ -3,9 +3,11 @@
 import net.minecraft.server.MinecraftServer;
 import net.minecraft.world.level.GameRules;
 import noobanidus.mods.laggenerator.common.mixins.AccessorMixinGameRules;
+import noobanidus.mods.laggenerator.common.mixins.AccessorMixinGameRules$BooleanValue;
 import noobanidus.mods.laggenerator.common.mixins.AccessorMixinGameRules$IntegerValue;
 
 public class LagGameRules {
+  public static GameRules.Key<GameRules.BooleanValue> ENABLE_LAG_GENERATOR;
   public static GameRules.Key<GameRules.IntegerValue> SERVER_PRE_TICK_LAG;
   public static GameRules.Key<GameRules.IntegerValue> SERVER_POST_TICK_LAG;
   public static GameRules.Key<GameRules.IntegerValue> ENTITY_TICK_LAG;
@@ -14,9 +16,14 @@ public static int getValue (MinecraftServer server, GameRules.Key<GameRules.Inte
     return server.getGameRules().getRule(key).get();
   }
 
+  public static boolean enabled (MinecraftServer server) {
+    return server.getGameRules().getRule(ENABLE_LAG_GENERATOR).get();
+  }
+
   public static void init () {
     SERVER_PRE_TICK_LAG = AccessorMixinGameRules.LagGenerator$invokeRegister("lag_generator_serverPreTickLag", GameRules.Category.UPDATES, AccessorMixinGameRules$IntegerValue.LagGenerator$invokeCreate(80, 0, Integer.MAX_VALUE, (a, b) -> {}));
     SERVER_POST_TICK_LAG = AccessorMixinGameRules.LagGenerator$invokeRegister("lag_generator_serverPostTickLag", GameRules.Category.UPDATES, AccessorMixinGameRules$IntegerValue.LagGenerator$invokeCreate(80, 0, Integer.MAX_VALUE, (a, b) -> {}));
-    ENTITY_TICK_LAG = AccessorMixinGameRules.LagGenerator$invokeRegister("lag_generator_entityTickLag", GameRules.Category.UPDATES, AccessorMixinGameRules$IntegerValue.LagGenerator$invokeCreate(80, 0, Integer.MAX_VALUE, (a, b) -> {}));
+    ENTITY_TICK_LAG = AccessorMixinGameRules.LagGenerator$invokeRegister("lag_generator_entityTickLag", GameRules.Category.UPDATES, AccessorMixinGameRules$IntegerValue.LagGenerator$invokeCreate(10, 0, Integer.MAX_VALUE, (a, b) -> {}));
+    ENABLE_LAG_GENERATOR = AccessorMixinGameRules.LagGenerator$invokeRegister("enable_lag_generator", GameRules.Category.UPDATES, AccessorMixinGameRules$BooleanValue.LagGenerator$invokeCreate(false, (a, b) -> {}));
   }
 }

common/src/main/resources/laggenerator-common.mixins.json

@@ -4,6 +4,7 @@
   "compatibilityLevel": "JAVA_17",
   "mixins": [
     "AccessorMixinGameRules",
+    "AccessorMixinGameRules$BooleanValue",
     "AccessorMixinGameRules$IntegerValue",
     "MixinEntity",
     "MixinMinecraftServer"