fix: throw EntryMissingException for null entries in hash palette (#497)

CaffeineMC · Mar 18, 2024 · 25304ec · 25304ec
1 parent 3a32d72
commit 25304ec
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/main/java/me/jellysquid/mods/lithium/common/world/chunk/LithiumHashPalette.java b/src/main/java/me/jellysquid/mods/lithium/common/world/chunk/LithiumHashPalette.java
@@ -7,6 +7,7 @@
 import net.minecraft.network.PacketByteBuf;
 import net.minecraft.network.encoding.VarInts;
 import net.minecraft.util.collection.IndexedIterable;
+import net.minecraft.world.chunk.EntryMissingException;
 import net.minecraft.world.chunk.Palette;
 import net.minecraft.world.chunk.PaletteResizeListener;
 
@@ -120,11 +121,16 @@ private void resize(int neededCapacity) {
     public T get(int id) {
         T[] entries = this.entries;
 
+        T entry = null;
         if (id >= 0 && id < entries.length) {
-            return entries[id];
+            entry = entries[id];
         }
 
-        return null;
+        if (entry != null) {
+            return entry;
+        } else {
+            throw new EntryMissingException(id);
+        }
     }
 
     @Override