Improve getEntityFromID (#1740)

src/main/java/com/comphenix/protocol/injector/EntityUtilities.java

@@ -28,6 +28,7 @@ import com.comphenix.protocol.utility.MinecraftFields;
 import com.comphenix.protocol.utility.MinecraftReflection;
 import com.comphenix.protocol.utility.MinecraftVersion;
 import com.comphenix.protocol.wrappers.WrappedIntHashMap;
+import java.lang.reflect.Method;
 import java.lang.reflect.Modifier;
 import java.util.ArrayList;
 import java.util.Collection;
@@ -56,6 +57,7 @@ class EntityUtilities {
 	private FieldAccessor trackedPlayersField;
 	private FieldAccessor trackedEntitiesField;
 
+	private MethodAccessor getEntity;
 	private MethodAccessor getChunkProvider;
 
 	private EntityUtilities() {
@@ -83,6 +85,20 @@ class EntityUtilities {
 				.invoke(trackerEntry, nmsPlayers);
 	}
 
+	public Entity getEntity(World world, int id) {
+		Object level = BukkitUnwrapper.getInstance().unwrapItem(world);
+		if (getEntity == null) {
+			Method entityGetter = FuzzyReflection.fromObject(level).getMethodByParameters(
+					"getEntity",
+					MinecraftReflection.getEntityClass(),
+					new Class[]{int.class});
+			getEntity = Accessors.getMethodAccessor(entityGetter);
+		}
+
+		Object entity = getEntity.invoke(level, id);
+		return (Entity) MinecraftReflection.getBukkitEntity(entity);
+	}
+
 	private MethodAccessor findScanPlayers(Class<?> trackerClass) {
 		MethodAccessor candidate = Accessors.getMethodAcccessorOrNull(trackerClass, "scanPlayers");
 		if (candidate != null) {

src/main/java/com/comphenix/protocol/injector/PacketFilterManager.java

@@ -401,12 +401,7 @@ public class PacketFilterManager implements ListenerInvoker, InternalManager {
 
 	@Override
 	public Entity getEntityFromID(World container, int id) {
-		for (Entity entity : container.getEntities()) {
+		return EntityUtilities.getInstance().getEntity(container, id);
-			if (entity.getEntityId() == id) {
-				return entity;
-			}
-		}
-		return null;
 	}
 
 	@Override