zhdev/griefus
1
0
Fork 0
Code Issues Pull requests Activity

Significantly improved performance of hopper logging (fixes #636)

Browse source
This commit is contained in:
Intelli 2025-03-10 17:42:43 -06:00
parent 3038e96c6e
commit 38ba3abc9a
2 changed files with 150 additions and 70 deletions
Download patch file Download diff file Expand all files Collapse all files

196
src/main/java/net/coreprotect/listener/player/HopperPushListener.java
View file

@ -4,6 +4,9 @@ import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicInteger;
import org.bukkit.Location;
import org.bukkit.inventory.Inventory;
@ -18,6 +21,13 @@ import net.coreprotect.utility.ItemUtils;
public final class HopperPushListener {
private static final ConcurrentLinkedQueue<Object[]> hopperQueue = new ConcurrentLinkedQueue<>();
private static final AtomicBoolean processorRunning = new AtomicBoolean(false);
private static final int BATCH_SIZE = 100;
private static final int PROCESS_DELAY_TICKS = 1;
private static final int MAX_CONCURRENT_PROCESSORS = 4;
private static final AtomicInteger activeProcessors = new AtomicInteger(0);
static void processHopperPush(Location location, String user, InventoryHolder sourceHolder, InventoryHolder destinationHolder, ItemStack item) {
Location destinationLocation = destinationHolder.getInventory().getLocation();
if (destinationLocation == null) {
@ -36,73 +46,127 @@ public final class HopperPushListener {
ItemStack[] destinationContainer = ItemUtils.getContainerState(destinationHolder.getInventory().getContents());
ItemStack movedItem = item.clone();
final long taskStarted = InventoryChangeListener.tasksStarted.incrementAndGet();
Scheduler.runTaskAsynchronously(CoreProtect.getInstance(), () -> {
try {
if (sourceHolder == null || destinationHolder == null) {
return;
}
hopperQueue.add(new Object[] { location, user, sourceHolder, destinationHolder, movedItem, destinationContainer, loggingChestId, lastAbort });
boolean abort = false;
boolean addedInventory = ItemUtils.canAddContainer(destinationContainer, movedItem, destinationHolder.getInventory().getMaxStackSize());
if (!addedInventory) {
abort = true;
}
if (abort) {
Set<ItemStack> movedItems = new HashSet<>();
ItemStack[] destinationContents = destinationHolder.getInventory().getContents();
if (lastAbort != null && Arrays.equals(destinationContents, (ItemStack[]) lastAbort[1])) {
((Set<?>) lastAbort[0]).forEach(itemStack -> movedItems.add((ItemStack) itemStack));
}
movedItems.add(movedItem);
ConfigHandler.hopperAbort.put(loggingChestId, new Object[] { movedItems, ItemUtils.getContainerState(destinationContents) });
return;
}
else {
ConfigHandler.hopperSuccess.put(loggingChestId, new Object[] { destinationContainer, movedItem });
}
List<Object> list = ConfigHandler.transactingChest.get(location.getWorld().getUID().toString() + "." + location.getBlockX() + "." + location.getBlockY() + "." + location.getBlockZ());
if (list != null) {
list.add(movedItem);
}
if (Config.getConfig(location.getWorld()).HOPPER_FILTER_META && !movedItem.hasItemMeta()) {
return;
}
Inventory destinationInventory = destinationHolder.getInventory();
ItemStack[] originalDestination = destinationInventory.getContents().clone();
int removeAmount = movedItem.getAmount();
for (int i = 0; i < originalDestination.length; i++) {
if (removeAmount == 0) {
break;
}
ItemStack itemStack = (originalDestination[i] != null ? originalDestination[i].clone() : null);
if (itemStack != null && itemStack.isSimilar(movedItem)) {
if (itemStack.getAmount() >= removeAmount) {
itemStack.setAmount(itemStack.getAmount() - removeAmount);
removeAmount = 0;
}
else {
removeAmount = removeAmount - itemStack.getAmount();
itemStack = null;
}
originalDestination[i] = itemStack;
}
}
InventoryChangeListener.checkTasks(taskStarted);
InventoryChangeListener.onInventoryInteract(user, destinationInventory, originalDestination, null, destinationInventory.getLocation(), true);
}
catch (Exception e) {
e.printStackTrace();
}
});
// Start the processor if it's not already running
if (processorRunning.compareAndSet(false, true)) {
startHopperProcessor();
}
}
private static void startHopperProcessor() {
if (activeProcessors.incrementAndGet() <= MAX_CONCURRENT_PROCESSORS) {
Scheduler.runTaskAsynchronously(CoreProtect.getInstance(), () -> {
try {
// Use the same server running check as Consumer class
while (!hopperQueue.isEmpty() && (ConfigHandler.serverRunning || ConfigHandler.converterRunning)) {
processHopperBatch();
}
}
catch (Exception e) {
e.printStackTrace();
}
finally {
activeProcessors.decrementAndGet();
processorRunning.set(false);
// If more items were added and we're still running, restart the processor
if (!hopperQueue.isEmpty() && (ConfigHandler.serverRunning || ConfigHandler.converterRunning) && activeProcessors.get() == 0) {
startHopperProcessor();
}
}
});
}
else {
activeProcessors.decrementAndGet();
}
}
private static void processHopperBatch() {
int processed = 0;
final long taskStarted = InventoryChangeListener.tasksStarted.incrementAndGet();
while (!hopperQueue.isEmpty() && processed < BATCH_SIZE && (ConfigHandler.serverRunning || ConfigHandler.converterRunning)) {
Object[] data = hopperQueue.poll();
if (data == null)
continue;
Location location = (Location) data[0];
String user = (String) data[1];
InventoryHolder sourceHolder = (InventoryHolder) data[2];
InventoryHolder destinationHolder = (InventoryHolder) data[3];
ItemStack movedItem = (ItemStack) data[4];
ItemStack[] destinationContainer = (ItemStack[]) data[5];
String loggingChestId = (String) data[6];
Object[] lastAbort = (Object[]) data[7];
processSingleHopperPush(location, user, sourceHolder, destinationHolder, movedItem, destinationContainer, loggingChestId, lastAbort);
processed++;
}
InventoryChangeListener.checkTasks(taskStarted);
}
private static void processSingleHopperPush(Location location, String user, InventoryHolder sourceHolder, InventoryHolder destinationHolder, ItemStack movedItem, ItemStack[] destinationContainer, String loggingChestId, Object[] lastAbort) {
if (sourceHolder == null || destinationHolder == null) {
return;
}
boolean abort = false;
boolean addedInventory = ItemUtils.canAddContainer(destinationContainer, movedItem, destinationHolder.getInventory().getMaxStackSize());
if (!addedInventory) {
abort = true;
}
if (abort) {
Set<ItemStack> movedItems = new HashSet<>();
ItemStack[] destinationContents = destinationHolder.getInventory().getContents();
if (lastAbort != null && Arrays.equals(destinationContents, (ItemStack[]) lastAbort[1])) {
((Set<?>) lastAbort[0]).forEach(itemStack -> movedItems.add((ItemStack) itemStack));
}
movedItems.add(movedItem);
ConfigHandler.hopperAbort.put(loggingChestId, new Object[] { movedItems, ItemUtils.getContainerState(destinationContents) });
return;
}
else {
ConfigHandler.hopperSuccess.put(loggingChestId, new Object[] { destinationContainer, movedItem });
}
List<Object> list = ConfigHandler.transactingChest.get(location.getWorld().getUID().toString() + "." + location.getBlockX() + "." + location.getBlockY() + "." + location.getBlockZ());
if (list != null) {
list.add(movedItem);
}
if (Config.getConfig(location.getWorld()).HOPPER_FILTER_META && !movedItem.hasItemMeta()) {
return;
}
Inventory destinationInventory = destinationHolder.getInventory();
ItemStack[] originalDestination = destinationInventory.getContents().clone();
int removeAmount = movedItem.getAmount();
for (int i = 0; i < originalDestination.length; i++) {
if (removeAmount == 0) {
break;
}
ItemStack itemStack = (originalDestination[i] != null ? originalDestination[i].clone() : null);
if (itemStack != null && itemStack.isSimilar(movedItem)) {
if (itemStack.getAmount() >= removeAmount) {
itemStack.setAmount(itemStack.getAmount() - removeAmount);
removeAmount = 0;
}
else {
removeAmount = removeAmount - itemStack.getAmount();
itemStack = null;
}
originalDestination[i] = itemStack;
}
}
InventoryChangeListener.onInventoryInteract(user, destinationInventory, originalDestination, null, destinationInventory.getLocation(), true);
}
}

24
src/main/java/net/coreprotect/listener/player/InventoryChangeListener.java
View file

@ -43,14 +43,30 @@ public final class InventoryChangeListener extends Queue implements Listener {
protected static AtomicLong tasksStarted = new AtomicLong();
protected static AtomicLong tasksCompleted = new AtomicLong();
private static ConcurrentHashMap<String, Boolean> inventoryProcessing = new ConcurrentHashMap<>();
private static final Object taskCompletionLock = new Object();
private static final long TASK_WAIT_MAX_MS = 50; // Maximum wait time in milliseconds
protected static void checkTasks(long taskStarted) {
try {
int waitCount = 0;
while (tasksCompleted.get() < (taskStarted - 1L) && waitCount++ <= 50) {
Thread.sleep(1);
// Skip checking if this is the first task or we're already caught up
if (taskStarted <= 1 || tasksCompleted.get() >= (taskStarted - 1L)) {
tasksCompleted.set(taskStarted);
return;
}
// Try to update without waiting if possible
if (tasksCompleted.compareAndSet(taskStarted - 1L, taskStarted)) {
return;
}
// Use proper synchronization instead of busy waiting
synchronized (taskCompletionLock) {
if (tasksCompleted.get() < (taskStarted - 1L)) {
taskCompletionLock.wait(TASK_WAIT_MAX_MS);
}
tasksCompleted.set(taskStarted);
taskCompletionLock.notifyAll(); // Notify other waiting threads
}
tasksCompleted.set(taskStarted);
}
catch (Exception e) {
e.printStackTrace();