124 lines
4.1 KiB
Java
124 lines
4.1 KiB
Java
package net.minemora.entitytrackerfixer.v1_16_R3.tasks;
|
|
|
|
import java.lang.reflect.Field;
|
|
import java.util.HashSet;
|
|
import java.util.Set;
|
|
|
|
import org.bukkit.Bukkit;
|
|
import org.bukkit.World;
|
|
import org.bukkit.craftbukkit.v1_16_R3.CraftWorld;
|
|
import org.bukkit.scheduler.BukkitRunnable;
|
|
|
|
import net.minecraft.server.v1_16_R3.ChunkProviderServer;
|
|
import net.minecraft.server.v1_16_R3.EntityArmorStand;
|
|
import net.minecraft.server.v1_16_R3.EntityComplexPart;
|
|
import net.minecraft.server.v1_16_R3.EntityEnderDragon;
|
|
import net.minecraft.server.v1_16_R3.EntityPlayer;
|
|
import net.minecraft.server.v1_16_R3.MinecraftServer;
|
|
import net.minecraft.server.v1_16_R3.WorldServer;
|
|
import net.minecraft.server.v1_16_R3.PlayerChunkMap.EntityTracker;
|
|
import net.minemora.entitytrackerfixer.EntityTrackerFixer;
|
|
import net.minemora.entitytrackerfixer.config.ConfigMain;
|
|
import net.minemora.entitytrackerfixer.util.ReflectionUtils;
|
|
import net.minemora.entitytrackerfixer.v1_16_R3.entityTick.EntityTickManager;
|
|
|
|
public class UntrackerTask extends BukkitRunnable {
|
|
|
|
private static boolean running = false;
|
|
|
|
private static Field trackerField;
|
|
|
|
static {
|
|
try {
|
|
trackerField = ReflectionUtils.getClassPrivateField(EntityTracker.class, "tracker");
|
|
} catch (NoSuchFieldException | SecurityException | IllegalArgumentException | IllegalAccessException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
@SuppressWarnings({ "deprecation", "resource" })
|
|
@Override
|
|
public void run() {
|
|
if(MinecraftServer.getServer().recentTps[0] > ConfigMain.getMinTps()) {
|
|
return;
|
|
}
|
|
running = true;
|
|
if(ConfigMain.isEnableOnAllWorlds()) {
|
|
for(World world : Bukkit.getWorlds()) {
|
|
untrackProcess(world.getName());
|
|
}
|
|
}
|
|
else {
|
|
for(String worldName : ConfigMain.getWorlds()) {
|
|
untrackProcess(worldName);
|
|
}
|
|
}
|
|
|
|
running = false;
|
|
}
|
|
|
|
private void untrackProcess(String worldName) {
|
|
if(Bukkit.getWorld(worldName) == null) {
|
|
return;
|
|
}
|
|
//Set<net.minecraft.server.v1_14_R2.Entity> toRemove = new HashSet<>();
|
|
Set<Integer> toRemove = new HashSet<>();
|
|
int removed = 0;
|
|
WorldServer ws = ((CraftWorld)Bukkit.getWorld(worldName)).getHandle();
|
|
ChunkProviderServer cps = ws.getChunkProvider();
|
|
|
|
try {
|
|
for(EntityTracker et : cps.playerChunkMap.trackedEntities.values()) {
|
|
net.minecraft.server.v1_16_R3.Entity nmsEnt = (net.minecraft.server.v1_16_R3.Entity) trackerField.get(et);
|
|
if(nmsEnt instanceof EntityPlayer || nmsEnt instanceof EntityEnderDragon || nmsEnt instanceof EntityComplexPart) {
|
|
continue;
|
|
}
|
|
if(nmsEnt instanceof EntityArmorStand && nmsEnt.getBukkitEntity().getCustomName() != null) {
|
|
continue;
|
|
}
|
|
boolean remove = false;
|
|
if(et.trackedPlayers.size() == 0) {
|
|
remove = true;
|
|
}
|
|
else if(et.trackedPlayers.size() == 1) {
|
|
for(EntityPlayer ep : et.trackedPlayers) {
|
|
if(!ep.getBukkitEntity().isOnline()) {
|
|
remove = true;
|
|
}
|
|
}
|
|
if(!remove) {
|
|
continue;
|
|
}
|
|
}
|
|
if(remove) {
|
|
//System.out.println("untracked: " + nmsEnt.getBukkitEntity().getType().name());
|
|
toRemove.add(nmsEnt.getId());
|
|
removed++;
|
|
}
|
|
}
|
|
} catch (IllegalArgumentException | IllegalAccessException e) {
|
|
e.printStackTrace();
|
|
}
|
|
|
|
for(int id : toRemove) {
|
|
cps.playerChunkMap.trackedEntities.remove(id);
|
|
if(ConfigMain.isDisableTickUntracked()) {
|
|
EntityTickManager.getInstance().disableTicking(ws.entitiesById.get(id));
|
|
}
|
|
}
|
|
|
|
if(ConfigMain.isLogToConsole()) {
|
|
if(removed > 0) {
|
|
EntityTrackerFixer.plugin.getLogger().info("Untracked " + removed + " entities in " + worldName);
|
|
}
|
|
}
|
|
|
|
//System.out.println("cache now contains " + UntrackedEntitiesCache.getInstance().getCache(worldName).size() + " entities");
|
|
}
|
|
|
|
public static boolean isRunning() {
|
|
return running;
|
|
}
|
|
|
|
}
|