123 lines
4.0 KiB
Java
123 lines
4.0 KiB
Java
package net.minemora.entitytrackerfixer;
|
|
|
|
import java.lang.reflect.Field;
|
|
import java.util.HashSet;
|
|
import java.util.Set;
|
|
|
|
import org.bukkit.Bukkit;
|
|
import org.bukkit.Location;
|
|
import org.bukkit.craftbukkit.v1_14_R1.CraftWorld;
|
|
import org.bukkit.scheduler.BukkitRunnable;
|
|
|
|
import net.minecraft.server.v1_14_R1.ChunkProviderServer;
|
|
import net.minecraft.server.v1_14_R1.EntityPlayer;
|
|
import net.minecraft.server.v1_14_R1.MinecraftServer;
|
|
import net.minecraft.server.v1_14_R1.WorldServer;
|
|
import net.minecraft.server.v1_14_R1.PlayerChunkMap.EntityTracker;
|
|
import net.minemora.entitytrackerfixer.config.ConfigMain;
|
|
import net.minemora.entitytrackerfixer.util.ReflectionUtils;
|
|
import net.minemora.entitytrackerfixer.util.Util;
|
|
|
|
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")
|
|
@Override
|
|
public void run() {
|
|
if(MinecraftServer.getServer().recentTps[0] > ConfigMain.getMinTps()) {
|
|
//String tps = String.format("%.2f", MinecraftServer.getServer().recentTps[0]);
|
|
//EntityTrackerFixer.plugin.getLogger().info("Not untraking because tps = " + tps);
|
|
return;
|
|
}
|
|
running = true;
|
|
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_R1.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_14_R1.Entity nmsEnt = (net.minecraft.server.v1_14_R1.Entity) trackerField.get(et);
|
|
if(nmsEnt instanceof EntityPlayer) {
|
|
continue;
|
|
}
|
|
if(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;
|
|
}
|
|
}
|
|
Location loc = nmsEnt.getBukkitEntity().getLocation();
|
|
if(!Util.isChunkLoaded(ws, loc.getBlockX() >> 4, loc.getBlockZ() >> 4)) {
|
|
UntrackedEntitiesCache.getInstance().addUFC(worldName, nmsEnt.getUniqueID());
|
|
}
|
|
if(remove) {
|
|
//System.out.println("untracked: " + nmsEnt.getBukkitEntity().getType().name());
|
|
toRemove.add(nmsEnt.getId());
|
|
removed++;
|
|
UntrackedEntitiesCache.getInstance().add(nmsEnt);
|
|
}
|
|
}
|
|
} catch (IllegalArgumentException | IllegalAccessException e) {
|
|
e.printStackTrace();
|
|
}
|
|
|
|
for(int id : toRemove) {
|
|
cps.playerChunkMap.trackedEntities.remove(id);
|
|
}
|
|
|
|
/*
|
|
new BukkitRunnable() {
|
|
@Override
|
|
public void run() {
|
|
NMSEntityTracker.untrackEntities(cps, toRemove);
|
|
}
|
|
}.runTask(EntityTrackerFixer.plugin);
|
|
*/
|
|
|
|
if(ConfigMain.isLogToConsole()) {
|
|
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;
|
|
}
|
|
|
|
}
|