Spaces:
Build error
Build error
File size: 2,806 Bytes
d46f4a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 |
package net.minecraft.util.thread;
import com.google.common.collect.Queues;
import java.util.Locale;
import java.util.Queue;
import java.util.concurrent.atomic.AtomicInteger;
import javax.annotation.Nullable;
public interface StrictQueue<T extends Runnable> {
@Nullable
Runnable pop();
boolean push(T p_365336_);
boolean isEmpty();
int size();
public static final class FixedPriorityQueue implements StrictQueue<StrictQueue.RunnableWithPriority> {
private final Queue<Runnable>[] queues;
private final AtomicInteger size = new AtomicInteger();
public FixedPriorityQueue(int p_18773_) {
this.queues = new Queue[p_18773_];
for (int i = 0; i < p_18773_; i++) {
this.queues[i] = Queues.newConcurrentLinkedQueue();
}
}
@Nullable
@Override
public Runnable pop() {
for (Queue<Runnable> queue : this.queues) {
Runnable runnable = queue.poll();
if (runnable != null) {
this.size.decrementAndGet();
return runnable;
}
}
return null;
}
public boolean push(StrictQueue.RunnableWithPriority p_361706_) {
int i = p_361706_.priority;
if (i < this.queues.length && i >= 0) {
this.queues[i].add(p_361706_);
this.size.incrementAndGet();
return true;
} else {
throw new IndexOutOfBoundsException(
String.format(Locale.ROOT, "Priority %d not supported. Expected range [0-%d]", i, this.queues.length - 1)
);
}
}
@Override
public boolean isEmpty() {
return this.size.get() == 0;
}
@Override
public int size() {
return this.size.get();
}
}
public static final class QueueStrictQueue implements StrictQueue<Runnable> {
private final Queue<Runnable> queue;
public QueueStrictQueue(Queue<Runnable> p_18792_) {
this.queue = p_18792_;
}
@Nullable
@Override
public Runnable pop() {
return this.queue.poll();
}
@Override
public boolean push(Runnable p_368428_) {
return this.queue.add(p_368428_);
}
@Override
public boolean isEmpty() {
return this.queue.isEmpty();
}
@Override
public int size() {
return this.queue.size();
}
}
public static record RunnableWithPriority(int priority, Runnable task) implements Runnable {
@Override
public void run() {
this.task.run();
}
}
} |