Improve packet warnings

GregTechCEu · Nov 10, 2023 · 3a47cda · 3a47cda
1 parent 08c7f57
commit 3a47cda
Showing 1 changed file with 20 additions and 2 deletions.
diff --git a/src/main/java/gregtech/api/metatileentity/SyncedTileEntityBase.java b/src/main/java/gregtech/api/metatileentity/SyncedTileEntityBase.java
@@ -2,6 +2,7 @@
 
 import gregtech.api.block.BlockStateTileEntity;
 import gregtech.api.metatileentity.interfaces.ISyncedTileEntity;
+import gregtech.api.metatileentity.interfaces.IGregTechTileEntity;
 import gregtech.api.network.PacketDataList;
 import gregtech.api.util.GTLog;
 import io.netty.buffer.ByteBuf;
@@ -72,8 +73,16 @@ public final void onDataPacket(@NotNull NetworkManager net, @NotNull SPacketUpda
                 ByteBuf backedBuffer = Unpooled.copiedBuffer(entryTag.getByteArray(discriminatorKey));
                 receiveCustomData(Integer.parseInt(discriminatorKey), new PacketBuffer(backedBuffer));
                 if (backedBuffer.readableBytes() != 0) {
+                    String className = null;
+                    if (this instanceof IGregTechTileEntity gtte) {
+                        MetaTileEntity mte = gtte.getMetaTileEntity();
+                        if (mte != null) className = mte.getClass().getName();
+                    }
+                    if (className == null) {
+                        className = this.getClass().getName();
+                    }
                     GTLog.logger.error("Class {} failed to finish reading receiveCustomData with discriminator {} and {} bytes remaining",
-                            getClass().getName(), discriminatorKey, backedBuffer.readableBytes());
+                            className, discriminatorKey, backedBuffer.readableBytes());
                 }
             }
         }
@@ -96,8 +105,17 @@ public final void handleUpdateTag(@NotNull NBTTagCompound tag) {
         ByteBuf backedBuffer = Unpooled.copiedBuffer(updateData);
         receiveInitialSyncData(new PacketBuffer(backedBuffer));
         if (backedBuffer.readableBytes() != 0) {
+            String className = null;
+            if (this instanceof IGregTechTileEntity gtte) {
+                MetaTileEntity mte = gtte.getMetaTileEntity();
+                if (mte != null) className = mte.getClass().getName();
+            }
+            if (className == null) {
+                className = this.getClass().getName();
+            }
+
             GTLog.logger.error("Class {} failed to finish reading initialSyncData with {} bytes remaining",
-                    getClass().getName(), backedBuffer.readableBytes());
+                    className, backedBuffer.readableBytes());
         }
     }
 }