1
mirror of https://github.com/StarWishsama/Slimefun4.git synced 2024-09-20 03:35:51 +00:00

Make round robin better

This commit is contained in:
qwertyuioplkjhgfd 2022-02-25 20:51:16 -08:00
parent e3efd3330c
commit 73c33e0100

View File

@ -168,6 +168,7 @@ class CargoNetworkTask implements Runnable {
boolean roundrobin = Objects.equals(cfg.getString("round-robin"), "true"); boolean roundrobin = Objects.equals(cfg.getString("round-robin"), "true");
boolean smartFill = Objects.equals(cfg.getString("smart-fill"), "true"); boolean smartFill = Objects.equals(cfg.getString("smart-fill"), "true");
int index = 0;
Collection<Location> destinations; Collection<Location> destinations;
if (roundrobin) { if (roundrobin) {
// Use an ArrayDeque to perform round-robin sorting // Use an ArrayDeque to perform round-robin sorting
@ -175,8 +176,9 @@ class CargoNetworkTask implements Runnable {
// An ArrayDequeue is preferable as opposed to a LinkedList: // An ArrayDequeue is preferable as opposed to a LinkedList:
// - The number of elements does not change. // - The number of elements does not change.
// - ArrayDequeue has better iterative performance // - ArrayDequeue has better iterative performance
index = network.roundRobin.getOrDefault(inputNode, 0);
Deque<Location> tempDestinations = new ArrayDeque<>(outputNodes); Deque<Location> tempDestinations = new ArrayDeque<>(outputNodes);
roundRobinSort(inputNode, tempDestinations); roundRobinSort(index, tempDestinations);
destinations = tempDestinations; destinations = tempDestinations;
} else { } else {
// Using an ArrayList here since we won't need to sort the destinations // Using an ArrayList here since we won't need to sort the destinations
@ -190,8 +192,12 @@ class CargoNetworkTask implements Runnable {
if (target.isPresent()) { if (target.isPresent()) {
ItemStackWrapper wrapper = ItemStackWrapper.wrap(item); ItemStackWrapper wrapper = ItemStackWrapper.wrap(item);
item = CargoUtils.insert(network, inventories, output.getBlock(), target.get(), smartFill, item, wrapper); item = CargoUtils.insert(network, inventories, output.getBlock(), target.get(), smartFill, item, wrapper);
index++;
if (item == null) { if (item == null) {
if (roundrobin) {
network.roundRobin.put(inputNode, index % outputNodes.size());
}
break; break;
} }
} }
@ -204,27 +210,19 @@ class CargoNetworkTask implements Runnable {
* This method sorts a given {@link Deque} of output node locations using a semi-accurate * This method sorts a given {@link Deque} of output node locations using a semi-accurate
* round-robin method. * round-robin method.
* *
* @param inputNode * @param index
* The {@link Location} of the input node * The round-robin index of the input node
* @param outputNodes * @param outputNodes
* A {@link Deque} of {@link Location Locations} of the output nodes * A {@link Deque} of {@link Location Locations} of the output nodes
*/ */
private void roundRobinSort(Location inputNode, Deque<Location> outputNodes) { private void roundRobinSort(int index, Deque<Location> outputNodes) {
int index = network.roundRobin.getOrDefault(inputNode, 0);
if (index < outputNodes.size()) { if (index < outputNodes.size()) {
// Not ideal but actually not bad performance-wise over more elegant alternatives // Not ideal but actually not bad performance-wise over more elegant alternatives
for (int i = 0; i < index; i++) { for (int i = 0; i < index; i++) {
Location temp = outputNodes.removeFirst(); Location temp = outputNodes.removeFirst();
outputNodes.add(temp); outputNodes.add(temp);
} }
index++;
} else {
index = 1;
} }
network.roundRobin.put(inputNode, index);
} }
} }