package dev.modzuozhi.mixin;

import dev.modzuozhi.core.compat.PayloadMainThreadDispatch;
import net.minecraft.server.MinecraftServer;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.injection.*;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
import java.util.function.BooleanSupplier;

@Mixin(MinecraftServer.class)
public abstract class MinecraftServerPayloadMixin {
    @Inject(method = "tickServer", at = @At("HEAD"))
    private void dimensionalripper$drainPayloads(BooleanSupplier budget, CallbackInfo ci) {
        PayloadMainThreadDispatch.drain((MinecraftServer) (Object) this);
    }
}