From 91a3f04bd855aca3a4436014d0c08e9441682da9 Mon Sep 17 00:00:00 2001
From: zimzaza4 <3625282098@qq.com>
Date: Sun, 7 Jul 2024 22:29:41 +0800
Subject: [PATCH] oof

---
 .../java/me/zimzaza4/geyserutils/geyser/GeyserUtils.java   | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/geyser/src/main/java/me/zimzaza4/geyserutils/geyser/GeyserUtils.java b/geyser/src/main/java/me/zimzaza4/geyserutils/geyser/GeyserUtils.java
index 3cf01fa..698b951 100644
--- a/geyser/src/main/java/me/zimzaza4/geyserutils/geyser/GeyserUtils.java
+++ b/geyser/src/main/java/me/zimzaza4/geyserutils/geyser/GeyserUtils.java
@@ -427,6 +427,13 @@ public class GeyserUtils implements Extension {
                 }
             }
             session.sendUpstreamPacket(animateEntityPacket);
+        } else if (customPacket instanceof CustomEntityPacket customEntityPacket) {
+            if (!LOADED_ENTITY_DEFINITIONS.containsKey(customEntityPacket.getIdentifier())) {
+                return;
+            }
+
+            Cache<Integer, String> cache = CUSTOM_ENTITIES.get(session);
+            cache.put(customEntityPacket.getEntityId(), customEntityPacket.getIdentifier());
         } else if (customPacket instanceof CameraInstructionCustomPayloadPacket cameraInstructionPacket) {
             if (cameraInstructionPacket.getInstruction() instanceof SetInstruction instruction) {
                 session.camera().sendCameraPosition(Converter.serializeSetInstruction(instruction));