Initial commit

src/main/java/eu/mhsl/minenet/minigames/util/BatchUtil.java

@@ -0,0 +1,32 @@
+package eu.mhsl.minenet.minigames.util;
+
+import it.unimi.dsi.fastutil.longs.Long2ObjectMap;
+import net.minestom.server.instance.InstanceContainer;
+import net.minestom.server.instance.batch.AbsoluteBlockBatch;
+import net.minestom.server.instance.batch.ChunkBatch;
+import net.minestom.server.utils.chunk.ChunkUtils;
+
+import java.lang.reflect.Field;
+
+public class BatchUtil {
+    public static long[] getAffectedChunks(AbsoluteBlockBatch batch) {
+        try {
+            Field field = batch.getClass().getDeclaredField("chunkBatchesMap");
+            field.setAccessible(true);
+
+            @SuppressWarnings("unchecked")
+            Long2ObjectMap<ChunkBatch> chunkBatchesMap = (Long2ObjectMap<ChunkBatch>) field.get(batch);
+
+            return chunkBatchesMap.keySet().toLongArray();
+        } catch (NoSuchFieldException | IllegalAccessException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    public static void loadAndApplyBatch(AbsoluteBlockBatch batch, InstanceContainer instance, Runnable onFinish) {
+        batch.awaitReady();
+        long[] affectedChunks = BatchUtil.getAffectedChunks(batch);
+        ChunkUtils.optionalLoadAll(instance, affectedChunks, null).thenRun(() -> batch.apply(instance, onFinish));
+    }
+
+}