OpenSource/PaperMC
OpenSource/PaperMC
1
0
Fork 0
mirror of https://github.com/PaperMC/Paper.git synced 2024-12-01 12:41:50 +01:00
Code Issues Projects Releases Packages Wiki Activity
PaperMC/patches/server/1037-fixup-ConcurrentUtil.patch
Spottedleaf 35ef2932be Finish rebase
2024-10-25 11:40:25 -07:00

6177 lines
229 KiB
Diff
Raw Blame History

From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
From: Spottedleaf <Spottedleaf@users.noreply.github.com>
Date: Mon, 21 Oct 2024 11:47:34 -0700
Subject: [PATCH] fixup! ConcurrentUtil
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/collection/SRSWLinkedQueue.java b/src/main/java/ca/spottedleaf/concurrentutil/collection/SRSWLinkedQueue.java
deleted file mode 100644
index 094eff418b4e3bffce020d650931b4d9e58fa9ed..0000000000000000000000000000000000000000
--- a/src/main/java/ca/spottedleaf/concurrentutil/collection/SRSWLinkedQueue.java
+++ /dev/null
@@ -1,149 +0,0 @@
-package ca.spottedleaf.concurrentutil.collection;
-
-import ca.spottedleaf.concurrentutil.util.ConcurrentUtil;
-import ca.spottedleaf.concurrentutil.util.Validate;
-
-import java.lang.invoke.VarHandle;
-import java.util.ConcurrentModificationException;
-
-/**
- * Single reader thread single writer thread queue. The reader side of the queue is ordered by acquire semantics,
- * and the writer side of the queue is ordered by release semantics.
- */
-// TODO test
-public class SRSWLinkedQueue<E> {
-
- // always non-null
- protected LinkedNode<E> head;
-
- // always non-null
- protected LinkedNode<E> tail;
-
- /* IMPL NOTE: Leave hashCode and equals to their defaults */
-
- public SRSWLinkedQueue() {
- final LinkedNode<E> dummy = new LinkedNode<>(null, null);
- this.head = this.tail = dummy;
- }
-
- /**
- * Must be the reader thread.
- *
- * <p>
- * Returns, without removing, the first element of this queue.
- * </p>
- * @return Returns, without removing, the first element of this queue.
- */
- public E peekFirst() {
- LinkedNode<E> head = this.head;
- E ret = head.getElementPlain();
- if (ret == null) {
- head = head.getNextAcquire();
- if (head == null) {
- // empty
- return null;
- }
- // update head reference for next poll() call
- this.head = head;
- // guaranteed to be non-null
- ret = head.getElementPlain();
- if (ret == null) {
- throw new ConcurrentModificationException("Multiple reader threads");
- }
- }
-
- return ret;
- }
-
- /**
- * Must be the reader thread.
- *
- * <p>
- * Returns and removes the first element of this queue.
- * </p>
- * @return Returns and removes the first element of this queue.
- */
- public E poll() {
- LinkedNode<E> head = this.head;
- E ret = head.getElementPlain();
- if (ret == null) {
- head = head.getNextAcquire();
- if (head == null) {
- // empty
- return null;
- }
- // guaranteed to be non-null
- ret = head.getElementPlain();
- if (ret == null) {
- throw new ConcurrentModificationException("Multiple reader threads");
- }
- }
-
- head.setElementPlain(null);
- LinkedNode<E> next = head.getNextAcquire();
- this.head = next == null ? head : next;
-
- return ret;
- }
-
- /**
- * Must be the writer thread.
- *
- * <p>
- * Adds the element to the end of the queue.
- * </p>
- *
- * @throws NullPointerException If the provided element is null
- */
- public void addLast(final E element) {
- Validate.notNull(element, "Provided element cannot be null");
- final LinkedNode<E> append = new LinkedNode<>(element, null);
-
- this.tail.setNextRelease(append);
- this.tail = append;
- }
-
- protected static final class LinkedNode<E> {
-
- protected volatile Object element;
- protected volatile LinkedNode<E> next;
-
- protected static final VarHandle ELEMENT_HANDLE = ConcurrentUtil.getVarHandle(LinkedNode.class, "element", Object.class);
- protected static final VarHandle NEXT_HANDLE = ConcurrentUtil.getVarHandle(LinkedNode.class, "next", LinkedNode.class);
-
- protected LinkedNode(final Object element, final LinkedNode<E> next) {
- ELEMENT_HANDLE.set(this, element);
- NEXT_HANDLE.set(this, next);
- }
-
- /* element */
-
- @SuppressWarnings("unchecked")
- protected final E getElementPlain() {
- return (E)ELEMENT_HANDLE.get(this);
- }
-
- protected final void setElementPlain(final E update) {
- ELEMENT_HANDLE.set(this, (Object)update);
- }
- /* next */
-
- @SuppressWarnings("unchecked")
- protected final LinkedNode<E> getNextPlain() {
- return (LinkedNode<E>)NEXT_HANDLE.get(this);
- }
-
- @SuppressWarnings("unchecked")
- protected final LinkedNode<E> getNextAcquire() {
- return (LinkedNode<E>)NEXT_HANDLE.getAcquire(this);
- }
-
- protected final void setNextPlain(final LinkedNode<E> next) {
- NEXT_HANDLE.set(this, next);
- }
-
- protected final void setNextRelease(final LinkedNode<E> next) {
- NEXT_HANDLE.setRelease(this, next);
- }
- }
-}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/completable/CallbackCompletable.java b/src/main/java/ca/spottedleaf/concurrentutil/completable/CallbackCompletable.java
new file mode 100644
index 0000000000000000000000000000000000000000..6bad6f8ecc0944d2f406924c7de7e227ff1e70fa
--- /dev/null
+++ b/src/main/java/ca/spottedleaf/concurrentutil/completable/CallbackCompletable.java
@@ -0,0 +1,110 @@
+package ca.spottedleaf.concurrentutil.completable;
+
+import ca.spottedleaf.concurrentutil.collection.MultiThreadedQueue;
+import ca.spottedleaf.concurrentutil.executor.Cancellable;
+import ca.spottedleaf.concurrentutil.util.ConcurrentUtil;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import java.util.function.BiConsumer;
+
+public final class CallbackCompletable<T> {
+
+ private static final Logger LOGGER = LoggerFactory.getLogger(CallbackCompletable.class);
+
+ private final MultiThreadedQueue<BiConsumer<T, Throwable>> waiters = new MultiThreadedQueue<>();
+ private T result;
+ private Throwable throwable;
+ private volatile boolean completed;
+
+ public boolean isCompleted() {
+ return this.completed;
+ }
+
+ /**
+ * Note: Can only use after calling {@link #addAsynchronousWaiter(BiConsumer)}, as this function performs zero
+ * synchronisation
+ */
+ public T getResult() {
+ return this.result;
+ }
+
+ /**
+ * Note: Can only use after calling {@link #addAsynchronousWaiter(BiConsumer)}, as this function performs zero
+ * synchronisation
+ */
+ public Throwable getThrowable() {
+ return this.throwable;
+ }
+
+ /**
+ * Adds a waiter that should only be completed asynchronously by the complete() calls. If complete()
+ * has already been called, returns {@code null} and does not invoke the specified consumer.
+ * @param consumer Consumer to be executed on completion
+ * @throws NullPointerException If consumer is null
+ * @return A cancellable which will control the execution of the specified consumer
+ */
+ public Cancellable addAsynchronousWaiter(final BiConsumer<T, Throwable> consumer) {
+ if (this.waiters.add(consumer)) {
+ return new CancellableImpl(consumer);
+ }
+ return null;
+ }
+
+ private void completeAllWaiters(final T result, final Throwable throwable) {
+ this.completed = true;
+ BiConsumer<T, Throwable> waiter;
+ while ((waiter = this.waiters.pollOrBlockAdds()) != null) {
+ this.completeWaiter(waiter, result, throwable);
+ }
+ }
+
+ private void completeWaiter(final BiConsumer<T, Throwable> consumer, final T result, final Throwable throwable) {
+ try {
+ consumer.accept(result, throwable);
+ } catch (final Throwable throwable2) {
+ LOGGER.error("Failed to complete callback " + ConcurrentUtil.genericToString(consumer), throwable2);
+ }
+ }
+
+ /**
+ * Adds a waiter that will be completed asynchronously by the complete() calls. If complete()
+ * has already been called, then invokes the consumer synchronously with the completed result.
+ * @param consumer Consumer to be executed on completion
+ * @throws NullPointerException If consumer is null
+ * @return A cancellable which will control the execution of the specified consumer
+ */
+ public Cancellable addWaiter(final BiConsumer<T, Throwable> consumer) {
+ if (this.waiters.add(consumer)) {
+ return new CancellableImpl(consumer);
+ }
+ this.completeWaiter(consumer, this.result, this.throwable);
+ return new CancellableImpl(consumer);
+ }
+
+ public void complete(final T result) {
+ this.result = result;
+ this.completeAllWaiters(result, null);
+ }
+
+ public void completeWithThrowable(final Throwable throwable) {
+ if (throwable == null) {
+ throw new NullPointerException("Throwable cannot be null");
+ }
+ this.throwable = throwable;
+ this.completeAllWaiters(null, throwable);
+ }
+
+ private final class CancellableImpl implements Cancellable {
+
+ private final BiConsumer<T, Throwable> waiter;
+
+ private CancellableImpl(final BiConsumer<T, Throwable> waiter) {
+ this.waiter = waiter;
+ }
+
+ @Override
+ public boolean cancel() {
+ return CallbackCompletable.this.waiters.remove(this.waiter);
+ }
+ }
+}
\ No newline at end of file
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/completable/Completable.java b/src/main/java/ca/spottedleaf/concurrentutil/completable/Completable.java
index 46d1bd01542ebeeffc0006a5c585a50dbbbff907..365616439fa079017d648ed7f6ddf6950a691adf 100644
--- a/src/main/java/ca/spottedleaf/concurrentutil/completable/Completable.java
+++ b/src/main/java/ca/spottedleaf/concurrentutil/completable/Completable.java
@@ -1,112 +1,737 @@
package ca.spottedleaf.concurrentutil.completable;
-import ca.spottedleaf.concurrentutil.collection.MultiThreadedQueue;
-import ca.spottedleaf.concurrentutil.executor.Cancellable;
import ca.spottedleaf.concurrentutil.util.ConcurrentUtil;
+import ca.spottedleaf.concurrentutil.util.Validate;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
+import java.lang.invoke.VarHandle;
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.CompletionException;
+import java.util.concurrent.CompletionStage;
+import java.util.concurrent.Executor;
+import java.util.concurrent.ForkJoinPool;
+import java.util.concurrent.locks.LockSupport;
import java.util.function.BiConsumer;
+import java.util.function.BiFunction;
+import java.util.function.Consumer;
+import java.util.function.Function;
+import java.util.function.Supplier;
public final class Completable<T> {
private static final Logger LOGGER = LoggerFactory.getLogger(Completable.class);
+ private static final Function<? super Throwable, ? extends Throwable> DEFAULT_EXCEPTION_HANDLER = (final Throwable thr) -> {
+ LOGGER.error("Unhandled exception during Completable operation", thr);
+ return thr;
+ };
- private final MultiThreadedQueue<BiConsumer<T, Throwable>> waiters = new MultiThreadedQueue<>();
- private T result;
- private Throwable throwable;
- private volatile boolean completed;
+ public static Executor getDefaultExecutor() {
+ return ForkJoinPool.commonPool();
+ }
+
+ private static final Transform<?, ?> COMPLETED_STACK = new Transform<>(null, null, null, null) {
+ @Override
+ public void run() {}
+ };
+ private volatile Transform<?, T> completeStack;
+ private static final VarHandle COMPLETE_STACK_HANDLE = ConcurrentUtil.getVarHandle(Completable.class, "completeStack", Transform.class);
+
+ private static final Object NULL_MASK = new Object();
+ private volatile Object result;
+ private static final VarHandle RESULT_HANDLE = ConcurrentUtil.getVarHandle(Completable.class, "result", Object.class);
- public boolean isCompleted() {
- return this.completed;
+ private Object getResultPlain() {
+ return (Object)RESULT_HANDLE.get(this);
}
- /**
- * Note: Can only use after calling {@link #addAsynchronousWaiter(BiConsumer)}, as this function performs zero
- * synchronisation
- */
- public T getResult() {
- return this.result;
+ private Object getResultVolatile() {
+ return (Object)RESULT_HANDLE.getVolatile(this);
}
- /**
- * Note: Can only use after calling {@link #addAsynchronousWaiter(BiConsumer)}, as this function performs zero
- * synchronisation
- */
- public Throwable getThrowable() {
- return this.throwable;
+ private void pushStackOrRun(final Transform<?, T> push) {
+ int failures = 0;
+ for (Transform<?, T> curr = (Transform<?, T>)COMPLETE_STACK_HANDLE.getVolatile(this);;) {
+ if (curr == COMPLETED_STACK) {
+ push.execute();
+ return;
+ }
+
+ push.next = curr;
+
+ for (int i = 0; i < failures; ++i) {
+ ConcurrentUtil.backoff();
+ }
+
+ if (curr == (curr = (Transform<?, T>)COMPLETE_STACK_HANDLE.compareAndExchange(this, curr, push))) {
+ return;
+ }
+ push.next = null;
+ ++failures;
+ }
}
- /**
- * Adds a waiter that should only be completed asynchronously by the complete() calls. If complete()
- * has already been called, returns {@code null} and does not invoke the specified consumer.
- * @param consumer Consumer to be executed on completion
- * @throws NullPointerException If consumer is null
- * @return A cancellable which will control the execution of the specified consumer
- */
- public Cancellable addAsynchronousWaiter(final BiConsumer<T, Throwable> consumer) {
- if (this.waiters.add(consumer)) {
- return new CancellableImpl(consumer);
+ private void propagateStack() {
+ Transform<?, T> topStack = (Transform<?, T>)COMPLETE_STACK_HANDLE.getAndSet(this, COMPLETED_STACK);
+ while (topStack != null) {
+ topStack.execute();
+ topStack = topStack.next;
}
- return null;
}
- private void completeAllWaiters(final T result, final Throwable throwable) {
- this.completed = true;
- BiConsumer<T, Throwable> waiter;
- while ((waiter = this.waiters.pollOrBlockAdds()) != null) {
- this.completeWaiter(waiter, result, throwable);
+ private static Object maskNull(final Object res) {
+ return res == null ? NULL_MASK : res;
+ }
+
+ private static Object unmaskNull(final Object res) {
+ return res == NULL_MASK ? null : res;
+ }
+
+ private static Executor checkExecutor(final Executor executor) {
+ return Validate.notNull(executor, "Executor may not be null");
+ }
+
+ public Completable() {}
+
+ private Completable(final Object complete) {
+ COMPLETE_STACK_HANDLE.set(this, COMPLETED_STACK);
+ RESULT_HANDLE.setRelease(this, complete);
+ }
+
+ public static <T> Completable<T> completed(final T value) {
+ return new Completable<>(maskNull(value));
+ }
+
+ public static <T> Completable<T> failed(final Throwable ex) {
+ Validate.notNull(ex, "Exception may not be null");
+
+ return new Completable<>(new ExceptionResult(ex));
+ }
+
+ public static <T> Completable<T> supplied(final Supplier<T> supplier) {
+ return supplied(supplier, DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public static <T> Completable<T> supplied(final Supplier<T> supplier, final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ try {
+ return completed(supplier.get());
+ } catch (final Throwable throwable) {
+ Throwable complete;
+ try {
+ complete = exceptionHandler.apply(throwable);
+ } catch (final Throwable thr2) {
+ throwable.addSuppressed(thr2);
+ complete = throwable;
+ }
+ return failed(complete);
}
}
- private void completeWaiter(final BiConsumer<T, Throwable> consumer, final T result, final Throwable throwable) {
+ public static <T> Completable<T> suppliedAsync(final Supplier<T> supplier, final Executor executor) {
+ return suppliedAsync(supplier, executor, DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public static <T> Completable<T> suppliedAsync(final Supplier<T> supplier, final Executor executor, final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ final Completable<T> ret = new Completable<>();
+
+ class AsyncSuppliedCompletable implements Runnable, CompletableFuture.AsynchronousCompletionTask {
+ @Override
+ public void run() {
+ try {
+ ret.complete(supplier.get());
+ } catch (final Throwable throwable) {
+ Throwable complete;
+ try {
+ complete = exceptionHandler.apply(throwable);
+ } catch (final Throwable thr2) {
+ throwable.addSuppressed(thr2);
+ complete = throwable;
+ }
+ ret.completeExceptionally(complete);
+ }
+ }
+ }
+
try {
- consumer.accept(result, throwable);
- } catch (final ThreadDeath death) {
- throw death;
- } catch (final Throwable throwable2) {
- LOGGER.error("Failed to complete callback " + ConcurrentUtil.genericToString(consumer), throwable2);
+ executor.execute(new AsyncSuppliedCompletable());
+ } catch (final Throwable throwable) {
+ Throwable complete;
+ try {
+ complete = exceptionHandler.apply(throwable);
+ } catch (final Throwable thr2) {
+ throwable.addSuppressed(thr2);
+ complete = throwable;
+ }
+ ret.completeExceptionally(complete);
+ }
+
+ return ret;
+ }
+
+ private boolean completeRaw(final Object value) {
+ if ((Object)RESULT_HANDLE.getVolatile(this) != null || !(boolean)RESULT_HANDLE.compareAndSet(this, (Object)null, value)) {
+ return false;
+ }
+
+ this.propagateStack();
+ return true;
+ }
+
+ public boolean complete(final T result) {
+ return this.completeRaw(maskNull(result));
+ }
+
+ public boolean completeExceptionally(final Throwable exception) {
+ Validate.notNull(exception, "Exception may not be null");
+
+ return this.completeRaw(new ExceptionResult(exception));
+ }
+
+ public boolean isDone() {
+ return this.getResultVolatile() != null;
+ }
+
+ public boolean isNormallyComplete() {
+ return this.getResultVolatile() != null && !(this.getResultVolatile() instanceof ExceptionResult);
+ }
+
+ public boolean isExceptionallyComplete() {
+ return this.getResultVolatile() instanceof ExceptionResult;
+ }
+
+ public Throwable getException() {
+ final Object res = this.getResultVolatile();
+ if (res == null) {
+ return null;
+ }
+
+ if (!(res instanceof ExceptionResult exRes)) {
+ throw new IllegalStateException("Not completed exceptionally");
+ }
+
+ return exRes.ex;
+ }
+
+ public T getNow(final T dfl) throws CompletionException {
+ final Object res = this.getResultVolatile();
+ if (res == null) {
+ return dfl;
+ }
+
+ if (res instanceof ExceptionResult exRes) {
+ throw new CompletionException(exRes.ex);
+ }
+
+ return (T)unmaskNull(res);
+ }
+
+ public T join() throws CompletionException {
+ if (this.isDone()) {
+ return this.getNow(null);
+ }
+
+ final UnparkTransform<T> unparkTransform = new UnparkTransform<>(this, Thread.currentThread());
+
+ this.pushStackOrRun(unparkTransform);
+
+ boolean interuptted = false;
+ while (!unparkTransform.isReleasable()) {
+ try {
+ ForkJoinPool.managedBlock(unparkTransform);
+ } catch (final InterruptedException ex) {
+ interuptted = true;
+ }
+ }
+
+ if (interuptted) {
+ Thread.currentThread().interrupt();
+ }
+
+ return this.getNow(null);
+ }
+
+ public CompletableFuture<T> toFuture() {
+ final Object rawResult = this.getResultVolatile();
+ if (rawResult != null) {
+ if (rawResult instanceof ExceptionResult exRes) {
+ return CompletableFuture.failedFuture(exRes.ex);
+ } else {
+ return CompletableFuture.completedFuture((T)unmaskNull(rawResult));
+ }
+ }
+
+ final CompletableFuture<T> ret = new CompletableFuture<>();
+
+ class ToFuture implements BiConsumer<T, Throwable> {
+
+ @Override
+ public void accept(final T res, final Throwable ex) {
+ if (ex != null) {
+ ret.completeExceptionally(ex);
+ } else {
+ ret.complete(res);
+ }
+ }
+ }
+
+ this.whenComplete(new ToFuture());
+
+ return ret;
+ }
+
+ public static <T> Completable<T> fromFuture(final CompletionStage<T> stage) {
+ final Completable<T> ret = new Completable<>();
+
+ class FromFuture implements BiConsumer<T, Throwable> {
+ @Override
+ public void accept(final T res, final Throwable ex) {
+ if (ex != null) {
+ ret.completeExceptionally(ex);
+ } else {
+ ret.complete(res);
+ }
+ }
+ }
+
+ stage.whenComplete(new FromFuture());
+
+ return ret;
+ }
+
+
+ public <U> Completable<U> thenApply(final Function<? super T, ? extends U> function) {
+ return this.thenApply(function, DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public <U> Completable<U> thenApply(final Function<? super T, ? extends U> function, final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ Validate.notNull(function, "Function may not be null");
+ Validate.notNull(exceptionHandler, "Exception handler may not be null");
+
+ final Completable<U> ret = new Completable<>();
+ this.pushStackOrRun(new ApplyTransform<>(null, this, ret, exceptionHandler, function));
+ return ret;
+ }
+
+ public <U> Completable<U> thenApplyAsync(final Function<? super T, ? extends U> function) {
+ return this.thenApplyAsync(function, getDefaultExecutor(), DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public <U> Completable<U> thenApplyAsync(final Function<? super T, ? extends U> function, final Executor executor) {
+ return this.thenApplyAsync(function, executor, DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public <U> Completable<U> thenApplyAsync(final Function<? super T, ? extends U> function, final Executor executor, final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ Validate.notNull(function, "Function may not be null");
+ Validate.notNull(exceptionHandler, "Exception handler may not be null");
+
+ final Completable<U> ret = new Completable<>();
+ this.pushStackOrRun(new ApplyTransform<>(checkExecutor(executor), this, ret, exceptionHandler, function));
+ return ret;
+ }
+
+
+ public Completable<Void> thenAccept(final Consumer<? super T> consumer) {
+ return this.thenAccept(consumer, DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public Completable<Void> thenAccept(final Consumer<? super T> consumer, final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ Validate.notNull(consumer, "Consumer may not be null");
+ Validate.notNull(exceptionHandler, "Exception handler may not be null");
+
+ final Completable<Void> ret = new Completable<>();
+ this.pushStackOrRun(new AcceptTransform<>(null, this, ret, exceptionHandler, consumer));
+ return ret;
+ }
+
+ public Completable<Void> thenAcceptAsync(final Consumer<? super T> consumer) {
+ return this.thenAcceptAsync(consumer, getDefaultExecutor(), DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public Completable<Void> thenAcceptAsync(final Consumer<? super T> consumer, final Executor executor) {
+ return this.thenAcceptAsync(consumer, executor, DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public Completable<Void> thenAcceptAsync(final Consumer<? super T> consumer, final Executor executor, final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ Validate.notNull(consumer, "Consumer may not be null");
+ Validate.notNull(exceptionHandler, "Exception handler may not be null");
+
+ final Completable<Void> ret = new Completable<>();
+ this.pushStackOrRun(new AcceptTransform<>(checkExecutor(executor), this, ret, exceptionHandler, consumer));
+ return ret;
+ }
+
+
+ public Completable<Void> thenRun(final Runnable run) {
+ return this.thenRun(run, DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public Completable<Void> thenRun(final Runnable run, final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ Validate.notNull(run, "Run may not be null");
+ Validate.notNull(exceptionHandler, "Exception handler may not be null");
+
+ final Completable<Void> ret = new Completable<>();
+ this.pushStackOrRun(new RunTransform<>(null, this, ret, exceptionHandler, run));
+ return ret;
+ }
+
+ public Completable<Void> thenRunAsync(final Runnable run) {
+ return this.thenRunAsync(run, getDefaultExecutor(), DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public Completable<Void> thenRunAsync(final Runnable run, final Executor executor) {
+ return this.thenRunAsync(run, executor, DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public Completable<Void> thenRunAsync(final Runnable run, final Executor executor, final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ Validate.notNull(run, "Run may not be null");
+ Validate.notNull(exceptionHandler, "Exception handler may not be null");
+
+ final Completable<Void> ret = new Completable<>();
+ this.pushStackOrRun(new RunTransform<>(checkExecutor(executor), this, ret, exceptionHandler, run));
+ return ret;
+ }
+
+
+ public <U> Completable<U> handle(final BiFunction<? super T, ? super Throwable, ? extends U> function) {
+ return this.handle(function, DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public <U> Completable<U> handle(final BiFunction<? super T, ? super Throwable, ? extends U> function,
+ final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ Validate.notNull(function, "Function may not be null");
+ Validate.notNull(exceptionHandler, "Exception handler may not be null");
+
+ final Completable<U> ret = new Completable<>();
+ this.pushStackOrRun(new HandleTransform<>(null, this, ret, exceptionHandler, function));
+ return ret;
+ }
+
+ public <U> Completable<U> handleAsync(final BiFunction<? super T, ? super Throwable, ? extends U> function) {
+ return this.handleAsync(function, getDefaultExecutor(), DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public <U> Completable<U> handleAsync(final BiFunction<? super T, ? super Throwable, ? extends U> function,
+ final Executor executor) {
+ return this.handleAsync(function, executor, DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public <U> Completable<U> handleAsync(final BiFunction<? super T, ? super Throwable, ? extends U> function,
+ final Executor executor,
+ final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ Validate.notNull(function, "Function may not be null");
+ Validate.notNull(exceptionHandler, "Exception handler may not be null");
+
+ final Completable<U> ret = new Completable<>();
+ this.pushStackOrRun(new HandleTransform<>(checkExecutor(executor), this, ret, exceptionHandler, function));
+ return ret;
+ }
+
+
+ public Completable<T> whenComplete(final BiConsumer<? super T, ? super Throwable> consumer) {
+ return this.whenComplete(consumer, DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public Completable<T> whenComplete(final BiConsumer<? super T, ? super Throwable> consumer, final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ Validate.notNull(consumer, "Consumer may not be null");
+ Validate.notNull(exceptionHandler, "Exception handler may not be null");
+
+ final Completable<T> ret = new Completable<>();
+ this.pushStackOrRun(new WhenTransform<>(null, this, ret, exceptionHandler, consumer));
+ return ret;
+ }
+
+ public Completable<T> whenCompleteAsync(final BiConsumer<? super T, ? super Throwable> consumer) {
+ return this.whenCompleteAsync(consumer, getDefaultExecutor(), DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public Completable<T> whenCompleteAsync(final BiConsumer<? super T, ? super Throwable> consumer, final Executor executor) {
+ return this.whenCompleteAsync(consumer, executor, DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public Completable<T> whenCompleteAsync(final BiConsumer<? super T, ? super Throwable> consumer, final Executor executor,
+ final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ Validate.notNull(consumer, "Consumer may not be null");
+ Validate.notNull(exceptionHandler, "Exception handler may not be null");
+
+ final Completable<T> ret = new Completable<>();
+ this.pushStackOrRun(new WhenTransform<>(checkExecutor(executor), this, ret, exceptionHandler, consumer));
+ return ret;
+ }
+
+
+ public Completable<T> exceptionally(final Function<Throwable, ? extends T> function) {
+ return this.exceptionally(function, DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public Completable<T> exceptionally(final Function<Throwable, ? extends T> function, final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ Validate.notNull(function, "Function may not be null");
+ Validate.notNull(exceptionHandler, "Exception handler may not be null");
+
+ final Completable<T> ret = new Completable<>();
+ this.pushStackOrRun(new ExceptionallyTransform<>(null, this, ret, exceptionHandler, function));
+ return ret;
+ }
+
+ public Completable<T> exceptionallyAsync(final Function<Throwable, ? extends T> function) {
+ return this.exceptionallyAsync(function, getDefaultExecutor(), DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public Completable<T> exceptionallyAsync(final Function<Throwable, ? extends T> function, final Executor executor) {
+ return this.exceptionallyAsync(function, executor, DEFAULT_EXCEPTION_HANDLER);
+ }
+
+ public Completable<T> exceptionallyAsync(final Function<Throwable, ? extends T> function, final Executor executor,
+ final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ Validate.notNull(function, "Function may not be null");
+ Validate.notNull(exceptionHandler, "Exception handler may not be null");
+
+ final Completable<T> ret = new Completable<>();
+ this.pushStackOrRun(new ExceptionallyTransform<>(checkExecutor(executor), this, ret, exceptionHandler, function));
+ return ret;
+ }
+
+ private static final class ExceptionResult {
+ public final Throwable ex;
+
+ public ExceptionResult(final Throwable ex) {
+ this.ex = ex;
}
}
- /**
- * Adds a waiter that will be completed asynchronously by the complete() calls. If complete()
- * has already been called, then invokes the consumer synchronously with the completed result.
- * @param consumer Consumer to be executed on completion
- * @throws NullPointerException If consumer is null
- * @return A cancellable which will control the execution of the specified consumer
- */
- public Cancellable addWaiter(final BiConsumer<T, Throwable> consumer) {
- if (this.waiters.add(consumer)) {
- return new CancellableImpl(consumer);
+ private static abstract class Transform<U, T> implements Runnable, CompletableFuture.AsynchronousCompletionTask {
+
+ private Transform<?, T> next;
+
+ private final Executor executor;
+ protected final Completable<T> from;
+ protected final Completable<U> to;
+ protected final Function<? super Throwable, ? extends Throwable> exceptionHandler;
+
+ protected Transform(final Executor executor, final Completable<T> from, final Completable<U> to,
+ final Function<? super Throwable, ? extends Throwable> exceptionHandler) {
+ this.executor = executor;
+ this.from = from;
+ this.to = to;
+ this.exceptionHandler = exceptionHandler;
+ }
+
+ // force interface call to become virtual call
+ @Override
+ public abstract void run();
+
+ protected void failed(final Throwable throwable) {
+ Throwable complete;
+ try {
+ complete = this.exceptionHandler.apply(throwable);
+ } catch (final Throwable thr2) {
+ throwable.addSuppressed(thr2);
+ complete = throwable;
+ }
+ this.to.completeExceptionally(complete);
+ }
+
+ public void execute() {
+ if (this.executor == null) {
+ this.run();
+ return;
+ }
+
+ try {
+ this.executor.execute(this);
+ } catch (final Throwable throwable) {
+ this.failed(throwable);
+ }
+ }
+ }
+
+ private static final class ApplyTransform<U, T> extends Transform<U, T> {
+
+ private final Function<? super T, ? extends U> function;
+
+ public ApplyTransform(final Executor executor, final Completable<T> from, final Completable<U> to,
+ final Function<? super Throwable, ? extends Throwable> exceptionHandler,
+ final Function<? super T, ? extends U> function) {
+ super(executor, from, to, exceptionHandler);
+ this.function = function;
+ }
+
+ @Override
+ public void run() {
+ final Object result = this.from.getResultPlain();
+ try {
+ if (result instanceof ExceptionResult exRes) {
+ this.to.completeExceptionally(exRes.ex);
+ } else {
+ this.to.complete(this.function.apply((T)unmaskNull(result)));
+ }
+ } catch (final Throwable throwable) {
+ this.failed(throwable);
+ }
}
- this.completeWaiter(consumer, this.result, this.throwable);
- return new CancellableImpl(consumer);
}
- public void complete(final T result) {
- this.result = result;
- this.completeAllWaiters(result, null);
+ private static final class AcceptTransform<T> extends Transform<Void, T> {
+ private final Consumer<? super T> consumer;
+
+ public AcceptTransform(final Executor executor, final Completable<T> from, final Completable<Void> to,
+ final Function<? super Throwable, ? extends Throwable> exceptionHandler,
+ final Consumer<? super T> consumer) {
+ super(executor, from, to, exceptionHandler);
+ this.consumer = consumer;
+ }
+
+ @Override
+ public void run() {
+ final Object result = this.from.getResultPlain();
+ try {
+ if (result instanceof ExceptionResult exRes) {
+ this.to.completeExceptionally(exRes.ex);
+ } else {
+ this.consumer.accept((T)unmaskNull(result));
+ this.to.complete(null);
+ }
+ } catch (final Throwable throwable) {
+ this.failed(throwable);
+ }
+ }
}
- public void completeWithThrowable(final Throwable throwable) {
- if (throwable == null) {
- throw new NullPointerException("Throwable cannot be null");
+ private static final class RunTransform<T> extends Transform<Void, T> {
+ private final Runnable run;
+
+ public RunTransform(final Executor executor, final Completable<T> from, final Completable<Void> to,
+ final Function<? super Throwable, ? extends Throwable> exceptionHandler,
+ final Runnable run) {
+ super(executor, from, to, exceptionHandler);
+ this.run = run;
+ }
+
+ @Override
+ public void run() {
+ final Object result = this.from.getResultPlain();
+ try {
+ if (result instanceof ExceptionResult exRes) {
+ this.to.completeExceptionally(exRes.ex);
+ } else {
+ this.run.run();
+ this.to.complete(null);
+ }
+ } catch (final Throwable throwable) {
+ this.failed(throwable);
+ }
}
- this.throwable = throwable;
- this.completeAllWaiters(null, throwable);
}
- private final class CancellableImpl implements Cancellable {
+ private static final class HandleTransform<U, T> extends Transform<U, T> {
+
+ private final BiFunction<? super T, ? super Throwable, ? extends U> function;
+
+ public HandleTransform(final Executor executor, final Completable<T> from, final Completable<U> to,
+ final Function<? super Throwable, ? extends Throwable> exceptionHandler,
+ final BiFunction<? super T, ? super Throwable, ? extends U> function) {
+ super(executor, from, to, exceptionHandler);
+ this.function = function;
+ }
- private final BiConsumer<T, Throwable> waiter;
+ @Override
+ public void run() {
+ final Object result = this.from.getResultPlain();
+ try {
+ if (result instanceof ExceptionResult exRes) {
+ this.to.complete(this.function.apply(null, exRes.ex));
+ } else {
+ this.to.complete(this.function.apply((T)unmaskNull(result), null));
+ }
+ } catch (final Throwable throwable) {
+ this.failed(throwable);
+ }
+ }
+ }
+
+ private static final class WhenTransform<T> extends Transform<T, T> {
+
+ private final BiConsumer<? super T, ? super Throwable> consumer;
+
+ public WhenTransform(final Executor executor, final Completable<T> from, final Completable<T> to,
+ final Function<? super Throwable, ? extends Throwable> exceptionHandler,
+ final BiConsumer<? super T, ? super Throwable> consumer) {
+ super(executor, from, to, exceptionHandler);
+ this.consumer = consumer;
+ }
+
+ @Override
+ public void run() {
+ final Object result = this.from.getResultPlain();
+ try {
+ if (result instanceof ExceptionResult exRes) {
+ this.consumer.accept(null, exRes.ex);
+ this.to.completeExceptionally(exRes.ex);
+ } else {
+ final T unmasked = (T)unmaskNull(result);
+ this.consumer.accept(unmasked, null);
+ this.to.complete(unmasked);
+ }
+ } catch (final Throwable throwable) {
+ this.failed(throwable);
+ }
+ }
+ }
+
+ private static final class ExceptionallyTransform<T> extends Transform<T, T> {
+ private final Function<Throwable, ? extends T> function;
+
+ public ExceptionallyTransform(final Executor executor, final Completable<T> from, final Completable<T> to,
+ final Function<? super Throwable, ? extends Throwable> exceptionHandler,
+ final Function<Throwable, ? extends T> function) {
+ super(executor, from, to, exceptionHandler);
+ this.function = function;
+ }
+
+ @Override
+ public void run() {
+ final Object result = this.from.getResultPlain();
+ try {
+ if (result instanceof ExceptionResult exRes) {
+ this.to.complete(this.function.apply(exRes.ex));
+ } else {
+ this.to.complete((T)unmaskNull(result));
+ }
+ } catch (final Throwable throwable) {
+ this.failed(throwable);
+ }
+ }
+ }
+
+ private static final class UnparkTransform<T> extends Transform<Void, T> implements ForkJoinPool.ManagedBlocker {
+
+ private volatile Thread thread;
+
+ public UnparkTransform(final Completable<T> from, final Thread target) {
+ super(null, from, null, null);
+ this.thread = target;
+ }
+
+ @Override
+ public void run() {
+ final Thread t = this.thread;
+ this.thread = null;
+ LockSupport.unpark(t);
+ }
+
+ @Override
+ public boolean block() throws InterruptedException {
+ while (!this.isReleasable()) {
+ if (Thread.interrupted()) {
+ throw new InterruptedException();
+ }
+ LockSupport.park(this);
+ }
- private CancellableImpl(final BiConsumer<T, Throwable> waiter) {
- this.waiter = waiter;
+ return true;
}
@Override
- public boolean cancel() {
- return Completable.this.waiters.remove(this.waiter);
+ public boolean isReleasable() {
+ return this.thread == null;
}
}
}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/executor/BaseExecutor.java b/src/main/java/ca/spottedleaf/concurrentutil/executor/BaseExecutor.java
deleted file mode 100644
index 18d646676fd022afd64afaac30ec1bd283a73b0e..0000000000000000000000000000000000000000
--- a/src/main/java/ca/spottedleaf/concurrentutil/executor/BaseExecutor.java
+++ /dev/null
@@ -1,208 +0,0 @@
-package ca.spottedleaf.concurrentutil.executor;
-
-import ca.spottedleaf.concurrentutil.util.ConcurrentUtil;
-import java.util.function.BooleanSupplier;
-
-/**
- * Base implementation for an abstract queue of tasks which are executed either synchronously or asynchronously.
- *
- * <p>
- * The implementation supports tracking task executions using {@link #getTotalTasksScheduled()} and
- * {@link #getTotalTasksExecuted()}, and optionally shutting down the executor using {@link #shutdown()}
- * </p>
- *
- * <p>
- * The base implementation does not provide a method to queue a task for execution, rather that is specified in
- * the specific implementation. However, it is required that a specific implementation provides a method to
- * <i>queue</i> a task or <i>create</i> a task. A <i>queued</i> task is one which will eventually be executed,
- * and a <i>created</i> task must be queued to execute via {@link BaseTask#queue()} or be executed manually via
- * {@link BaseTask#execute()}. This choice of delaying the queueing of a task may be useful to provide a task handle
- * which may be cancelled or adjusted before the actual real task logic is ready to be executed.
- * </p>
- */
-public interface BaseExecutor {
-
- /**
- * Returns whether every task scheduled to this queue has been removed and executed or cancelled. If no tasks have been queued,
- * returns {@code true}.
- *
- * @return {@code true} if all tasks that have been queued have finished executing or no tasks have been queued, {@code false} otherwise.
- */
- public default boolean haveAllTasksExecuted() {
- // order is important
- // if new tasks are scheduled between the reading of these variables, scheduled is guaranteed to be higher -
- // so our check fails, and we try again
- final long completed = this.getTotalTasksExecuted();
- final long scheduled = this.getTotalTasksScheduled();
-
- return completed == scheduled;
- }
-
- /**
- * Returns the number of tasks that have been scheduled or execute or are pending to be scheduled.
- */
- public long getTotalTasksScheduled();
-
- /**
- * Returns the number of tasks that have fully been executed.
- */
- public long getTotalTasksExecuted();
-
- /**
- * Waits until this queue has had all of its tasks executed (NOT removed). See {@link #haveAllTasksExecuted()}
- * <p>
- * This call is most effective after a {@link #shutdown()} call, as the shutdown call guarantees no tasks can
- * be executed and the waitUntilAllExecuted call makes sure the queue is empty. Effectively, using shutdown then using
- * waitUntilAllExecuted ensures this queue is empty - and most importantly, will remain empty.
- * </p>
- * <p>
- * This method is not guaranteed to be immediately responsive to queue state, so calls may take significantly more
- * time than expected. Effectively, do not rely on this call being fast - even if there are few tasks scheduled.
- * </p>
- * <p>
- * Note: Interruptions to the the current thread have no effect. Interrupt status is also not affected by this call.
- * </p>
- *
- * @throws IllegalStateException If the current thread is not allowed to wait
- */
- public default void waitUntilAllExecuted() throws IllegalStateException {
- long failures = 1L; // start at 0.25ms
-
- while (!this.haveAllTasksExecuted()) {
- Thread.yield();
- failures = ConcurrentUtil.linearLongBackoff(failures, 250_000L, 5_000_000L); // 500us, 5ms
- }
- }
-
- /**
- * Executes the next available task.
- *
- * @return {@code true} if a task was executed, {@code false} otherwise
- * @throws IllegalStateException If the current thread is not allowed to execute a task
- */
- public boolean executeTask() throws IllegalStateException;
-
- /**
- * Executes all queued tasks.
- *
- * @return {@code true} if a task was executed, {@code false} otherwise
- * @throws IllegalStateException If the current thread is not allowed to execute a task
- */
- public default boolean executeAll() {
- if (!this.executeTask()) {
- return false;
- }
-
- while (this.executeTask());
-
- return true;
- }
-
- /**
- * Waits and executes tasks until the condition returns {@code true}.
- * <p>
- * WARNING: This function is <i>not</i> suitable for waiting until a deadline!
- * Use {@link #executeUntil(long)} or {@link #executeConditionally(BooleanSupplier, long)} instead.
- * </p>
- */
- public default void executeConditionally(final BooleanSupplier condition) {
- long failures = 0;
- while (!condition.getAsBoolean()) {
- if (this.executeTask()) {
- failures = failures >>> 2;
- } else {
- failures = ConcurrentUtil.linearLongBackoff(failures, 100_000L, 10_000_000L); // 100us, 10ms
- }
- }
- }
-
- /**
- * Waits and executes tasks until the condition returns {@code true} or {@code System.nanoTime() - deadline >= 0}.
- */
- public default void executeConditionally(final BooleanSupplier condition, final long deadline) {
- long failures = 0;
- // double check deadline; we don't know how expensive the condition is
- while ((System.nanoTime() - deadline < 0L) && !condition.getAsBoolean() && (System.nanoTime() - deadline < 0L)) {
- if (this.executeTask()) {
- failures = failures >>> 2;
- } else {
- failures = ConcurrentUtil.linearLongBackoffDeadline(failures, 100_000L, 10_000_000L, deadline); // 100us, 10ms
- }
- }
- }
-
- /**
- * Waits and executes tasks until {@code System.nanoTime() - deadline >= 0}.
- */
- public default void executeUntil(final long deadline) {
- long failures = 0;
- while (System.nanoTime() - deadline < 0L) {
- if (this.executeTask()) {
- failures = failures >>> 2;
- } else {
- failures = ConcurrentUtil.linearLongBackoffDeadline(failures, 100_000L, 10_000_000L, deadline); // 100us, 10ms
- }
- }
- }
-
- /**
- * Prevent further additions to this queue. Attempts to add after this call has completed (potentially during) will
- * result in {@link IllegalStateException} being thrown.
- * <p>
- * This operation is atomic with respect to other shutdown calls
- * </p>
- * <p>
- * After this call has completed, regardless of return value, this queue will be shutdown.
- * </p>
- *
- * @return {@code true} if the queue was shutdown, {@code false} if it has shut down already
- * @throws UnsupportedOperationException If this queue does not support shutdown
- * @see #isShutdown()
- */
- public default boolean shutdown() throws UnsupportedOperationException {
- throw new UnsupportedOperationException();
- }
-
- /**
- * Returns whether this queue has shut down. Effectively, whether new tasks will be rejected - this method
- * does not indicate whether all the tasks scheduled have been executed.
- * @return Returns whether this queue has shut down.
- * @see #waitUntilAllExecuted()
- */
- public default boolean isShutdown() {
- return false;
- }
-
- /**
- * Task object returned for any {@link BaseExecutor} scheduled task.
- * @see BaseExecutor
- */
- public static interface BaseTask extends Cancellable {
-
- /**
- * Causes a lazily queued task to become queued or executed
- *
- * @throws IllegalStateException If the backing queue has shutdown
- * @return {@code true} If the task was queued, {@code false} if the task was already queued/cancelled/executed
- */
- public boolean queue();
-
- /**
- * Forces this task to be marked as completed.
- *
- * @return {@code true} if the task was cancelled, {@code false} if the task has already completed or is being completed.
- */
- @Override
- public boolean cancel();
-
- /**
- * Executes this task. This will also mark the task as completing.
- * <p>
- * Exceptions thrown from the runnable will be rethrown.
- * </p>
- *
- * @return {@code true} if this task was executed, {@code false} if it was already marked as completed.
- */
- public boolean execute();
- }
-}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/executor/PrioritisedExecutor.java b/src/main/java/ca/spottedleaf/concurrentutil/executor/PrioritisedExecutor.java
new file mode 100644
index 0000000000000000000000000000000000000000..17cbaee1e89bd3f6d905e640d20d0119ab0570a0
--- /dev/null
+++ b/src/main/java/ca/spottedleaf/concurrentutil/executor/PrioritisedExecutor.java
@@ -0,0 +1,271 @@
+package ca.spottedleaf.concurrentutil.executor;
+
+import ca.spottedleaf.concurrentutil.util.Priority;
+
+public interface PrioritisedExecutor {
+
+ /**
+ * Returns the number of tasks that have been scheduled are pending to be scheduled.
+ */
+ public long getTotalTasksScheduled();
+
+ /**
+ * Returns the number of tasks that have been executed.
+ */
+ public long getTotalTasksExecuted();
+
+ /**
+ * Generates the next suborder id.
+ * @return The next suborder id.
+ */
+ public long generateNextSubOrder();
+
+ /**
+ * Executes the next available task.
+ * <p>
+ * If there is a task with priority {@link Priority#BLOCKING} available, then that such task is executed.
+ * </p>
+ * <p>
+ * If there is a task with priority {@link Priority#IDLE} available then that task is only executed
+ * when there are no other tasks available with a higher priority.
+ * </p>
+ * <p>
+ * If there are no tasks that have priority {@link Priority#BLOCKING} or {@link Priority#IDLE}, then
+ * this function will be biased to execute tasks that have higher priorities.
+ * </p>
+ *
+ * @return {@code true} if a task was executed, {@code false} otherwise
+ * @throws IllegalStateException If the current thread is not allowed to execute a task
+ */
+ public boolean executeTask() throws IllegalStateException;
+
+ /**
+ * Prevent further additions to this executor. Attempts to add after this call has completed (potentially during) will
+ * result in {@link IllegalStateException} being thrown.
+ * <p>
+ * This operation is atomic with respect to other shutdown calls
+ * </p>
+ * <p>
+ * After this call has completed, regardless of return value, this executor will be shutdown.
+ * </p>
+ *
+ * @return {@code true} if the executor was shutdown, {@code false} if it has shut down already
+ * @see #isShutdown()
+ */
+ public boolean shutdown();
+
+ /**
+ * Returns whether this executor has shut down. Effectively, returns whether new tasks will be rejected.
+ * This method does not indicate whether all the tasks scheduled have been executed.
+ * @return Returns whether this executor has shut down.
+ */
+ public boolean isShutdown();
+
+ /**
+ * Queues or executes a task at {@link Priority#NORMAL} priority.
+ * @param task The task to run.
+ *
+ * @throws IllegalStateException If this executor has shutdown.
+ * @throws NullPointerException If the task is null
+ * @return {@code null} if the current thread immediately executed the task, else returns the prioritised task
+ * associated with the parameter
+ */
+ public PrioritisedTask queueTask(final Runnable task);
+
+ /**
+ * Queues or executes a task.
+ *
+ * @param task The task to run.
+ * @param priority The priority for the task.
+ *
+ * @throws IllegalStateException If this executor has shutdown.
+ * @throws NullPointerException If the task is null
+ * @throws IllegalArgumentException If the priority is invalid.
+ * @return {@code null} if the current thread immediately executed the task, else returns the prioritised task
+ * associated with the parameter
+ */
+ public PrioritisedTask queueTask(final Runnable task, final Priority priority);
+
+ /**
+ * Queues or executes a task.
+ *
+ * @param task The task to run.
+ * @param priority The priority for the task.
+ * @param subOrder The task's suborder.
+ *
+ * @throws IllegalStateException If this executor has shutdown.
+ * @throws NullPointerException If the task is null
+ * @throws IllegalArgumentException If the priority is invalid.
+ * @return {@code null} if the current thread immediately executed the task, else returns the prioritised task
+ * associated with the parameter
+ */
+ public PrioritisedTask queueTask(final Runnable task, final Priority priority, final long subOrder);
+
+ /**
+ * Creates, but does not queue or execute, a task at {@link Priority#NORMAL} priority.
+ * @param task The task to run.
+ *
+ * @throws NullPointerException If the task is null
+ * @return {@code null} if the current thread immediately executed the task, else returns the prioritised task
+ * associated with the parameter
+ */
+ public PrioritisedTask createTask(final Runnable task);
+
+ /**
+ * Creates, but does not queue or execute, a task at {@link Priority#NORMAL} priority.
+ *
+ * @param task The task to run.
+ * @param priority The priority for the task.
+ *
+ * @throws NullPointerException If the task is null
+ * @throws IllegalArgumentException If the priority is invalid.
+ * @return {@code null} if the current thread immediately executed the task, else returns the prioritised task
+ * associated with the parameter
+ */
+ public PrioritisedTask createTask(final Runnable task, final Priority priority);
+
+ /**
+ * Creates, but does not queue or execute, a task at {@link Priority#NORMAL} priority.
+ *
+ * @param task The task to run.
+ * @param priority The priority for the task.
+ * @param subOrder The task's suborder.
+ *
+ * @throws NullPointerException If the task is null
+ * @throws IllegalArgumentException If the priority is invalid.
+ * @return {@code null} if the current thread immediately executed the task, else returns the prioritised task
+ * associated with the parameter
+ */
+ public PrioritisedTask createTask(final Runnable task, final Priority priority, final long subOrder);
+
+ public static interface PrioritisedTask extends Cancellable {
+
+ /**
+ * Returns the executor associated with this task.
+ * @return The executor associated with this task.
+ */
+ public PrioritisedExecutor getExecutor();
+
+ /**
+ * Causes a lazily queued task to become queued or executed
+ *
+ * @throws IllegalStateException If the backing executor has shutdown
+ * @return {@code true} If the task was queued, {@code false} if the task was already queued/cancelled/executed
+ */
+ public boolean queue();
+
+ /**
+ * Returns whether this task has been queued and is not completing.
+ * @return {@code true} If the task has been queued, {@code false} if the task has not been queued or is marked
+ * as completing.
+ */
+ public boolean isQueued();
+
+ /**
+ * Forces this task to be marked as completed.
+ *
+ * @return {@code true} if the task was cancelled, {@code false} if the task has already completed
+ * or is being completed.
+ */
+ @Override
+ public boolean cancel();
+
+ /**
+ * Executes this task. This will also mark the task as completing.
+ * <p>
+ * Exceptions thrown from the runnable will be rethrown.
+ * </p>
+ *
+ * @return {@code true} if this task was executed, {@code false} if it was already marked as completed.
+ */
+ public boolean execute();
+
+ /**
+ * Returns the current priority. Note that {@link Priority#COMPLETING} will be returned
+ * if this task is completing or has completed.
+ */
+ public Priority getPriority();
+
+ /**
+ * Attempts to set this task's priority level to the level specified.
+ *
+ * @param priority Specified priority level.
+ *
+ * @throws IllegalArgumentException If the priority is invalid
+ * @return {@code true} if successful, {@code false} if this task is completing or has completed or the queue
+ * this task was scheduled on was shutdown, or if the priority was already at the specified level.
+ */
+ public boolean setPriority(final Priority priority);
+
+ /**
+ * Attempts to raise the priority to the priority level specified.
+ *
+ * @param priority Priority specified
+ *
+ * @throws IllegalArgumentException If the priority is invalid
+ * @return {@code false} if the current task is completing, {@code true} if the priority was raised to the
+ * specified level or was already at the specified level or higher.
+ */
+ public boolean raisePriority(final Priority priority);
+
+ /**
+ * Attempts to lower the priority to the priority level specified.
+ *
+ * @param priority Priority specified
+ *
+ * @throws IllegalArgumentException If the priority is invalid
+ * @return {@code false} if the current task is completing, {@code true} if the priority was lowered to the
+ * specified level or was already at the specified level or lower.
+ */
+ public boolean lowerPriority(final Priority priority);
+
+ /**
+ * Returns the suborder id associated with this task.
+ * @return The suborder id associated with this task.
+ */
+ public long getSubOrder();
+
+ /**
+ * Sets the suborder id associated with this task. Ths function has no effect when this task
+ * is completing or is completed.
+ *
+ * @param subOrder Specified new sub order.
+ *
+ * @return {@code true} if successful, {@code false} if this task is completing or has completed or the queue
+ * this task was scheduled on was shutdown, or if the current suborder is the same as the new sub order.
+ */
+ public boolean setSubOrder(final long subOrder);
+
+ /**
+ * Attempts to raise the suborder to the suborder specified.
+ *
+ * @param subOrder Specified new sub order.
+ *
+ * @return {@code false} if the current task is completing, {@code true} if the suborder was raised to the
+ * specified suborder or was already at the specified suborder or higher.
+ */
+ public boolean raiseSubOrder(final long subOrder);
+
+ /**
+ * Attempts to lower the suborder to the suborder specified.
+ *
+ * @param subOrder Specified new sub order.
+ *
+ * @return {@code false} if the current task is completing, {@code true} if the suborder was lowered to the
+ * specified suborder or was already at the specified suborder or lower.
+ */
+ public boolean lowerSubOrder(final long subOrder);
+
+ /**
+ * Sets the priority and suborder id associated with this task. Ths function has no effect when this task
+ * is completing or is completed.
+ *
+ * @param priority Priority specified
+ * @param subOrder Specified new sub order.
+ * @return {@code true} if successful, {@code false} if this task is completing or has completed or the queue
+ * this task was scheduled on was shutdown, or if the current priority and suborder are the same as
+ * the parameters.
+ */
+ public boolean setPriorityAndSubOrder(final Priority priority, final long subOrder);
+ }
+}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/executor/queue/PrioritisedTaskQueue.java b/src/main/java/ca/spottedleaf/concurrentutil/executor/queue/PrioritisedTaskQueue.java
new file mode 100644
index 0000000000000000000000000000000000000000..edb8c6611bdc9aced2714b963e00bbb7829603d2
--- /dev/null
+++ b/src/main/java/ca/spottedleaf/concurrentutil/executor/queue/PrioritisedTaskQueue.java
@@ -0,0 +1,454 @@
+package ca.spottedleaf.concurrentutil.executor.queue;
+
+import ca.spottedleaf.concurrentutil.executor.PrioritisedExecutor;
+import ca.spottedleaf.concurrentutil.util.ConcurrentUtil;
+import ca.spottedleaf.concurrentutil.util.Priority;
+import java.lang.invoke.VarHandle;
+import java.util.Comparator;
+import java.util.Map;
+import java.util.concurrent.ConcurrentSkipListMap;
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.AtomicLong;
+
+public final class PrioritisedTaskQueue implements PrioritisedExecutor {
+
+ /**
+ * Required for tie-breaking in the queue
+ */
+ private final AtomicLong taskIdGenerator = new AtomicLong();
+ private final AtomicLong scheduledTasks = new AtomicLong();
+ private final AtomicLong executedTasks = new AtomicLong();
+ private final AtomicLong subOrderGenerator = new AtomicLong();
+ private final AtomicBoolean shutdown = new AtomicBoolean();
+ private final ConcurrentSkipListMap<PrioritisedQueuedTask.Holder, Boolean> tasks = new ConcurrentSkipListMap<>(PrioritisedQueuedTask.COMPARATOR);
+
+ @Override
+ public long getTotalTasksScheduled() {
+ return this.scheduledTasks.get();
+ }
+
+ @Override
+ public long getTotalTasksExecuted() {
+ return this.executedTasks.get();
+ }
+
+ @Override
+ public long generateNextSubOrder() {
+ return this.subOrderGenerator.getAndIncrement();
+ }
+
+ @Override
+ public boolean shutdown() {
+ return !this.shutdown.getAndSet(true);
+ }
+
+ @Override
+ public boolean isShutdown() {
+ return this.shutdown.get();
+ }
+
+ public PrioritisedTask peekFirst() {
+ final Map.Entry<PrioritisedQueuedTask.Holder, Boolean> firstEntry = this.tasks.firstEntry();
+ return firstEntry == null ? null : firstEntry.getKey().task;
+ }
+
+ public Priority getHighestPriority() {
+ final Map.Entry<PrioritisedQueuedTask.Holder, Boolean> firstEntry = this.tasks.firstEntry();
+ return firstEntry == null ? null : Priority.getPriority(firstEntry.getKey().priority);
+ }
+
+ public boolean hasNoScheduledTasks() {
+ final long executedTasks = this.executedTasks.get();
+ final long scheduledTasks = this.scheduledTasks.get();
+
+ return executedTasks == scheduledTasks;
+ }
+
+ public PrioritySubOrderPair getHighestPrioritySubOrder() {
+ final Map.Entry<PrioritisedQueuedTask.Holder, Boolean> firstEntry = this.tasks.firstEntry();
+ if (firstEntry == null) {
+ return null;
+ }
+
+ final PrioritisedQueuedTask.Holder holder = firstEntry.getKey();
+
+ return new PrioritySubOrderPair(Priority.getPriority(holder.priority), holder.subOrder);
+ }
+
+ public Runnable pollTask() {
+ for (;;) {
+ final Map.Entry<PrioritisedQueuedTask.Holder, Boolean> firstEntry = this.tasks.pollFirstEntry();
+ if (firstEntry != null) {
+ final PrioritisedQueuedTask.Holder task = firstEntry.getKey();
+ task.markRemoved();
+ if (!task.task.cancel()) {
+ continue;
+ }
+ return task.task.execute;
+ }
+
+ return null;
+ }
+ }
+
+ @Override
+ public boolean executeTask() {
+ for (;;) {
+ final Map.Entry<PrioritisedQueuedTask.Holder, Boolean> firstEntry = this.tasks.pollFirstEntry();
+ if (firstEntry != null) {
+ final PrioritisedQueuedTask.Holder task = firstEntry.getKey();
+ task.markRemoved();
+ if (!task.task.execute()) {
+ continue;
+ }
+ return true;
+ }
+
+ return false;
+ }
+ }
+
+ @Override
+ public PrioritisedTask createTask(final Runnable task) {
+ return this.createTask(task, Priority.NORMAL, this.generateNextSubOrder());
+ }
+
+ @Override
+ public PrioritisedTask createTask(final Runnable task, final Priority priority) {
+ return this.createTask(task, priority, this.generateNextSubOrder());
+ }
+
+ @Override
+ public PrioritisedTask createTask(final Runnable task, final Priority priority, final long subOrder) {
+ return new PrioritisedQueuedTask(task, priority, subOrder);
+ }
+
+ @Override
+ public PrioritisedTask queueTask(final Runnable task) {
+ return this.queueTask(task, Priority.NORMAL, this.generateNextSubOrder());
+ }
+
+ @Override
+ public PrioritisedTask queueTask(final Runnable task, final Priority priority) {
+ return this.queueTask(task, priority, this.generateNextSubOrder());
+ }
+
+ @Override
+ public PrioritisedTask queueTask(final Runnable task, final Priority priority, final long subOrder) {
+ final PrioritisedQueuedTask ret = new PrioritisedQueuedTask(task, priority, subOrder);
+
+ ret.queue();
+
+ return ret;
+ }
+
+ private final class PrioritisedQueuedTask implements PrioritisedExecutor.PrioritisedTask {
+ public static final Comparator<PrioritisedQueuedTask.Holder> COMPARATOR = (final PrioritisedQueuedTask.Holder t1, final PrioritisedQueuedTask.Holder t2) -> {
+ final int priorityCompare = t1.priority - t2.priority;
+ if (priorityCompare != 0) {
+ return priorityCompare;
+ }
+
+ final int subOrderCompare = Long.compare(t1.subOrder, t2.subOrder);
+ if (subOrderCompare != 0) {
+ return subOrderCompare;
+ }
+
+ return Long.compare(t1.id, t2.id);
+ };
+
+ private static final class Holder {
+ private final PrioritisedQueuedTask task;
+ private final int priority;
+ private final long subOrder;
+ private final long id;
+
+ private volatile boolean removed;
+ private static final VarHandle REMOVED_HANDLE = ConcurrentUtil.getVarHandle(Holder.class, "removed", boolean.class);
+
+ private Holder(final PrioritisedQueuedTask task, final int priority, final long subOrder,
+ final long id) {
+ this.task = task;
+ this.priority = priority;
+ this.subOrder = subOrder;
+ this.id = id;
+ }
+
+ /**
+ * Returns true if marked as removed
+ */
+ public boolean markRemoved() {
+ return !(boolean)REMOVED_HANDLE.getAndSet((Holder)this, (boolean)true);
+ }
+ }
+
+ private final long id;
+ private final Runnable execute;
+
+ private Priority priority;
+ private long subOrder;
+ private Holder holder;
+
+ public PrioritisedQueuedTask(final Runnable execute, final Priority priority, final long subOrder) {
+ if (!Priority.isValidPriority(priority)) {
+ throw new IllegalArgumentException("Invalid priority " + priority);
+ }
+
+ this.execute = execute;
+ this.priority = priority;
+ this.subOrder = subOrder;
+ this.id = PrioritisedTaskQueue.this.taskIdGenerator.getAndIncrement();
+ }
+
+ @Override
+ public PrioritisedExecutor getExecutor() {
+ return PrioritisedTaskQueue.this;
+ }
+
+ @Override
+ public boolean queue() {
+ synchronized (this) {
+ if (this.holder != null || this.priority == Priority.COMPLETING) {
+ return false;
+ }
+
+ if (PrioritisedTaskQueue.this.isShutdown()) {
+ throw new IllegalStateException("Queue is shutdown");
+ }
+
+ final Holder holder = new Holder(this, this.priority.priority, this.subOrder, this.id);
+ this.holder = holder;
+
+ PrioritisedTaskQueue.this.scheduledTasks.getAndIncrement();
+ PrioritisedTaskQueue.this.tasks.put(holder, Boolean.TRUE);
+ }
+
+ if (PrioritisedTaskQueue.this.isShutdown()) {
+ this.cancel();
+ throw new IllegalStateException("Queue is shutdown");
+ }
+
+
+ return true;
+ }
+
+ @Override
+ public boolean isQueued() {
+ synchronized (this) {
+ return this.holder != null && this.priority != Priority.COMPLETING;
+ }
+ }
+
+ @Override
+ public boolean cancel() {
+ synchronized (this) {
+ if (this.priority == Priority.COMPLETING) {
+ return false;
+ }
+
+ this.priority = Priority.COMPLETING;
+
+ if (this.holder != null) {
+ if (this.holder.markRemoved()) {
+ PrioritisedTaskQueue.this.tasks.remove(this.holder);
+ }
+ PrioritisedTaskQueue.this.executedTasks.getAndIncrement();
+ }
+
+ return true;
+ }
+ }
+
+ @Override
+ public boolean execute() {
+ final boolean increaseExecuted;
+
+ synchronized (this) {
+ if (this.priority == Priority.COMPLETING) {
+ return false;
+ }
+
+ this.priority = Priority.COMPLETING;
+
+ if (increaseExecuted = (this.holder != null)) {
+ if (this.holder.markRemoved()) {
+ PrioritisedTaskQueue.this.tasks.remove(this.holder);
+ }
+ }
+ }
+
+ try {
+ this.execute.run();
+ return true;
+ } finally {
+ if (increaseExecuted) {
+ PrioritisedTaskQueue.this.executedTasks.getAndIncrement();
+ }
+ }
+ }
+
+ @Override
+ public Priority getPriority() {
+ synchronized (this) {
+ return this.priority;
+ }
+ }
+
+ @Override
+ public boolean setPriority(final Priority priority) {
+ synchronized (this) {
+ if (this.priority == Priority.COMPLETING || this.priority == priority) {
+ return false;
+ }
+
+ this.priority = priority;
+
+ if (this.holder != null) {
+ if (this.holder.markRemoved()) {
+ PrioritisedTaskQueue.this.tasks.remove(this.holder);
+ }
+ this.holder = new Holder(this, priority.priority, this.subOrder, this.id);
+ PrioritisedTaskQueue.this.tasks.put(this.holder, Boolean.TRUE);
+ }
+
+ return true;
+ }
+ }
+
+ @Override
+ public boolean raisePriority(final Priority priority) {
+ synchronized (this) {
+ if (this.priority == Priority.COMPLETING || this.priority.isHigherOrEqualPriority(priority)) {
+ return false;
+ }
+
+ this.priority = priority;
+
+ if (this.holder != null) {
+ if (this.holder.markRemoved()) {
+ PrioritisedTaskQueue.this.tasks.remove(this.holder);
+ }
+ this.holder = new Holder(this, priority.priority, this.subOrder, this.id);
+ PrioritisedTaskQueue.this.tasks.put(this.holder, Boolean.TRUE);
+ }
+
+ return true;
+ }
+ }
+
+ @Override
+ public boolean lowerPriority(Priority priority) {
+ synchronized (this) {
+ if (this.priority == Priority.COMPLETING || this.priority.isLowerOrEqualPriority(priority)) {
+ return false;
+ }
+
+ this.priority = priority;
+
+ if (this.holder != null) {
+ if (this.holder.markRemoved()) {
+ PrioritisedTaskQueue.this.tasks.remove(this.holder);
+ }
+ this.holder = new Holder(this, priority.priority, this.subOrder, this.id);
+ PrioritisedTaskQueue.this.tasks.put(this.holder, Boolean.TRUE);
+ }
+
+ return true;
+ }
+ }
+
+ @Override
+ public long getSubOrder() {
+ synchronized (this) {
+ return this.subOrder;
+ }
+ }
+
+ @Override
+ public boolean setSubOrder(final long subOrder) {
+ synchronized (this) {
+ if (this.priority == Priority.COMPLETING || this.subOrder == subOrder) {
+ return false;
+ }
+
+ this.subOrder = subOrder;
+
+ if (this.holder != null) {
+ if (this.holder.markRemoved()) {
+ PrioritisedTaskQueue.this.tasks.remove(this.holder);
+ }
+ this.holder = new Holder(this, priority.priority, this.subOrder, this.id);
+ PrioritisedTaskQueue.this.tasks.put(this.holder, Boolean.TRUE);
+ }
+
+ return true;
+ }
+ }
+
+ @Override
+ public boolean raiseSubOrder(long subOrder) {
+ synchronized (this) {
+ if (this.priority == Priority.COMPLETING || this.subOrder >= subOrder) {
+ return false;
+ }
+
+ this.subOrder = subOrder;
+
+ if (this.holder != null) {
+ if (this.holder.markRemoved()) {
+ PrioritisedTaskQueue.this.tasks.remove(this.holder);
+ }
+ this.holder = new Holder(this, priority.priority, this.subOrder, this.id);
+ PrioritisedTaskQueue.this.tasks.put(this.holder, Boolean.TRUE);
+ }
+
+ return true;
+ }
+ }
+
+ @Override
+ public boolean lowerSubOrder(final long subOrder) {
+ synchronized (this) {
+ if (this.priority == Priority.COMPLETING || this.subOrder <= subOrder) {
+ return false;
+ }
+
+ this.subOrder = subOrder;
+
+ if (this.holder != null) {
+ if (this.holder.markRemoved()) {
+ PrioritisedTaskQueue.this.tasks.remove(this.holder);
+ }
+ this.holder = new Holder(this, priority.priority, this.subOrder, this.id);
+ PrioritisedTaskQueue.this.tasks.put(this.holder, Boolean.TRUE);
+ }
+
+ return true;
+ }
+ }
+
+ @Override
+ public boolean setPriorityAndSubOrder(final Priority priority, final long subOrder) {
+ synchronized (this) {
+ if (this.priority == Priority.COMPLETING || (this.priority == priority && this.subOrder == subOrder)) {
+ return false;
+ }
+
+ this.priority = priority;
+ this.subOrder = subOrder;
+
+ if (this.holder != null) {
+ if (this.holder.markRemoved()) {
+ PrioritisedTaskQueue.this.tasks.remove(this.holder);
+ }
+ this.holder = new Holder(this, priority.priority, this.subOrder, this.id);
+ PrioritisedTaskQueue.this.tasks.put(this.holder, Boolean.TRUE);
+ }
+
+ return true;
+ }
+ }
+ }
+
+ public static record PrioritySubOrderPair(Priority priority, long subOrder) {}
+}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/executor/standard/DelayedPrioritisedTask.java b/src/main/java/ca/spottedleaf/concurrentutil/executor/standard/DelayedPrioritisedTask.java
deleted file mode 100644
index 3ce10053d4ec51855ad7012abb5d97df1c0e557a..0000000000000000000000000000000000000000
--- a/src/main/java/ca/spottedleaf/concurrentutil/executor/standard/DelayedPrioritisedTask.java
+++ /dev/null
@@ -1,170 +0,0 @@
-package ca.spottedleaf.concurrentutil.executor.standard;
-
-import ca.spottedleaf.concurrentutil.util.ConcurrentUtil;
-import java.lang.invoke.VarHandle;
-
-public class DelayedPrioritisedTask {
-
- protected volatile int priority;
- protected static final VarHandle PRIORITY_HANDLE = ConcurrentUtil.getVarHandle(DelayedPrioritisedTask.class, "priority", int.class);
-
- protected static final int PRIORITY_SET = Integer.MIN_VALUE >>> 0;
-
- protected final int getPriorityVolatile() {
- return (int)PRIORITY_HANDLE.getVolatile((DelayedPrioritisedTask)this);
- }
-
- protected final int compareAndExchangePriorityVolatile(final int expect, final int update) {
- return (int)PRIORITY_HANDLE.compareAndExchange((DelayedPrioritisedTask)this, (int)expect, (int)update);
- }
-
- protected final int getAndOrPriorityVolatile(final int val) {
- return (int)PRIORITY_HANDLE.getAndBitwiseOr((DelayedPrioritisedTask)this, (int)val);
- }
-
- protected final void setPriorityPlain(final int val) {
- PRIORITY_HANDLE.set((DelayedPrioritisedTask)this, (int)val);
- }
-
- protected volatile PrioritisedExecutor.PrioritisedTask task;
- protected static final VarHandle TASK_HANDLE = ConcurrentUtil.getVarHandle(DelayedPrioritisedTask.class, "task", PrioritisedExecutor.PrioritisedTask.class);
-
- protected PrioritisedExecutor.PrioritisedTask getTaskPlain() {
- return (PrioritisedExecutor.PrioritisedTask)TASK_HANDLE.get((DelayedPrioritisedTask)this);
- }
-
- protected PrioritisedExecutor.PrioritisedTask getTaskVolatile() {
- return (PrioritisedExecutor.PrioritisedTask)TASK_HANDLE.getVolatile((DelayedPrioritisedTask)this);
- }
-
- protected final PrioritisedExecutor.PrioritisedTask compareAndExchangeTaskVolatile(final PrioritisedExecutor.PrioritisedTask expect, final PrioritisedExecutor.PrioritisedTask update) {
- return (PrioritisedExecutor.PrioritisedTask)TASK_HANDLE.compareAndExchange((DelayedPrioritisedTask)this, (PrioritisedExecutor.PrioritisedTask)expect, (PrioritisedExecutor.PrioritisedTask)update);
- }
-
- public DelayedPrioritisedTask(final PrioritisedExecutor.Priority priority) {
- this.setPriorityPlain(priority.priority);
- }
-
- // only public for debugging
- public int getPriorityInternal() {
- return this.getPriorityVolatile();
- }
-
- public PrioritisedExecutor.PrioritisedTask getTask() {
- return this.getTaskVolatile();
- }
-
- public void setTask(final PrioritisedExecutor.PrioritisedTask task) {
- int priority = this.getPriorityVolatile();
-
- if (this.compareAndExchangeTaskVolatile(null, task) != null) {
- throw new IllegalStateException("setTask() called twice");
- }
-
- int failures = 0;
- for (;;) {
- task.setPriority(PrioritisedExecutor.Priority.getPriority(priority));
-
- if (priority == (priority = this.compareAndExchangePriorityVolatile(priority, priority | PRIORITY_SET))) {
- return;
- }
-
- ++failures;
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
- }
- }
-
- public PrioritisedExecutor.Priority getPriority() {
- final int priority = this.getPriorityVolatile();
- if ((priority & PRIORITY_SET) != 0) {
- return this.task.getPriority();
- }
-
- return PrioritisedExecutor.Priority.getPriority(priority);
- }
-
- public void raisePriority(final PrioritisedExecutor.Priority priority) {
- if (!PrioritisedExecutor.Priority.isValidPriority(priority)) {
- throw new IllegalArgumentException("Invalid priority " + priority);
- }
-
- int failures = 0;
- for (int curr = this.getPriorityVolatile();;) {
- if ((curr & PRIORITY_SET) != 0) {
- this.getTaskPlain().raisePriority(priority);
- return;
- }
-
- if (!priority.isLowerPriority(curr)) {
- return;
- }
-
- if (curr == (curr = this.compareAndExchangePriorityVolatile(curr, priority.priority))) {
- return;
- }
-
- // failed, retry
-
- ++failures;
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
- }
- }
-
- public void setPriority(final PrioritisedExecutor.Priority priority) {
- if (!PrioritisedExecutor.Priority.isValidPriority(priority)) {
- throw new IllegalArgumentException("Invalid priority " + priority);
- }
-
- int failures = 0;
- for (int curr = this.getPriorityVolatile();;) {
- if ((curr & PRIORITY_SET) != 0) {
- this.getTaskPlain().setPriority(priority);
- return;
- }
-
- if (curr == (curr = this.compareAndExchangePriorityVolatile(curr, priority.priority))) {
- return;
- }
-
- // failed, retry
-
- ++failures;
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
- }
- }
-
- public void lowerPriority(final PrioritisedExecutor.Priority priority) {
- if (!PrioritisedExecutor.Priority.isValidPriority(priority)) {
- throw new IllegalArgumentException("Invalid priority " + priority);
- }
-
- int failures = 0;
- for (int curr = this.getPriorityVolatile();;) {
- if ((curr & PRIORITY_SET) != 0) {
- this.getTaskPlain().lowerPriority(priority);
- return;
- }
-
- if (!priority.isHigherPriority(curr)) {
- return;
- }
-
- if (curr == (curr = this.compareAndExchangePriorityVolatile(curr, priority.priority))) {
- return;
- }
-
- // failed, retry
-
- ++failures;
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
- }
- }
-}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/executor/standard/PrioritisedExecutor.java b/src/main/java/ca/spottedleaf/concurrentutil/executor/standard/PrioritisedExecutor.java
deleted file mode 100644
index 91beb6f23f257cf265fe3150f760892e605f217a..0000000000000000000000000000000000000000
--- a/src/main/java/ca/spottedleaf/concurrentutil/executor/standard/PrioritisedExecutor.java
+++ /dev/null
@@ -1,276 +0,0 @@
-package ca.spottedleaf.concurrentutil.executor.standard;
-
-import ca.spottedleaf.concurrentutil.executor.BaseExecutor;
-
-/**
- * Implementation of {@link BaseExecutor} which schedules tasks to be executed by a given priority.
- * @see BaseExecutor
- */
-public interface PrioritisedExecutor extends BaseExecutor {
-
- public static enum Priority {
-
- /**
- * Priority value indicating the task has completed or is being completed.
- * This priority cannot be used to schedule tasks.
- */
- COMPLETING(-1),
-
- /**
- * Absolute highest priority, should only be used for when a task is blocking a time-critical thread.
- */
- BLOCKING(),
-
- /**
- * Should only be used for urgent but not time-critical tasks.
- */
- HIGHEST(),
-
- /**
- * Two priorities above normal.
- */
- HIGHER(),
-
- /**
- * One priority above normal.
- */
- HIGH(),
-
- /**
- * Default priority.
- */
- NORMAL(),
-
- /**
- * One priority below normal.
- */
- LOW(),
-
- /**
- * Two priorities below normal.
- */
- LOWER(),
-
- /**
- * Use for tasks that should eventually execute, but are not needed to.
- */
- LOWEST(),
-
- /**
- * Use for tasks that can be delayed indefinitely.
- */
- IDLE();
-
- // returns whether the priority can be scheduled
- public static boolean isValidPriority(final Priority priority) {
- return priority != null && priority != Priority.COMPLETING;
- }
-
- // returns the higher priority of the two
- public static Priority max(final Priority p1, final Priority p2) {
- return p1.isHigherOrEqualPriority(p2) ? p1 : p2;
- }
-
- // returns the lower priroity of the two
- public static Priority min(final Priority p1, final Priority p2) {
- return p1.isLowerOrEqualPriority(p2) ? p1 : p2;
- }
-
- public boolean isHigherOrEqualPriority(final Priority than) {
- return this.priority <= than.priority;
- }
-
- public boolean isHigherPriority(final Priority than) {
- return this.priority < than.priority;
- }
-
- public boolean isLowerOrEqualPriority(final Priority than) {
- return this.priority >= than.priority;
- }
-
- public boolean isLowerPriority(final Priority than) {
- return this.priority > than.priority;
- }
-
- public boolean isHigherOrEqualPriority(final int than) {
- return this.priority <= than;
- }
-
- public boolean isHigherPriority(final int than) {
- return this.priority < than;
- }
-
- public boolean isLowerOrEqualPriority(final int than) {
- return this.priority >= than;
- }
-
- public boolean isLowerPriority(final int than) {
- return this.priority > than;
- }
-
- public static boolean isHigherOrEqualPriority(final int priority, final int than) {
- return priority <= than;
- }
-
- public static boolean isHigherPriority(final int priority, final int than) {
- return priority < than;
- }
-
- public static boolean isLowerOrEqualPriority(final int priority, final int than) {
- return priority >= than;
- }
-
- public static boolean isLowerPriority(final int priority, final int than) {
- return priority > than;
- }
-
- static final Priority[] PRIORITIES = Priority.values();
-
- /** includes special priorities */
- public static final int TOTAL_PRIORITIES = PRIORITIES.length;
-
- public static final int TOTAL_SCHEDULABLE_PRIORITIES = TOTAL_PRIORITIES - 1;
-
- public static Priority getPriority(final int priority) {
- return PRIORITIES[priority + 1];
- }
-
- private static int priorityCounter;
-
- private static int nextCounter() {
- return priorityCounter++;
- }
-
- public final int priority;
-
- Priority() {
- this(nextCounter());
- }
-
- Priority(final int priority) {
- this.priority = priority;
- }
- }
-
- /**
- * Executes the next available task.
- * <p>
- * If there is a task with priority {@link PrioritisedExecutor.Priority#BLOCKING} available, then that such task is executed.
- * </p>
- * <p>
- * If there is a task with priority {@link PrioritisedExecutor.Priority#IDLE} available then that task is only executed
- * when there are no other tasks available with a higher priority.
- * </p>
- * <p>
- * If there are no tasks that have priority {@link PrioritisedExecutor.Priority#BLOCKING} or {@link PrioritisedExecutor.Priority#IDLE}, then
- * this function will be biased to execute tasks that have higher priorities.
- * </p>
- *
- * @return {@code true} if a task was executed, {@code false} otherwise
- * @throws IllegalStateException If the current thread is not allowed to execute a task
- */
- @Override
- public boolean executeTask() throws IllegalStateException;
-
- /**
- * Queues or executes a task at {@link Priority#NORMAL} priority.
- * @param task The task to run.
- *
- * @throws IllegalStateException If this queue has shutdown.
- * @throws NullPointerException If the task is null
- * @return {@code null} if the current thread immediately executed the task, else returns the prioritised task
- * associated with the parameter
- */
- public default PrioritisedTask queueRunnable(final Runnable task) {
- return this.queueRunnable(task, Priority.NORMAL);
- }
-
- /**
- * Queues or executes a task.
- *
- * @param task The task to run.
- * @param priority The priority for the task.
- *
- * @throws IllegalStateException If this queue has shutdown.
- * @throws NullPointerException If the task is null
- * @throws IllegalArgumentException If the priority is invalid.
- * @return {@code null} if the current thread immediately executed the task, else returns the prioritised task
- * associated with the parameter
- */
- public PrioritisedTask queueRunnable(final Runnable task, final Priority priority);
-
- /**
- * Creates, but does not execute or queue the task. The task must later be queued via {@link BaseTask#queue()}.
- *
- * @param task The task to run.
- *
- * @throws IllegalStateException If this queue has shutdown.
- * @throws NullPointerException If the task is null
- * @throws IllegalArgumentException If the priority is invalid.
- * @throws UnsupportedOperationException If this executor does not support lazily queueing tasks
- * @return The prioritised task associated with the parameters
- */
- public default PrioritisedTask createTask(final Runnable task) {
- return this.createTask(task, Priority.NORMAL);
- }
-
- /**
- * Creates, but does not execute or queue the task. The task must later be queued via {@link BaseTask#queue()}.
- *
- * @param task The task to run.
- * @param priority The priority for the task.
- *
- * @throws IllegalStateException If this queue has shutdown.
- * @throws NullPointerException If the task is null
- * @throws IllegalArgumentException If the priority is invalid.
- * @throws UnsupportedOperationException If this executor does not support lazily queueing tasks
- * @return The prioritised task associated with the parameters
- */
- public PrioritisedTask createTask(final Runnable task, final Priority priority);
-
- /**
- * Extension of {@link ca.spottedleaf.concurrentutil.executor.BaseExecutor.BaseTask} which adds functions
- * to retrieve and modify the task's associated priority.
- *
- * @see ca.spottedleaf.concurrentutil.executor.BaseExecutor.BaseTask
- */
- public static interface PrioritisedTask extends BaseTask {
-
- /**
- * Returns the current priority. Note that {@link Priority#COMPLETING} will be returned
- * if this task is completing or has completed.
- */
- public Priority getPriority();
-
- /**
- * Attempts to set this task's priority level to the level specified.
- *
- * @param priority Specified priority level.
- *
- * @throws IllegalArgumentException If the priority is invalid
- * @return {@code true} if successful, {@code false} if this task is completing or has completed or the queue
- * this task was scheduled on was shutdown, or if the priority was already at the specified level.
- */
- public boolean setPriority(final Priority priority);
-
- /**
- * Attempts to raise the priority to the priority level specified.
- *
- * @param priority Priority specified
- *
- * @throws IllegalArgumentException If the priority is invalid
- * @return {@code false} if the current task is completing, {@code true} if the priority was raised to the specified level or was already at the specified level or higher.
- */
- public boolean raisePriority(final Priority priority);
-
- /**
- * Attempts to lower the priority to the priority level specified.
- *
- * @param priority Priority specified
- *
- * @throws IllegalArgumentException If the priority is invalid
- * @return {@code false} if the current task is completing, {@code true} if the priority was lowered to the specified level or was already at the specified level or lower.
- */
- public boolean lowerPriority(final Priority priority);
- }
-}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/executor/standard/PrioritisedThreadPool.java b/src/main/java/ca/spottedleaf/concurrentutil/executor/standard/PrioritisedThreadPool.java
deleted file mode 100644
index 2ba36e29d0d8693f2f5e6c6d195ca27f2a5099aa..0000000000000000000000000000000000000000
--- a/src/main/java/ca/spottedleaf/concurrentutil/executor/standard/PrioritisedThreadPool.java
+++ /dev/null
@@ -1,632 +0,0 @@
-package ca.spottedleaf.concurrentutil.executor.standard;
-
-import it.unimi.dsi.fastutil.objects.ReferenceOpenHashSet;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.Comparator;
-import java.util.TreeSet;
-import java.util.concurrent.atomic.AtomicBoolean;
-import java.util.function.BiConsumer;
-
-public final class PrioritisedThreadPool {
-
- private static final Logger LOGGER = LoggerFactory.getLogger(PrioritisedThreadPool.class);
-
- private final PrioritisedThread[] threads;
- private final TreeSet<PrioritisedPoolExecutorImpl> queues = new TreeSet<>(PrioritisedPoolExecutorImpl.comparator());
- private final String name;
- private final long queueMaxHoldTime;
-
- private final ReferenceOpenHashSet<PrioritisedPoolExecutorImpl> nonShutdownQueues = new ReferenceOpenHashSet<>();
- private final ReferenceOpenHashSet<PrioritisedPoolExecutorImpl> activeQueues = new ReferenceOpenHashSet<>();
-
- private boolean shutdown;
-
- private long schedulingIdGenerator;
-
- private static final long DEFAULT_QUEUE_HOLD_TIME = (long)(5.0e6);
-
- /**
- * @param name Specified debug name of this thread pool
- * @param threads The number of threads to use
- */
- public PrioritisedThreadPool(final String name, final int threads) {
- this(name, threads, null);
- }
-
- /**
- * @param name Specified debug name of this thread pool
- * @param threads The number of threads to use
- * @param threadModifier Invoked for each created thread with its incremental id before starting them
- */
- public PrioritisedThreadPool(final String name, final int threads, final BiConsumer<Thread, Integer> threadModifier) {
- this(name, threads, threadModifier, DEFAULT_QUEUE_HOLD_TIME); // 5ms
- }
-
- /**
- * @param name Specified debug name of this thread pool
- * @param threads The number of threads to use
- * @param threadModifier Invoked for each created thread with its incremental id before starting them
- * @param queueHoldTime The maximum amount of time to spend executing tasks in a specific queue before attempting
- * to switch to another queue, per thread
- */
- public PrioritisedThreadPool(final String name, final int threads, final BiConsumer<Thread, Integer> threadModifier,
- final long queueHoldTime) { // in ns
- if (threads <= 0) {
- throw new IllegalArgumentException("Thread count must be > 0, not " + threads);
- }
- if (name == null) {
- throw new IllegalArgumentException("Name cannot be null");
- }
- this.name = name;
- this.queueMaxHoldTime = queueHoldTime;
-
- this.threads = new PrioritisedThread[threads];
- for (int i = 0; i < threads; ++i) {
- this.threads[i] = new PrioritisedThread(this);
-
- // set default attributes
- this.threads[i].setName("Prioritised thread for pool '" + name + "' #" + i);
- this.threads[i].setUncaughtExceptionHandler((final Thread thread, final Throwable throwable) -> {
- LOGGER.error("Uncaught exception in thread " + thread.getName(), throwable);
- });
-
- // let thread modifier override defaults
- if (threadModifier != null) {
- threadModifier.accept(this.threads[i], Integer.valueOf(i));
- }
-
- // now the thread can start
- this.threads[i].start();
- }
- }
-
- /**
- * Returns an array representing the threads backing this thread pool.
- */
- public Thread[] getThreads() {
- return Arrays.copyOf(this.threads, this.threads.length, Thread[].class);
- }
-
- /**
- * Creates and returns a {@link PrioritisedPoolExecutor} to schedule tasks onto. The returned executor will execute
- * tasks on this thread pool only.
- * @param name The debug name of the executor.
- * @param minParallelism The minimum number of threads to be executing tasks from the returned executor
- * before threads may be allocated to other queues in this thread pool.
- * @param parallelism The maximum number of threads which may be executing tasks from the returned executor.
- * @throws IllegalStateException If this thread pool is shut down
- */
- public PrioritisedPoolExecutor createExecutor(final String name, final int minParallelism, final int parallelism) {
- synchronized (this.nonShutdownQueues) {
- if (this.shutdown) {
- throw new IllegalStateException("Queue is shutdown: " + this.toString());
- }
- final PrioritisedPoolExecutorImpl ret = new PrioritisedPoolExecutorImpl(
- this, name,
- Math.min(Math.max(1, parallelism), this.threads.length),
- Math.min(Math.max(0, minParallelism), this.threads.length)
- );
-
- this.nonShutdownQueues.add(ret);
-
- synchronized (this.activeQueues) {
- this.activeQueues.add(ret);
- }
-
- return ret;
- }
- }
-
- /**
- * Prevents creation of new queues, shutdowns all non-shutdown queues if specified
- */
- public void halt(final boolean shutdownQueues) {
- synchronized (this.nonShutdownQueues) {
- this.shutdown = true;
- }
- if (shutdownQueues) {
- final ArrayList<PrioritisedPoolExecutorImpl> queuesToShutdown;
- synchronized (this.nonShutdownQueues) {
- this.shutdown = true;
- queuesToShutdown = new ArrayList<>(this.nonShutdownQueues);
- }
-
- for (final PrioritisedPoolExecutorImpl queue : queuesToShutdown) {
- queue.shutdown();
- }
- }
-
-
- for (final PrioritisedThread thread : this.threads) {
- // can't kill queue, queue is null
- thread.halt(false);
- }
- }
-
- /**
- * Waits until all threads in this pool have shutdown, or until the specified time has passed.
- * @param msToWait Maximum time to wait.
- * @return {@code false} if the maximum time passed, {@code true} otherwise.
- */
- public boolean join(final long msToWait) {
- try {
- return this.join(msToWait, false);
- } catch (final InterruptedException ex) {
- throw new IllegalStateException(ex);
- }
- }
-
- /**
- * Waits until all threads in this pool have shutdown, or until the specified time has passed.
- * @param msToWait Maximum time to wait.
- * @return {@code false} if the maximum time passed, {@code true} otherwise.
- * @throws InterruptedException If this thread is interrupted.
- */
- public boolean joinInterruptable(final long msToWait) throws InterruptedException {
- return this.join(msToWait, true);
- }
-
- protected final boolean join(final long msToWait, final boolean interruptable) throws InterruptedException {
- final long nsToWait = msToWait * (1000 * 1000);
- final long start = System.nanoTime();
- final long deadline = start + nsToWait;
- boolean interrupted = false;
- try {
- for (final PrioritisedThread thread : this.threads) {
- for (;;) {
- if (!thread.isAlive()) {
- break;
- }
- final long current = System.nanoTime();
- if (current >= deadline) {
- return false;
- }
-
- try {
- thread.join(Math.max(1L, (deadline - current) / (1000 * 1000)));
- } catch (final InterruptedException ex) {
- if (interruptable) {
- throw ex;
- }
- interrupted = true;
- }
- }
- }
-
- return true;
- } finally {
- if (interrupted) {
- Thread.currentThread().interrupt();
- }
- }
- }
-
- /**
- * Shuts down this thread pool, optionally waiting for all tasks to be executed.
- * This function will invoke {@link PrioritisedPoolExecutor#shutdown()} on all created executors on this
- * thread pool.
- * @param wait Whether to wait for tasks to be executed
- */
- public void shutdown(final boolean wait) {
- final ArrayList<PrioritisedPoolExecutorImpl> queuesToShutdown;
- synchronized (this.nonShutdownQueues) {
- this.shutdown = true;
- queuesToShutdown = new ArrayList<>(this.nonShutdownQueues);
- }
-
- for (final PrioritisedPoolExecutorImpl queue : queuesToShutdown) {
- queue.shutdown();
- }
-
- for (final PrioritisedThread thread : this.threads) {
- // none of these can be true or else NPE
- thread.close(false, false);
- }
-
- if (wait) {
- final ArrayList<PrioritisedPoolExecutorImpl> queues;
- synchronized (this.activeQueues) {
- queues = new ArrayList<>(this.activeQueues);
- }
- for (final PrioritisedPoolExecutorImpl queue : queues) {
- queue.waitUntilAllExecuted();
- }
- }
- }
-
- protected static final class PrioritisedThread extends PrioritisedQueueExecutorThread {
-
- protected final PrioritisedThreadPool pool;
- protected final AtomicBoolean alertedHighPriority = new AtomicBoolean();
-
- public PrioritisedThread(final PrioritisedThreadPool pool) {
- super(null);
- this.pool = pool;
- }
-
- public boolean alertHighPriorityExecutor() {
- if (!this.notifyTasks()) {
- if (!this.alertedHighPriority.get()) {
- this.alertedHighPriority.set(true);
- }
- return false;
- }
-
- return true;
- }
-
- private boolean isAlertedHighPriority() {
- return this.alertedHighPriority.get() && this.alertedHighPriority.getAndSet(false);
- }
-
- @Override
- protected boolean pollTasks() {
- final PrioritisedThreadPool pool = this.pool;
- final TreeSet<PrioritisedPoolExecutorImpl> queues = this.pool.queues;
-
- boolean ret = false;
- for (;;) {
- if (this.halted) {
- break;
- }
- // try to find a queue
- // note that if and ONLY IF the queues set is empty, this means there are no tasks for us to execute.
- // so we can only break when it's empty
- final PrioritisedPoolExecutorImpl queue;
- // select queue
- synchronized (queues) {
- queue = queues.pollFirst();
- if (queue == null) {
- // no tasks to execute
- break;
- }
-
- queue.schedulingId = ++pool.schedulingIdGenerator;
- // we own this queue now, so increment the executor count
- // do we also need to push this queue up for grabs for another executor?
- if (++queue.concurrentExecutors < queue.maximumExecutors) {
- // re-add to queues
- // it's very important this is done in the same synchronised block for polling, as this prevents
- // us from possibly later adding a queue that should not exist in the set
- queues.add(queue);
- queue.isQueued = true;
- } else {
- queue.isQueued = false;
- }
- // note: we cannot drain entries from the queue while holding this lock, as it will cause deadlock
- // the queue addition holds the per-queue lock first then acquires the lock we have now, but if we
- // try to poll now we don't hold the per queue lock but we do hold the global lock...
- }
-
- // parse tasks as long as we are allowed
- final long start = System.nanoTime();
- final long deadline = start + pool.queueMaxHoldTime;
- do {
- try {
- if (this.halted) {
- break;
- }
- if (!queue.executeTask()) {
- // no more tasks, try next queue
- break;
- }
- ret = true;
- } catch (final ThreadDeath death) {
- throw death; // goodbye world...
- } catch (final Throwable throwable) {
- LOGGER.error("Exception thrown from thread '" + this.getName() + "' in queue '" + queue.toString() + "'", throwable);
- }
- } while (!this.isAlertedHighPriority() && System.nanoTime() <= deadline);
-
- synchronized (queues) {
- // decrement executors, we are no longer executing
- if (queue.isQueued) {
- queues.remove(queue);
- queue.isQueued = false;
- }
- if (--queue.concurrentExecutors == 0 && queue.scheduledPriority == null) {
- // reset scheduling id once the queue is empty again
- // this will ensure empty queues are not prioritised suddenly over active queues once tasks are
- // queued
- queue.schedulingId = 0L;
- }
-
- // ensure the executor is queued for execution again
- if (!queue.isHalted && queue.scheduledPriority != null) { // make sure it actually has tasks
- queues.add(queue);
- queue.isQueued = true;
- }
- }
- }
-
- return ret;
- }
- }
-
- public interface PrioritisedPoolExecutor extends PrioritisedExecutor {
-
- /**
- * Removes this queue from the thread pool without shutting the queue down or waiting for queued tasks to be executed
- */
- public void halt();
-
- /**
- * Returns whether this executor is scheduled to run tasks or is running tasks, otherwise it returns whether
- * this queue is not halted and not shutdown.
- */
- public boolean isActive();
- }
-
- protected static final class PrioritisedPoolExecutorImpl extends PrioritisedThreadedTaskQueue implements PrioritisedPoolExecutor {
-
- protected final PrioritisedThreadPool pool;
- protected final long[] priorityCounts = new long[Priority.TOTAL_SCHEDULABLE_PRIORITIES];
- protected long schedulingId;
- protected int concurrentExecutors;
- protected Priority scheduledPriority;
-
- protected final String name;
- protected final int maximumExecutors;
- protected final int minimumExecutors;
- protected boolean isQueued;
-
- public PrioritisedPoolExecutorImpl(final PrioritisedThreadPool pool, final String name, final int maximumExecutors, final int minimumExecutors) {
- this.pool = pool;
- this.name = name;
- this.maximumExecutors = maximumExecutors;
- this.minimumExecutors = minimumExecutors;
- }
-
- public static Comparator<PrioritisedPoolExecutorImpl> comparator() {
- return (final PrioritisedPoolExecutorImpl p1, final PrioritisedPoolExecutorImpl p2) -> {
- if (p1 == p2) {
- return 0;
- }
-
- final int belowMin1 = p1.minimumExecutors - p1.concurrentExecutors;
- final int belowMin2 = p2.minimumExecutors - p2.concurrentExecutors;
-
- // test minimum executors
- if (belowMin1 > 0 || belowMin2 > 0) {
- // want the largest belowMin to be first
- final int minCompare = Integer.compare(belowMin2, belowMin1);
-
- if (minCompare != 0) {
- return minCompare;
- }
- }
-
- // prefer higher priority
- final int priorityCompare = p1.scheduledPriority.ordinal() - p2.scheduledPriority.ordinal();
- if (priorityCompare != 0) {
- return priorityCompare;
- }
-
- // try to spread out the executors so that each can have threads executing
- final int executorCompare = p1.concurrentExecutors - p2.concurrentExecutors;
- if (executorCompare != 0) {
- return executorCompare;
- }
-
- // if all else fails here we just choose whichever executor was queued first
- return Long.compare(p1.schedulingId, p2.schedulingId);
- };
- }
-
- private boolean isHalted;
-
- @Override
- public void halt() {
- final PrioritisedThreadPool pool = this.pool;
- final TreeSet<PrioritisedPoolExecutorImpl> queues = pool.queues;
- synchronized (queues) {
- if (this.isHalted) {
- return;
- }
- this.isHalted = true;
- if (this.isQueued) {
- queues.remove(this);
- this.isQueued = false;
- }
- }
- synchronized (pool.nonShutdownQueues) {
- pool.nonShutdownQueues.remove(this);
- }
- synchronized (pool.activeQueues) {
- pool.activeQueues.remove(this);
- }
- }
-
- @Override
- public boolean isActive() {
- final PrioritisedThreadPool pool = this.pool;
- final TreeSet<PrioritisedPoolExecutorImpl> queues = pool.queues;
-
- synchronized (queues) {
- if (this.concurrentExecutors != 0) {
- return true;
- }
- synchronized (pool.activeQueues) {
- if (pool.activeQueues.contains(this)) {
- return true;
- }
- }
- }
-
- return false;
- }
-
- private long totalQueuedTasks = 0L;
-
- @Override
- protected void priorityChange(final PrioritisedThreadedTaskQueue.PrioritisedTask task, final Priority from, final Priority to) {
- // Note: The superclass' queue lock is ALWAYS held when inside this method. So we do NOT need to do any additional synchronisation
- // for accessing this queue's state.
- final long[] priorityCounts = this.priorityCounts;
- final boolean shutdown = this.isShutdown();
-
- if (from == null && to == Priority.COMPLETING) {
- throw new IllegalStateException("Cannot complete task without queueing it first");
- }
-
- // we should only notify for queueing of tasks, not changing priorities
- final boolean shouldNotifyTasks = from == null;
-
- final Priority scheduledPriority = this.scheduledPriority;
- if (from != null) {
- --priorityCounts[from.priority];
- }
- if (to != Priority.COMPLETING) {
- ++priorityCounts[to.priority];
- }
- final long totalQueuedTasks;
- if (to == Priority.COMPLETING) {
- totalQueuedTasks = --this.totalQueuedTasks;
- } else if (from == null) {
- totalQueuedTasks = ++this.totalQueuedTasks;
- } else {
- totalQueuedTasks = this.totalQueuedTasks;
- }
-
- // find new highest priority
- int highest = Math.min(to == Priority.COMPLETING ? Priority.IDLE.priority : to.priority, scheduledPriority == null ? Priority.IDLE.priority : scheduledPriority.priority);
- int lowestPriority = priorityCounts.length; // exclusive
- for (;highest < lowestPriority; ++highest) {
- final long count = priorityCounts[highest];
- if (count < 0) {
- throw new IllegalStateException("Priority " + highest + " has " + count + " scheduled tasks");
- }
-
- if (count != 0) {
- break;
- }
- }
-
- final Priority newPriority;
- if (highest == lowestPriority) {
- // no tasks left
- newPriority = null;
- } else if (shutdown) {
- // whichever is lower, the actual greatest priority or simply HIGHEST
- // this is so shutdown automatically gets priority
- newPriority = Priority.getPriority(Math.min(highest, Priority.HIGHEST.priority));
- } else {
- newPriority = Priority.getPriority(highest);
- }
-
- final int executorsWanted;
- boolean shouldNotifyHighPriority = false;
-
- final PrioritisedThreadPool pool = this.pool;
- final TreeSet<PrioritisedPoolExecutorImpl> queues = pool.queues;
-
- synchronized (queues) {
- if (!this.isQueued) {
- // see if we need to be queued
- if (newPriority != null) {
- if (this.schedulingId == 0L) {
- this.schedulingId = ++pool.schedulingIdGenerator;
- }
- this.scheduledPriority = newPriority; // must be updated before queue add
- if (!this.isHalted && this.concurrentExecutors < this.maximumExecutors) {
- shouldNotifyHighPriority = newPriority.isHigherOrEqualPriority(Priority.HIGH);
- queues.add(this);
- this.isQueued = true;
- }
- } else {
- // do not queue
- this.scheduledPriority = null;
- }
- } else {
- // see if we need to NOT be queued
- if (newPriority == null) {
- queues.remove(this);
- this.scheduledPriority = null;
- this.isQueued = false;
- } else if (scheduledPriority != newPriority) {
- // if our priority changed, we need to update it - which means removing and re-adding into the queue
- queues.remove(this);
- // only now can we update scheduledPriority, since we are no longer in queue
- this.scheduledPriority = newPriority;
- queues.add(this);
- shouldNotifyHighPriority = (scheduledPriority == null || scheduledPriority.isLowerPriority(Priority.HIGH)) && newPriority.isHigherOrEqualPriority(Priority.HIGH);
- }
- }
-
- if (this.isQueued) {
- executorsWanted = Math.min(this.maximumExecutors - this.concurrentExecutors, (int)totalQueuedTasks);
- } else {
- executorsWanted = 0;
- }
- }
-
- if (newPriority == null && shutdown) {
- synchronized (pool.activeQueues) {
- pool.activeQueues.remove(this);
- }
- }
-
- // Wake up the number of executors we want
- if (executorsWanted > 0 || (shouldNotifyTasks | shouldNotifyHighPriority)) {
- int notified = 0;
- for (final PrioritisedThread thread : pool.threads) {
- if ((shouldNotifyHighPriority ? thread.alertHighPriorityExecutor() : thread.notifyTasks())
- && (++notified >= executorsWanted)) {
- break;
- }
- }
- }
- }
-
- @Override
- public boolean shutdown() {
- final boolean ret = super.shutdown();
- if (!ret) {
- return ret;
- }
-
- final PrioritisedThreadPool pool = this.pool;
-
- // remove from active queues
- synchronized (pool.nonShutdownQueues) {
- pool.nonShutdownQueues.remove(this);
- }
-
- final TreeSet<PrioritisedPoolExecutorImpl> queues = pool.queues;
-
- // try and shift around our priority
- synchronized (queues) {
- if (this.scheduledPriority == null) {
- // no tasks are queued, ensure we aren't in activeQueues
- synchronized (pool.activeQueues) {
- pool.activeQueues.remove(this);
- }
-
- return ret;
- }
-
- // try to set scheduled priority to HIGHEST so it drains faster
-
- if (this.scheduledPriority.isHigherOrEqualPriority(Priority.HIGHEST)) {
- // already at target priority (highest or above)
- return ret;
- }
-
- // shift priority to HIGHEST
-
- if (this.isQueued) {
- queues.remove(this);
- this.scheduledPriority = Priority.HIGHEST;
- queues.add(this);
- } else {
- this.scheduledPriority = Priority.HIGHEST;
- }
- }
-
- return ret;
- }
- }
-}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/executor/standard/PrioritisedThreadedTaskQueue.java b/src/main/java/ca/spottedleaf/concurrentutil/executor/standard/PrioritisedThreadedTaskQueue.java
deleted file mode 100644
index 3e8401b1b1f833c4f01bc87059a2f48d761d989f..0000000000000000000000000000000000000000
--- a/src/main/java/ca/spottedleaf/concurrentutil/executor/standard/PrioritisedThreadedTaskQueue.java
+++ /dev/null
@@ -1,378 +0,0 @@
-package ca.spottedleaf.concurrentutil.executor.standard;
-
-import java.util.ArrayDeque;
-import java.util.concurrent.atomic.AtomicLong;
-
-public class PrioritisedThreadedTaskQueue implements PrioritisedExecutor {
-
- protected final ArrayDeque<PrioritisedTask>[] queues = new ArrayDeque[Priority.TOTAL_SCHEDULABLE_PRIORITIES]; {
- for (int i = 0; i < Priority.TOTAL_SCHEDULABLE_PRIORITIES; ++i) {
- this.queues[i] = new ArrayDeque<>();
- }
- }
-
- // Use AtomicLong to separate from the queue field, we don't want false sharing here.
- protected final AtomicLong totalScheduledTasks = new AtomicLong();
- protected final AtomicLong totalCompletedTasks = new AtomicLong();
-
- // this is here to prevent failures to queue stalling flush() calls (as the schedule calls would increment totalScheduledTasks without this check)
- protected volatile boolean hasShutdown;
-
- protected long taskIdGenerator = 0;
-
- @Override
- public PrioritisedExecutor.PrioritisedTask queueRunnable(final Runnable task, final Priority priority) throws IllegalStateException, IllegalArgumentException {
- if (!Priority.isValidPriority(priority)) {
- throw new IllegalArgumentException("Priority " + priority + " is invalid");
- }
- if (task == null) {
- throw new NullPointerException("Task cannot be null");
- }
-
- if (this.hasShutdown) {
- // prevent us from stalling flush() calls by incrementing scheduled tasks when we really didn't schedule something
- throw new IllegalStateException("Queue has shutdown");
- }
-
- final PrioritisedTask ret;
-
- synchronized (this.queues) {
- if (this.hasShutdown) {
- throw new IllegalStateException("Queue has shutdown");
- }
- this.getAndAddTotalScheduledTasksVolatile(1L);
-
- ret = new PrioritisedTask(this.taskIdGenerator++, task, priority, this);
-
- this.queues[ret.priority.priority].add(ret);
-
- // call priority change callback (note: only after we successfully queue!)
- this.priorityChange(ret, null, priority);
- }
-
- return ret;
- }
-
- @Override
- public PrioritisedExecutor.PrioritisedTask createTask(final Runnable task, final Priority priority) {
- if (!Priority.isValidPriority(priority)) {
- throw new IllegalArgumentException("Priority " + priority + " is invalid");
- }
- if (task == null) {
- throw new NullPointerException("Task cannot be null");
- }
-
- return new PrioritisedTask(task, priority, this);
- }
-
- @Override
- public long getTotalTasksScheduled() {
- return this.totalScheduledTasks.get();
- }
-
- @Override
- public long getTotalTasksExecuted() {
- return this.totalCompletedTasks.get();
- }
-
- // callback method for subclasses to override
- // from is null when a task is immediately created
- protected void priorityChange(final PrioritisedTask task, final Priority from, final Priority to) {}
-
- /**
- * Polls the highest priority task currently available. {@code null} if none. This will mark the
- * returned task as completed.
- */
- protected PrioritisedTask poll() {
- return this.poll(Priority.IDLE);
- }
-
- protected PrioritisedTask poll(final Priority minPriority) {
- final ArrayDeque<PrioritisedTask>[] queues = this.queues;
- synchronized (queues) {
- final int max = minPriority.priority;
- for (int i = 0; i <= max; ++i) {
- final ArrayDeque<PrioritisedTask> queue = queues[i];
- PrioritisedTask task;
- while ((task = queue.pollFirst()) != null) {
- if (task.trySetCompleting(i)) {
- return task;
- }
- }
- }
- }
-
- return null;
- }
-
- /**
- * Polls and executes the highest priority task currently available. Exceptions thrown during task execution will
- * be rethrown.
- * @return {@code true} if a task was executed, {@code false} otherwise.
- */
- @Override
- public boolean executeTask() {
- final PrioritisedTask task = this.poll();
-
- if (task != null) {
- task.executeInternal();
- return true;
- }
-
- return false;
- }
-
- @Override
- public boolean shutdown() {
- synchronized (this.queues) {
- if (this.hasShutdown) {
- return false;
- }
- this.hasShutdown = true;
- }
- return true;
- }
-
- @Override
- public boolean isShutdown() {
- return this.hasShutdown;
- }
-
- /* totalScheduledTasks */
-
- protected final long getTotalScheduledTasksVolatile() {
- return this.totalScheduledTasks.get();
- }
-
- protected final long getAndAddTotalScheduledTasksVolatile(final long value) {
- return this.totalScheduledTasks.getAndAdd(value);
- }
-
- /* totalCompletedTasks */
-
- protected final long getTotalCompletedTasksVolatile() {
- return this.totalCompletedTasks.get();
- }
-
- protected final long getAndAddTotalCompletedTasksVolatile(final long value) {
- return this.totalCompletedTasks.getAndAdd(value);
- }
-
- protected static final class PrioritisedTask implements PrioritisedExecutor.PrioritisedTask {
- protected final PrioritisedThreadedTaskQueue queue;
- protected long id;
- protected static final long NOT_SCHEDULED_ID = -1L;
-
- protected Runnable runnable;
- protected volatile Priority priority;
-
- protected PrioritisedTask(final long id, final Runnable runnable, final Priority priority, final PrioritisedThreadedTaskQueue queue) {
- if (!Priority.isValidPriority(priority)) {
- throw new IllegalArgumentException("Invalid priority " + priority);
- }
-
- this.priority = priority;
- this.runnable = runnable;
- this.queue = queue;
- this.id = id;
- }
-
- protected PrioritisedTask(final Runnable runnable, final Priority priority, final PrioritisedThreadedTaskQueue queue) {
- if (!Priority.isValidPriority(priority)) {
- throw new IllegalArgumentException("Invalid priority " + priority);
- }
-
- this.priority = priority;
- this.runnable = runnable;
- this.queue = queue;
- this.id = NOT_SCHEDULED_ID;
- }
-
- @Override
- public boolean queue() {
- if (this.queue.hasShutdown) {
- throw new IllegalStateException("Queue has shutdown");
- }
-
- synchronized (this.queue.queues) {
- if (this.queue.hasShutdown) {
- throw new IllegalStateException("Queue has shutdown");
- }
-
- final Priority priority = this.priority;
- if (priority == Priority.COMPLETING) {
- return false;
- }
-
- if (this.id != NOT_SCHEDULED_ID) {
- return false;
- }
-
- this.queue.getAndAddTotalScheduledTasksVolatile(1L);
- this.id = this.queue.taskIdGenerator++;
- this.queue.queues[priority.priority].add(this);
-
- this.queue.priorityChange(this, null, priority);
-
- return true;
- }
- }
-
- protected boolean trySetCompleting(final int minPriority) {
- final Priority oldPriority = this.priority;
- if (oldPriority != Priority.COMPLETING && oldPriority.isHigherOrEqualPriority(minPriority)) {
- this.priority = Priority.COMPLETING;
- if (this.id != NOT_SCHEDULED_ID) {
- this.queue.priorityChange(this, oldPriority, Priority.COMPLETING);
- }
- return true;
- }
-
- return false;
- }
-
- @Override
- public Priority getPriority() {
- return this.priority;
- }
-
- @Override
- public boolean setPriority(final Priority priority) {
- if (!Priority.isValidPriority(priority)) {
- throw new IllegalArgumentException("Invalid priority " + priority);
- }
- synchronized (this.queue.queues) {
- final Priority curr = this.priority;
-
- if (curr == Priority.COMPLETING) {
- return false;
- }
-
- if (curr == priority) {
- return true;
- }
-
- this.priority = priority;
- if (this.id != NOT_SCHEDULED_ID) {
- this.queue.queues[priority.priority].add(this);
-
- // call priority change callback
- this.queue.priorityChange(this, curr, priority);
- }
- }
-
- return true;
- }
-
- @Override
- public boolean raisePriority(final Priority priority) {
- if (!Priority.isValidPriority(priority)) {
- throw new IllegalArgumentException("Invalid priority " + priority);
- }
-
- synchronized (this.queue.queues) {
- final Priority curr = this.priority;
-
- if (curr == Priority.COMPLETING) {
- return false;
- }
-
- if (curr.isHigherOrEqualPriority(priority)) {
- return true;
- }
-
- this.priority = priority;
- if (this.id != NOT_SCHEDULED_ID) {
- this.queue.queues[priority.priority].add(this);
-
- // call priority change callback
- this.queue.priorityChange(this, curr, priority);
- }
- }
-
- return true;
- }
-
- @Override
- public boolean lowerPriority(final Priority priority) {
- if (!Priority.isValidPriority(priority)) {
- throw new IllegalArgumentException("Invalid priority " + priority);
- }
-
- synchronized (this.queue.queues) {
- final Priority curr = this.priority;
-
- if (curr == Priority.COMPLETING) {
- return false;
- }
-
- if (curr.isLowerOrEqualPriority(priority)) {
- return true;
- }
-
- this.priority = priority;
- if (this.id != NOT_SCHEDULED_ID) {
- this.queue.queues[priority.priority].add(this);
-
- // call priority change callback
- this.queue.priorityChange(this, curr, priority);
- }
- }
-
- return true;
- }
-
- @Override
- public boolean cancel() {
- final long id;
- synchronized (this.queue.queues) {
- final Priority oldPriority = this.priority;
- if (oldPriority == Priority.COMPLETING) {
- return false;
- }
-
- this.priority = Priority.COMPLETING;
- // call priority change callback
- if ((id = this.id) != NOT_SCHEDULED_ID) {
- this.queue.priorityChange(this, oldPriority, Priority.COMPLETING);
- }
- }
- this.runnable = null;
- if (id != NOT_SCHEDULED_ID) {
- this.queue.getAndAddTotalCompletedTasksVolatile(1L);
- }
- return true;
- }
-
- protected void executeInternal() {
- try {
- final Runnable execute = this.runnable;
- this.runnable = null;
- execute.run();
- } finally {
- if (this.id != NOT_SCHEDULED_ID) {
- this.queue.getAndAddTotalCompletedTasksVolatile(1L);
- }
- }
- }
-
- @Override
- public boolean execute() {
- synchronized (this.queue.queues) {
- final Priority oldPriority = this.priority;
- if (oldPriority == Priority.COMPLETING) {
- return false;
- }
-
- this.priority = Priority.COMPLETING;
- // call priority change callback
- if (this.id != NOT_SCHEDULED_ID) {
- this.queue.priorityChange(this, oldPriority, Priority.COMPLETING);
- }
- }
-
- this.executeInternal();
- return true;
- }
- }
-}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/executor/standard/PrioritisedQueueExecutorThread.java b/src/main/java/ca/spottedleaf/concurrentutil/executor/thread/PrioritisedQueueExecutorThread.java
similarity index 60%
rename from src/main/java/ca/spottedleaf/concurrentutil/executor/standard/PrioritisedQueueExecutorThread.java
rename to src/main/java/ca/spottedleaf/concurrentutil/executor/thread/PrioritisedQueueExecutorThread.java
index d1683ba6350e530373944f98192c0f2baf241e70..f5367a13aaa02f0f929813c00a67e6ac7c8652cb 100644
--- a/src/main/java/ca/spottedleaf/concurrentutil/executor/standard/PrioritisedQueueExecutorThread.java
+++ b/src/main/java/ca/spottedleaf/concurrentutil/executor/thread/PrioritisedQueueExecutorThread.java
@@ -1,6 +1,8 @@
-package ca.spottedleaf.concurrentutil.executor.standard;
+package ca.spottedleaf.concurrentutil.executor.thread;
+import ca.spottedleaf.concurrentutil.executor.PrioritisedExecutor;
import ca.spottedleaf.concurrentutil.util.ConcurrentUtil;
+import ca.spottedleaf.concurrentutil.util.Priority;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.lang.invoke.VarHandle;
@@ -11,8 +13,7 @@ import java.util.concurrent.locks.LockSupport;
* <p>
* Note: When using this thread, queue additions to the underlying {@link #queue} are not sufficient to get this thread
* to execute the task. The function {@link #notifyTasks()} must be used after scheduling a task. For expected behaviour
- * of task scheduling (thread wakes up after tasks are scheduled), use the methods provided on {@link PrioritisedExecutor}
- * methods.
+ * of task scheduling, use the methods provided on this class to schedule tasks.
* </p>
*/
public class PrioritisedQueueExecutorThread extends Thread implements PrioritisedExecutor {
@@ -30,7 +31,7 @@ public class PrioritisedQueueExecutorThread extends Thread implements Prioritise
protected final long spinWaitTime;
- static final long DEFAULT_SPINWAIT_TIME = (long)(0.1e6);// 0.1ms
+ protected static final long DEFAULT_SPINWAIT_TIME = (long)(0.1e6);// 0.1ms
public PrioritisedQueueExecutorThread(final PrioritisedExecutor queue) {
this(queue, DEFAULT_SPINWAIT_TIME); // 0.1ms
@@ -42,7 +43,16 @@ public class PrioritisedQueueExecutorThread extends Thread implements Prioritise
}
@Override
- public void run() {
+ public final void run() {
+ try {
+ this.begin();
+ this.doRun();
+ } finally {
+ this.die();
+ }
+ }
+
+ public final void doRun() {
final long spinWaitTime = this.spinWaitTime;
main_loop:
@@ -80,7 +90,7 @@ public class PrioritisedQueueExecutorThread extends Thread implements Prioritise
this.setThreadParkedVolatile(true);
// We need to parse here to avoid a race condition where a thread queues a task before we set parked to true
- // (i.e it will not notify us)
+ // (i.e. it will not notify us)
if (this.pollTasks()) {
this.setThreadParkedVolatile(false);
continue;
@@ -99,6 +109,10 @@ public class PrioritisedQueueExecutorThread extends Thread implements Prioritise
}
}
+ protected void begin() {}
+
+ protected void die() {}
+
/**
* Attempts to poll as many tasks as possible, returning when finished.
* @return Whether any tasks were executed.
@@ -115,8 +129,6 @@ public class PrioritisedQueueExecutorThread extends Thread implements Prioritise
break;
}
ret = true;
- } catch (final ThreadDeath death) {
- throw death; // goodbye world...
} catch (final Throwable throwable) {
LOGGER.error("Exception thrown from prioritized runnable task in thread '" + this.getName() + "'", throwable);
}
@@ -146,95 +158,86 @@ public class PrioritisedQueueExecutorThread extends Thread implements Prioritise
}
@Override
- public PrioritisedTask createTask(final Runnable task, final Priority priority) {
- final PrioritisedTask queueTask = this.queue.createTask(task, priority);
-
- // need to override queue() to notify us of tasks
- return new PrioritisedTask() {
- @Override
- public Priority getPriority() {
- return queueTask.getPriority();
- }
-
- @Override
- public boolean setPriority(final Priority priority) {
- return queueTask.setPriority(priority);
- }
+ public long getTotalTasksExecuted() {
+ return this.queue.getTotalTasksExecuted();
+ }
- @Override
- public boolean raisePriority(final Priority priority) {
- return queueTask.raisePriority(priority);
- }
+ @Override
+ public long getTotalTasksScheduled() {
+ return this.queue.getTotalTasksScheduled();
+ }
- @Override
- public boolean lowerPriority(final Priority priority) {
- return queueTask.lowerPriority(priority);
- }
+ @Override
+ public long generateNextSubOrder() {
+ return this.queue.generateNextSubOrder();
+ }
- @Override
- public boolean queue() {
- final boolean ret = queueTask.queue();
- if (ret) {
- PrioritisedQueueExecutorThread.this.notifyTasks();
- }
- return ret;
- }
+ @Override
+ public boolean shutdown() {
+ throw new UnsupportedOperationException();
+ }
- @Override
- public boolean cancel() {
- return queueTask.cancel();
- }
+ @Override
+ public boolean isShutdown() {
+ return false;
+ }
- @Override
- public boolean execute() {
- return queueTask.execute();
- }
- };
+ /**
+ * {@inheritDoc}
+ * @throws IllegalStateException Always
+ */
+ @Override
+ public boolean executeTask() throws IllegalStateException {
+ throw new IllegalStateException();
}
@Override
- public PrioritisedTask queueRunnable(final Runnable task, final Priority priority) {
- final PrioritisedTask ret = this.queue.queueRunnable(task, priority);
+ public PrioritisedTask queueTask(final Runnable task) {
+ final PrioritisedTask ret = this.createTask(task);
- this.notifyTasks();
+ ret.queue();
return ret;
}
@Override
- public boolean haveAllTasksExecuted() {
- return this.queue.haveAllTasksExecuted();
+ public PrioritisedTask queueTask(final Runnable task, final Priority priority) {
+ final PrioritisedTask ret = this.createTask(task, priority);
+
+ ret.queue();
+
+ return ret;
}
@Override
- public long getTotalTasksExecuted() {
- return this.queue.getTotalTasksExecuted();
+ public PrioritisedTask queueTask(final Runnable task, final Priority priority, final long subOrder) {
+ final PrioritisedTask ret = this.createTask(task, priority, subOrder);
+
+ ret.queue();
+
+ return ret;
}
+
@Override
- public long getTotalTasksScheduled() {
- return this.queue.getTotalTasksScheduled();
+ public PrioritisedTask createTask(Runnable task) {
+ final PrioritisedTask queueTask = this.queue.createTask(task);
+
+ return new WrappedTask(queueTask);
}
- /**
- * {@inheritDoc}
- * @throws IllegalStateException If the current thread is {@code this} thread, or the underlying queue throws this exception.
- */
@Override
- public void waitUntilAllExecuted() throws IllegalStateException {
- if (Thread.currentThread() == this) {
- throw new IllegalStateException("Cannot block on our own queue");
- }
- this.queue.waitUntilAllExecuted();
+ public PrioritisedTask createTask(final Runnable task, final Priority priority) {
+ final PrioritisedTask queueTask = this.queue.createTask(task, priority);
+
+ return new WrappedTask(queueTask);
}
- /**
- * {@inheritDoc}
- * @throws IllegalStateException Always
- */
@Override
- public boolean executeTask() throws IllegalStateException {
- throw new IllegalStateException();
+ public PrioritisedTask createTask(final Runnable task, final Priority priority, final long subOrder) {
+ final PrioritisedTask queueTask = this.queue.createTask(task, priority, subOrder);
+
+ return new WrappedTask(queueTask);
}
/**
@@ -242,7 +245,7 @@ public class PrioritisedQueueExecutorThread extends Thread implements Prioritise
* <p>
* This function is MT-Safe.
* </p>
- * @param wait If this call is to wait until the queue is empty and there are no tasks executing in the queue.
+ * @param wait If this call is to wait until this thread shuts down.
* @param killQueue Whether to shutdown this thread's queue
* @return whether this thread shut down the queue
* @see #halt(boolean)
@@ -256,7 +259,20 @@ public class PrioritisedQueueExecutorThread extends Thread implements Prioritise
LockSupport.unpark(this);
if (wait) {
- this.waitUntilAllExecuted();
+ boolean interrupted = false;
+ for (;;) {
+ if (this.isAlive()) {
+ if (interrupted) {
+ Thread.currentThread().interrupt();
+ }
+ break;
+ }
+ try {
+ this.join();
+ } catch (final InterruptedException ex) {
+ interrupted = true;
+ }
+ }
}
return ret;
@@ -298,4 +314,89 @@ public class PrioritisedQueueExecutorThread extends Thread implements Prioritise
protected final void setThreadParkedVolatile(final boolean value) {
THREAD_PARKED_HANDLE.setVolatile(this, value);
}
+
+ /**
+ * Required so that queue() can notify (unpark) this thread
+ */
+ private final class WrappedTask implements PrioritisedTask {
+ private final PrioritisedTask queueTask;
+
+ public WrappedTask(final PrioritisedTask queueTask) {
+ this.queueTask = queueTask;
+ }
+
+ @Override
+ public PrioritisedExecutor getExecutor() {
+ return PrioritisedQueueExecutorThread.this;
+ }
+
+ @Override
+ public boolean queue() {
+ final boolean ret = this.queueTask.queue();
+ if (ret) {
+ PrioritisedQueueExecutorThread.this.notifyTasks();
+ }
+ return ret;
+ }
+
+ @Override
+ public boolean isQueued() {
+ return this.queueTask.isQueued();
+ }
+
+ @Override
+ public boolean cancel() {
+ return this.queueTask.cancel();
+ }
+
+ @Override
+ public boolean execute() {
+ return this.queueTask.execute();
+ }
+
+ @Override
+ public Priority getPriority() {
+ return this.queueTask.getPriority();
+ }
+
+ @Override
+ public boolean setPriority(final Priority priority) {
+ return this.queueTask.setPriority(priority);
+ }
+
+ @Override
+ public boolean raisePriority(final Priority priority) {
+ return this.queueTask.raisePriority(priority);
+ }
+
+ @Override
+ public boolean lowerPriority(final Priority priority) {
+ return this.queueTask.lowerPriority(priority);
+ }
+
+ @Override
+ public long getSubOrder() {
+ return this.queueTask.getSubOrder();
+ }
+
+ @Override
+ public boolean setSubOrder(final long subOrder) {
+ return this.queueTask.setSubOrder(subOrder);
+ }
+
+ @Override
+ public boolean raiseSubOrder(final long subOrder) {
+ return this.queueTask.raiseSubOrder(subOrder);
+ }
+
+ @Override
+ public boolean lowerSubOrder(final long subOrder) {
+ return this.queueTask.lowerSubOrder(subOrder);
+ }
+
+ @Override
+ public boolean setPriorityAndSubOrder(final Priority priority, final long subOrder) {
+ return this.queueTask.setPriorityAndSubOrder(priority, subOrder);
+ }
+ }
}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/executor/thread/PrioritisedThreadPool.java b/src/main/java/ca/spottedleaf/concurrentutil/executor/thread/PrioritisedThreadPool.java
new file mode 100644
index 0000000000000000000000000000000000000000..cb9df914a9a6d0d3f58fa58d8c93f4f583416cd1
--- /dev/null
+++ b/src/main/java/ca/spottedleaf/concurrentutil/executor/thread/PrioritisedThreadPool.java
@@ -0,0 +1,741 @@
+package ca.spottedleaf.concurrentutil.executor.thread;
+
+import ca.spottedleaf.concurrentutil.executor.PrioritisedExecutor;
+import ca.spottedleaf.concurrentutil.executor.queue.PrioritisedTaskQueue;
+import ca.spottedleaf.concurrentutil.util.Priority;
+import ca.spottedleaf.concurrentutil.util.TimeUtil;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import java.lang.reflect.Array;
+import java.util.Arrays;
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.AtomicLong;
+import java.util.function.Consumer;
+
+public final class PrioritisedThreadPool {
+
+ private static final Logger LOGGER = LoggerFactory.getLogger(PrioritisedThreadPool.class);
+
+ private final Consumer<Thread> threadModifier;
+ private final COWArrayList<ExecutorGroup> executors = new COWArrayList<>(ExecutorGroup.class);
+ private final COWArrayList<PrioritisedThread> threads = new COWArrayList<>(PrioritisedThread.class);
+ private final COWArrayList<PrioritisedThread> aliveThreads = new COWArrayList<>(PrioritisedThread.class);
+
+ private static final Priority HIGH_PRIORITY_NOTIFY_THRESHOLD = Priority.HIGH;
+ private static final Priority QUEUE_SHUTDOWN_PRIORITY = Priority.HIGH;
+
+ private boolean shutdown;
+
+ public PrioritisedThreadPool(final Consumer<Thread> threadModifier) {
+ this.threadModifier = threadModifier;
+
+ if (threadModifier == null) {
+ throw new NullPointerException("Thread factory may not be null");
+ }
+ }
+
+ public Thread[] getAliveThreads() {
+ final PrioritisedThread[] threads = this.aliveThreads.getArray();
+
+ return Arrays.copyOf(threads, threads.length, Thread[].class);
+ }
+
+ public Thread[] getCoreThreads() {
+ final PrioritisedThread[] threads = this.threads.getArray();
+
+ return Arrays.copyOf(threads, threads.length, Thread[].class);
+ }
+
+ /**
+ * Prevents creation of new queues, shutdowns all non-shutdown queues if specified
+ */
+ public void halt(final boolean shutdownQueues) {
+ synchronized (this) {
+ this.shutdown = true;
+ }
+
+ if (shutdownQueues) {
+ for (final ExecutorGroup group : this.executors.getArray()) {
+ for (final ExecutorGroup.ThreadPoolExecutor executor : group.executors.getArray()) {
+ executor.shutdown();
+ }
+ }
+ }
+
+ for (final PrioritisedThread thread : this.threads.getArray()) {
+ thread.halt(false);
+ }
+ }
+
+ /**
+ * Waits until all threads in this pool have shutdown, or until the specified time has passed.
+ * @param msToWait Maximum time to wait.
+ * @return {@code false} if the maximum time passed, {@code true} otherwise.
+ */
+ public boolean join(final long msToWait) {
+ try {
+ return this.join(msToWait, false);
+ } catch (final InterruptedException ex) {
+ throw new IllegalStateException(ex);
+ }
+ }
+
+ /**
+ * Waits until all threads in this pool have shutdown, or until the specified time has passed.
+ * @param msToWait Maximum time to wait.
+ * @return {@code false} if the maximum time passed, {@code true} otherwise.
+ * @throws InterruptedException If this thread is interrupted.
+ */
+ public boolean joinInterruptable(final long msToWait) throws InterruptedException {
+ return this.join(msToWait, true);
+ }
+
+ protected final boolean join(final long msToWait, final boolean interruptable) throws InterruptedException {
+ final long nsToWait = msToWait * (1000 * 1000);
+ final long start = System.nanoTime();
+ final long deadline = start + nsToWait;
+ boolean interrupted = false;
+ try {
+ for (final PrioritisedThread thread : this.aliveThreads.getArray()) {
+ for (;;) {
+ if (!thread.isAlive()) {
+ break;
+ }
+ final long current = System.nanoTime();
+ if (current >= deadline && msToWait > 0L) {
+ return false;
+ }
+
+ try {
+ thread.join(msToWait <= 0L ? 0L : Math.max(1L, (deadline - current) / (1000 * 1000)));
+ } catch (final InterruptedException ex) {
+ if (interruptable) {
+ throw ex;
+ }
+ interrupted = true;
+ }
+ }
+ }
+
+ return true;
+ } finally {
+ if (interrupted) {
+ Thread.currentThread().interrupt();
+ }
+ }
+ }
+
+ /**
+ * Shuts down this thread pool, optionally waiting for all tasks to be executed.
+ * This function will invoke {@link PrioritisedExecutor#shutdown()} on all created executors on this
+ * thread pool.
+ * @param wait Whether to wait for tasks to be executed
+ */
+ public void shutdown(final boolean wait) {
+ synchronized (this) {
+ this.shutdown = true;
+ }
+
+ for (final ExecutorGroup group : this.executors.getArray()) {
+ for (final ExecutorGroup.ThreadPoolExecutor executor : group.executors.getArray()) {
+ executor.shutdown();
+ }
+ }
+
+
+ for (final PrioritisedThread thread : this.threads.getArray()) {
+ // none of these can be true or else NPE
+ thread.close(false, false);
+ }
+
+ if (wait) {
+ this.join(0L);
+ }
+ }
+
+ private void die(final PrioritisedThread thread) {
+ this.aliveThreads.remove(thread);
+ }
+
+ public void adjustThreadCount(final int threads) {
+ synchronized (this) {
+ if (this.shutdown) {
+ return;
+ }
+
+ final PrioritisedThread[] currentThreads = this.threads.getArray();
+ if (threads == currentThreads.length) {
+ // no adjustment needed
+ return;
+ }
+
+ if (threads < currentThreads.length) {
+ // we need to trim threads
+ for (int i = 0, difference = currentThreads.length - threads; i < difference; ++i) {
+ final PrioritisedThread remove = currentThreads[currentThreads.length - i - 1];
+
+ remove.halt(false);
+ this.threads.remove(remove);
+ }
+ } else {
+ // we need to add threads
+ for (int i = 0, difference = threads - currentThreads.length; i < difference; ++i) {
+ final PrioritisedThread thread = new PrioritisedThread();
+
+ this.threadModifier.accept(thread);
+ this.aliveThreads.add(thread);
+ this.threads.add(thread);
+
+ thread.start();
+ }
+ }
+ }
+ }
+
+ private static int compareInsideGroup(final ExecutorGroup.ThreadPoolExecutor src, final Priority srcPriority,
+ final ExecutorGroup.ThreadPoolExecutor dst, final Priority dstPriority) {
+ final int priorityCompare = srcPriority.ordinal() - dstPriority.ordinal();
+ if (priorityCompare != 0) {
+ return priorityCompare;
+ }
+
+ final int parallelismCompare = src.currentParallelism - dst.currentParallelism;
+ if (parallelismCompare != 0) {
+ return parallelismCompare;
+ }
+
+ return TimeUtil.compareTimes(src.lastRetrieved, dst.lastRetrieved);
+ }
+
+ private static int compareOutsideGroup(final ExecutorGroup.ThreadPoolExecutor src, final Priority srcPriority,
+ final ExecutorGroup.ThreadPoolExecutor dst, final Priority dstPriority) {
+ if (src.getGroup().division == dst.getGroup().division) {
+ // can only compare priorities inside the same division
+ final int priorityCompare = srcPriority.ordinal() - dstPriority.ordinal();
+ if (priorityCompare != 0) {
+ return priorityCompare;
+ }
+ }
+
+ final int parallelismCompare = src.getGroup().currentParallelism - dst.getGroup().currentParallelism;
+ if (parallelismCompare != 0) {
+ return parallelismCompare;
+ }
+
+ return TimeUtil.compareTimes(src.lastRetrieved, dst.lastRetrieved);
+ }
+
+ private ExecutorGroup.ThreadPoolExecutor obtainQueue() {
+ final long time = System.nanoTime();
+ synchronized (this) {
+ ExecutorGroup.ThreadPoolExecutor ret = null;
+ Priority retPriority = null;
+
+ for (final ExecutorGroup executorGroup : this.executors.getArray()) {
+ ExecutorGroup.ThreadPoolExecutor highest = null;
+ Priority highestPriority = null;
+ for (final ExecutorGroup.ThreadPoolExecutor executor : executorGroup.executors.getArray()) {
+ final int maxParallelism = executor.maxParallelism;
+ if (maxParallelism > 0 && executor.currentParallelism >= maxParallelism) {
+ continue;
+ }
+
+ final Priority priority = executor.getTargetPriority();
+
+ if (priority == null) {
+ continue;
+ }
+
+ if (highestPriority == null || compareInsideGroup(highest, highestPriority, executor, priority) > 0) {
+ highest = executor;
+ highestPriority = priority;
+ }
+ }
+
+ if (highest == null) {
+ continue;
+ }
+
+ if (ret == null || compareOutsideGroup(ret, retPriority, highest, highestPriority) > 0) {
+ ret = highest;
+ retPriority = highestPriority;
+ }
+ }
+
+ if (ret != null) {
+ ret.lastRetrieved = time;
+ ++ret.currentParallelism;
+ ++ret.getGroup().currentParallelism;
+ return ret;
+ }
+
+ return ret;
+ }
+ }
+
+ private void returnQueue(final ExecutorGroup.ThreadPoolExecutor executor) {
+ synchronized (this) {
+ --executor.currentParallelism;
+ --executor.getGroup().currentParallelism;
+ }
+
+ if (executor.isShutdown() && executor.queue.hasNoScheduledTasks()) {
+ executor.getGroup().executors.remove(executor);
+ }
+ }
+
+ private void notifyAllThreads() {
+ for (final PrioritisedThread thread : this.threads.getArray()) {
+ thread.notifyTasks();
+ }
+ }
+
+ public ExecutorGroup createExecutorGroup(final int division, final int flags) {
+ synchronized (this) {
+ if (this.shutdown) {
+ throw new IllegalStateException("Queue is shutdown: " + this.toString());
+ }
+
+ final ExecutorGroup ret = new ExecutorGroup(division, flags);
+
+ this.executors.add(ret);
+
+ return ret;
+ }
+ }
+
+ private final class PrioritisedThread extends PrioritisedQueueExecutorThread {
+
+ private final AtomicBoolean alertedHighPriority = new AtomicBoolean();
+
+ public PrioritisedThread() {
+ super(null);
+ }
+
+ public boolean alertHighPriorityExecutor() {
+ if (!this.notifyTasks()) {
+ if (!this.alertedHighPriority.get()) {
+ this.alertedHighPriority.set(true);
+ }
+ return false;
+ }
+
+ return true;
+ }
+
+ private boolean isAlertedHighPriority() {
+ return this.alertedHighPriority.get() && this.alertedHighPriority.getAndSet(false);
+ }
+
+ @Override
+ protected void die() {
+ PrioritisedThreadPool.this.die(this);
+ }
+
+ @Override
+ protected boolean pollTasks() {
+ boolean ret = false;
+
+ for (;;) {
+ if (this.halted) {
+ break;
+ }
+
+ final ExecutorGroup.ThreadPoolExecutor executor = PrioritisedThreadPool.this.obtainQueue();
+ if (executor == null) {
+ break;
+ }
+ final long deadline = System.nanoTime() + executor.queueMaxHoldTime;
+ do {
+ try {
+ if (this.halted || executor.halt) {
+ break;
+ }
+ if (!executor.executeTask()) {
+ // no more tasks, try next queue
+ break;
+ }
+ ret = true;
+ } catch (final Throwable throwable) {
+ LOGGER.error("Exception thrown from thread '" + this.getName() + "' in queue '" + executor.toString() + "'", throwable);
+ }
+ } while (!this.isAlertedHighPriority() && System.nanoTime() <= deadline);
+
+ PrioritisedThreadPool.this.returnQueue(executor);
+ }
+
+
+ return ret;
+ }
+ }
+
+ public final class ExecutorGroup {
+
+ private final AtomicLong subOrderGenerator = new AtomicLong();
+ private final COWArrayList<ThreadPoolExecutor> executors = new COWArrayList<>(ThreadPoolExecutor.class);
+
+ private final int division;
+ private int currentParallelism;
+
+ private ExecutorGroup(final int division, final int flags) {
+ this.division = division;
+ }
+
+ public ThreadPoolExecutor[] getAllExecutors() {
+ return this.executors.getArray().clone();
+ }
+
+ private PrioritisedThreadPool getThreadPool() {
+ return PrioritisedThreadPool.this;
+ }
+
+ public ThreadPoolExecutor createExecutor(final int maxParallelism, final long queueMaxHoldTime, final int flags) {
+ synchronized (PrioritisedThreadPool.this) {
+ if (PrioritisedThreadPool.this.shutdown) {
+ throw new IllegalStateException("Queue is shutdown: " + PrioritisedThreadPool.this.toString());
+ }
+
+ final ThreadPoolExecutor ret = new ThreadPoolExecutor(maxParallelism, queueMaxHoldTime, flags);
+
+ this.executors.add(ret);
+
+ return ret;
+ }
+ }
+
+ public final class ThreadPoolExecutor implements PrioritisedExecutor {
+
+ private final PrioritisedTaskQueue queue = new PrioritisedTaskQueue();
+
+ private volatile int maxParallelism;
+ private final long queueMaxHoldTime;
+ private volatile int currentParallelism;
+ private volatile boolean halt;
+ private long lastRetrieved = System.nanoTime();
+
+ private ThreadPoolExecutor(final int maxParallelism, final long queueMaxHoldTime, final int flags) {
+ this.maxParallelism = maxParallelism;
+ this.queueMaxHoldTime = queueMaxHoldTime;
+ }
+
+ private ExecutorGroup getGroup() {
+ return ExecutorGroup.this;
+ }
+
+ private boolean canNotify() {
+ if (this.halt) {
+ return false;
+ }
+
+ final int max = this.maxParallelism;
+ return max < 0 || this.currentParallelism < max;
+ }
+
+ private void notifyHighPriority() {
+ if (!this.canNotify()) {
+ return;
+ }
+ for (final PrioritisedThread thread : this.getGroup().getThreadPool().threads.getArray()) {
+ if (thread.alertHighPriorityExecutor()) {
+ return;
+ }
+ }
+ }
+
+ private void notifyScheduled() {
+ if (!this.canNotify()) {
+ return;
+ }
+ for (final PrioritisedThread thread : this.getGroup().getThreadPool().threads.getArray()) {
+ if (thread.notifyTasks()) {
+ return;
+ }
+ }
+ }
+
+ /**
+ * Removes this queue from the thread pool without shutting the queue down or waiting for queued tasks to be executed
+ */
+ public void halt() {
+ this.halt = true;
+
+ ExecutorGroup.this.executors.remove(this);
+ }
+
+ /**
+ * Returns whether this executor is scheduled to run tasks or is running tasks, otherwise it returns whether
+ * this queue is not halted and not shutdown.
+ */
+ public boolean isActive() {
+ if (this.halt) {
+ return this.currentParallelism > 0;
+ } else {
+ if (!this.isShutdown()) {
+ return true;
+ }
+
+ return !this.queue.hasNoScheduledTasks();
+ }
+ }
+
+ @Override
+ public boolean shutdown() {
+ if (!this.queue.shutdown()) {
+ return false;
+ }
+
+ if (this.queue.hasNoScheduledTasks()) {
+ ExecutorGroup.this.executors.remove(this);
+ }
+
+ return true;
+ }
+
+ @Override
+ public boolean isShutdown() {
+ return this.queue.isShutdown();
+ }
+
+ public void setMaxParallelism(final int maxParallelism) {
+ this.maxParallelism = maxParallelism;
+ // assume that we could have increased the parallelism
+ if (this.getTargetPriority() != null) {
+ ExecutorGroup.this.getThreadPool().notifyAllThreads();
+ }
+ }
+
+ Priority getTargetPriority() {
+ final Priority ret = this.queue.getHighestPriority();
+ if (!this.isShutdown()) {
+ return ret;
+ }
+
+ return ret == null ? QUEUE_SHUTDOWN_PRIORITY : Priority.max(ret, QUEUE_SHUTDOWN_PRIORITY);
+ }
+
+ @Override
+ public long getTotalTasksScheduled() {
+ return this.queue.getTotalTasksScheduled();
+ }
+
+ @Override
+ public long getTotalTasksExecuted() {
+ return this.queue.getTotalTasksExecuted();
+ }
+
+ @Override
+ public long generateNextSubOrder() {
+ return ExecutorGroup.this.subOrderGenerator.getAndIncrement();
+ }
+
+ @Override
+ public boolean executeTask() {
+ return this.queue.executeTask();
+ }
+
+ @Override
+ public PrioritisedTask queueTask(final Runnable task) {
+ final PrioritisedTask ret = this.createTask(task);
+
+ ret.queue();
+
+ return ret;
+ }
+
+ @Override
+ public PrioritisedTask queueTask(final Runnable task, final Priority priority) {
+ final PrioritisedTask ret = this.createTask(task, priority);
+
+ ret.queue();
+
+ return ret;
+ }
+
+ @Override
+ public PrioritisedTask queueTask(final Runnable task, final Priority priority, final long subOrder) {
+ final PrioritisedTask ret = this.createTask(task, priority, subOrder);
+
+ ret.queue();
+
+ return ret;
+ }
+
+ @Override
+ public PrioritisedTask createTask(final Runnable task) {
+ return this.createTask(task, Priority.NORMAL);
+ }
+
+ @Override
+ public PrioritisedTask createTask(final Runnable task, final Priority priority) {
+ return this.createTask(task, priority, this.generateNextSubOrder());
+ }
+
+ @Override
+ public PrioritisedTask createTask(final Runnable task, final Priority priority, final long subOrder) {
+ return new WrappedTask(this.queue.createTask(task, priority, subOrder));
+ }
+
+ private final class WrappedTask implements PrioritisedTask {
+
+ private final PrioritisedTask wrapped;
+
+ private WrappedTask(final PrioritisedTask wrapped) {
+ this.wrapped = wrapped;
+ }
+
+ @Override
+ public PrioritisedExecutor getExecutor() {
+ return ThreadPoolExecutor.this;
+ }
+
+ @Override
+ public boolean queue() {
+ if (this.wrapped.queue()) {
+ final Priority priority = this.getPriority();
+ if (priority != Priority.COMPLETING) {
+ if (priority.isHigherOrEqualPriority(HIGH_PRIORITY_NOTIFY_THRESHOLD)) {
+ ThreadPoolExecutor.this.notifyHighPriority();
+ } else {
+ ThreadPoolExecutor.this.notifyScheduled();
+ }
+ }
+ return true;
+ }
+
+ return false;
+ }
+
+ @Override
+ public boolean isQueued() {
+ return this.wrapped.isQueued();
+ }
+
+ @Override
+ public boolean cancel() {
+ return this.wrapped.cancel();
+ }
+
+ @Override
+ public boolean execute() {
+ return this.wrapped.execute();
+ }
+
+ @Override
+ public Priority getPriority() {
+ return this.wrapped.getPriority();
+ }
+
+ @Override
+ public boolean setPriority(final Priority priority) {
+ if (this.wrapped.setPriority(priority)) {
+ if (priority.isHigherOrEqualPriority(HIGH_PRIORITY_NOTIFY_THRESHOLD)) {
+ ThreadPoolExecutor.this.notifyHighPriority();
+ }
+ return true;
+ }
+
+ return false;
+ }
+
+ @Override
+ public boolean raisePriority(final Priority priority) {
+ if (this.wrapped.raisePriority(priority)) {
+ if (priority.isHigherOrEqualPriority(HIGH_PRIORITY_NOTIFY_THRESHOLD)) {
+ ThreadPoolExecutor.this.notifyHighPriority();
+ }
+ return true;
+ }
+
+ return false;
+ }
+
+ @Override
+ public boolean lowerPriority(final Priority priority) {
+ return this.wrapped.lowerPriority(priority);
+ }
+
+ @Override
+ public long getSubOrder() {
+ return this.wrapped.getSubOrder();
+ }
+
+ @Override
+ public boolean setSubOrder(final long subOrder) {
+ return this.wrapped.setSubOrder(subOrder);
+ }
+
+ @Override
+ public boolean raiseSubOrder(final long subOrder) {
+ return this.wrapped.raiseSubOrder(subOrder);
+ }
+
+ @Override
+ public boolean lowerSubOrder(final long subOrder) {
+ return this.wrapped.lowerSubOrder(subOrder);
+ }
+
+ @Override
+ public boolean setPriorityAndSubOrder(final Priority priority, final long subOrder) {
+ if (this.wrapped.setPriorityAndSubOrder(priority, subOrder)) {
+ if (priority.isHigherOrEqualPriority(HIGH_PRIORITY_NOTIFY_THRESHOLD)) {
+ ThreadPoolExecutor.this.notifyHighPriority();
+ }
+ return true;
+ }
+
+ return false;
+ }
+ }
+ }
+ }
+
+ private static final class COWArrayList<E> {
+
+ private volatile E[] array;
+
+ public COWArrayList(final Class<E> clazz) {
+ this.array = (E[])Array.newInstance(clazz, 0);
+ }
+
+ public E[] getArray() {
+ return this.array;
+ }
+
+ public void add(final E element) {
+ synchronized (this) {
+ final E[] array = this.array;
+
+ final E[] copy = Arrays.copyOf(array, array.length + 1);
+ copy[array.length] = element;
+
+ this.array = copy;
+ }
+ }
+
+ public boolean remove(final E element) {
+ synchronized (this) {
+ final E[] array = this.array;
+ int index = -1;
+ for (int i = 0, len = array.length; i < len; ++i) {
+ if (array[i] == element) {
+ index = i;
+ break;
+ }
+ }
+
+ if (index == -1) {
+ return false;
+ }
+
+ final E[] copy = (E[])Array.newInstance(array.getClass().getComponentType(), array.length - 1);
+
+ System.arraycopy(array, 0, copy, 0, index);
+ System.arraycopy(array, index + 1, copy, index, (array.length - 1) - index);
+
+ this.array = copy;
+ }
+
+ return true;
+ }
+ }
+}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/map/ConcurrentLong2ReferenceChainedHashTable.java b/src/main/java/ca/spottedleaf/concurrentutil/map/ConcurrentLong2ReferenceChainedHashTable.java
index d701998b376579ec652fb94823befa3cc0bc4090..6918f130099e6c19e20a47bfdb54915cdd13732a 100644
--- a/src/main/java/ca/spottedleaf/concurrentutil/map/ConcurrentLong2ReferenceChainedHashTable.java
+++ b/src/main/java/ca/spottedleaf/concurrentutil/map/ConcurrentLong2ReferenceChainedHashTable.java
@@ -43,7 +43,7 @@ import java.util.function.Predicate;
* @param <V>
* @see java.util.concurrent.ConcurrentMap
*/
-public class ConcurrentLong2ReferenceChainedHashTable<V> {
+public class ConcurrentLong2ReferenceChainedHashTable<V> implements Iterable<ConcurrentLong2ReferenceChainedHashTable.TableEntry<V>> {
protected static final int DEFAULT_CAPACITY = 16;
protected static final float DEFAULT_LOAD_FACTOR = 0.75f;
@@ -192,6 +192,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
}
if (node.resize) {
+ // noinspection unchecked
table = (TableEntry<V>[])node.getValuePlain();
continue;
}
@@ -274,10 +275,10 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
public int size() {
final long ret = this.size.sum();
- if (ret <= 0L) {
+ if (ret < 0L) {
return 0;
}
- if (ret >= (long)Integer.MAX_VALUE) {
+ if (ret > (long)Integer.MAX_VALUE) {
return Integer.MAX_VALUE;
}
@@ -341,6 +342,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
// create new table data
+ // noinspection unchecked
final TableEntry<V>[] newTable = new TableEntry[capacity];
// noinspection unchecked
final TableEntry<V> resizeNode = new TableEntry<>(0L, (V)newTable, true);
@@ -365,6 +367,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
throw new IllegalStateException("Resizing to same size");
}
+ // noinspection unchecked
final TableEntry<V>[] work = new TableEntry[1 << capDiffShift]; // typically, capDiffShift = 1
for (int i = 0, len = oldTable.length; i < len; ++i) {
@@ -538,6 +541,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
}
if (node.resize) {
+ // noinspection unchecked
table = (TableEntry<V>[])node.getValuePlain();
continue table_loop;
}
@@ -599,6 +603,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
}
if (node.resize) {
+ // noinspection unchecked
table = (TableEntry<V>[])node.getValuePlain();
continue table_loop;
}
@@ -653,6 +658,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
}
if (node.resize) {
+ // noinspection unchecked
table = (TableEntry<V>[])node.getValuePlain();
continue table_loop;
}
@@ -704,6 +710,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
}
if (node.resize) {
+ // noinspection unchecked
table = (TableEntry<V>[])node.getValuePlain();
continue table_loop;
}
@@ -772,6 +779,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
}
if (node.resize) {
+ // noinspection unchecked
table = (TableEntry<V>[])node.getValuePlain();
continue table_loop;
}
@@ -848,6 +856,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
}
if (node.resize) {
+ // noinspection unchecked
table = (TableEntry<V>[])node.getValuePlain();
continue table_loop;
}
@@ -944,6 +953,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
}
if (node.resize) {
+ // noinspection unchecked
table = (TableEntry<V>[])node.getValuePlain();
continue table_loop;
}
@@ -1058,6 +1068,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
}
if (node.resize) {
+ // noinspection unchecked
table = (TableEntry<V>[])node.getValuePlain();
continue table_loop;
}
@@ -1126,6 +1137,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
}
if (node.resize) {
+ // noinspection unchecked
table = (TableEntry<V>[])node.getValuePlain();
continue table_loop;
}
@@ -1200,6 +1212,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
}
if (node.resize) {
+ // noinspection unchecked
table = (TableEntry<V>[])node.getValuePlain();
continue table_loop;
}
@@ -1288,6 +1301,11 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
return new EntryIterator<>(this);
}
+ @Override
+ public final Iterator<TableEntry<V>> iterator() {
+ return this.entryIterator();
+ }
+
/**
* Returns an iterator over the keys in this map. The iterator is only guaranteed to see keys that were
* added before the beginning of this call, but it may see keys added during.
@@ -1306,7 +1324,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
protected static final class EntryIterator<V> extends BaseIteratorImpl<V, TableEntry<V>> {
- protected EntryIterator(final ConcurrentLong2ReferenceChainedHashTable<V> map) {
+ public EntryIterator(final ConcurrentLong2ReferenceChainedHashTable<V> map) {
super(map);
}
@@ -1326,7 +1344,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
protected static final class KeyIterator<V> extends BaseIteratorImpl<V, Long> implements PrimitiveIterator.OfLong {
- protected KeyIterator(final ConcurrentLong2ReferenceChainedHashTable<V> map) {
+ public KeyIterator(final ConcurrentLong2ReferenceChainedHashTable<V> map) {
super(map);
}
@@ -1365,7 +1383,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
protected static final class ValueIterator<V> extends BaseIteratorImpl<V, V> {
- protected ValueIterator(final ConcurrentLong2ReferenceChainedHashTable<V> map) {
+ public ValueIterator(final ConcurrentLong2ReferenceChainedHashTable<V> map) {
super(map);
}
@@ -1420,7 +1438,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
@Override
public final void remove() {
- final TableEntry<V> lastReturned = this.nextToReturn;
+ final TableEntry<V> lastReturned = this.lastReturned;
if (lastReturned == null) {
throw new NoSuchElementException();
}
@@ -1492,6 +1510,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
final ResizeChain<V> chain = this.resizeChain;
if (chain == null) {
+ // noinspection unchecked
final TableEntry<V>[] nextTable = (TableEntry<V>[])entry.getValuePlain();
final ResizeChain<V> oldChain = new ResizeChain<>(table, null, null);
@@ -1506,6 +1525,7 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
} else {
ResizeChain<V> currChain = chain.next;
if (currChain == null) {
+ // noinspection unchecked
final TableEntry<V>[] ret = (TableEntry<V>[])entry.getValuePlain();
currChain = new ResizeChain<>(ret, chain, null);
chain.next = currChain;
@@ -1586,11 +1606,11 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
protected static final class ResizeChain<V> {
- protected final TableEntry<V>[] table;
- protected final ResizeChain<V> prev;
- protected ResizeChain<V> next;
+ public final TableEntry<V>[] table;
+ public final ResizeChain<V> prev;
+ public ResizeChain<V> next;
- protected ResizeChain(final TableEntry<V>[] table, final ResizeChain<V> prev, final ResizeChain<V> next) {
+ public ResizeChain(final TableEntry<V>[] table, final ResizeChain<V> prev, final ResizeChain<V> next) {
this.table = table;
this.prev = prev;
this.next = next;
@@ -1600,64 +1620,64 @@ public class ConcurrentLong2ReferenceChainedHashTable<V> {
public static final class TableEntry<V> {
- protected static final VarHandle TABLE_ENTRY_ARRAY_HANDLE = ConcurrentUtil.getArrayHandle(TableEntry[].class);
+ private static final VarHandle TABLE_ENTRY_ARRAY_HANDLE = ConcurrentUtil.getArrayHandle(TableEntry[].class);
- protected final boolean resize;
+ private final boolean resize;
- protected final long key;
+ private final long key;
- protected volatile V value;
- protected static final VarHandle VALUE_HANDLE = ConcurrentUtil.getVarHandle(TableEntry.class, "value", Object.class);
+ private volatile V value;
+ private static final VarHandle VALUE_HANDLE = ConcurrentUtil.getVarHandle(TableEntry.class, "value", Object.class);
- protected final V getValuePlain() {
+ private V getValuePlain() {
//noinspection unchecked
return (V)VALUE_HANDLE.get(this);
}
- protected final V getValueAcquire() {
+ private V getValueAcquire() {
//noinspection unchecked
return (V)VALUE_HANDLE.getAcquire(this);
}
- protected final V getValueVolatile() {
+ private V getValueVolatile() {
//noinspection unchecked
return (V)VALUE_HANDLE.getVolatile(this);
}
- protected final void setValuePlain(final V value) {
+ private void setValuePlain(final V value) {
VALUE_HANDLE.set(this, (Object)value);
}
- protected final void setValueRelease(final V value) {
+ private void setValueRelease(final V value) {
VALUE_HANDLE.setRelease(this, (Object)value);
}
- protected final void setValueVolatile(final V value) {
+ private void setValueVolatile(final V value) {
VALUE_HANDLE.setVolatile(this, (Object)value);
}
- protected volatile TableEntry<V> next;
- protected static final VarHandle NEXT_HANDLE = ConcurrentUtil.getVarHandle(TableEntry.class, "next", TableEntry.class);
+ private volatile TableEntry<V> next;
+ private static final VarHandle NEXT_HANDLE = ConcurrentUtil.getVarHandle(TableEntry.class, "next", TableEntry.class);
- protected final TableEntry<V> getNextPlain() {
+ private TableEntry<V> getNextPlain() {
//noinspection unchecked
return (TableEntry<V>)NEXT_HANDLE.get(this);
}
- protected final TableEntry<V> getNextVolatile() {
+ private TableEntry<V> getNextVolatile() {
//noinspection unchecked
return (TableEntry<V>)NEXT_HANDLE.getVolatile(this);
}
- protected final void setNextPlain(final TableEntry<V> next) {
+ private void setNextPlain(final TableEntry<V> next) {
NEXT_HANDLE.set(this, next);
}
- protected final void setNextRelease(final TableEntry<V> next) {
+ private void setNextRelease(final TableEntry<V> next) {
NEXT_HANDLE.setRelease(this, next);
}
- protected final void setNextVolatile(final TableEntry<V> next) {
+ private void setNextVolatile(final TableEntry<V> next) {
NEXT_HANDLE.setVolatile(this, next);
}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/scheduler/SchedulerThreadPool.java b/src/main/java/ca/spottedleaf/concurrentutil/scheduler/SchedulerThreadPool.java
index 8197ccb1c4e5878dbd8007b5fb514640765ec8e4..85e6ef636d435a0ee4bf3e0760b0c87422c520a1 100644
--- a/src/main/java/ca/spottedleaf/concurrentutil/scheduler/SchedulerThreadPool.java
+++ b/src/main/java/ca/spottedleaf/concurrentutil/scheduler/SchedulerThreadPool.java
@@ -13,6 +13,10 @@ import java.util.concurrent.atomic.AtomicLong;
import java.util.concurrent.locks.LockSupport;
import java.util.function.BooleanSupplier;
+/**
+ * @deprecated To be replaced
+ */
+@Deprecated
public class SchedulerThreadPool {
public static final long DEADLINE_NOT_SET = Long.MIN_VALUE;
@@ -297,7 +301,9 @@ public class SchedulerThreadPool {
* is invoked for any scheduled task - otherwise, {@link #runTasks(BooleanSupplier)} may not be invoked to
* parse intermediate tasks.
* </p>
+ * @deprecated To be replaced
*/
+ @Deprecated
public static abstract class SchedulableTick {
private static final AtomicLong ID_GENERATOR = new AtomicLong();
public final long id = ID_GENERATOR.getAndIncrement();
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/set/LinkedSortedSet.java b/src/main/java/ca/spottedleaf/concurrentutil/set/LinkedSortedSet.java
index 212bc9ae2fc7d37d4a089a2921b00de1e97f7cc1..82c4c11b0b564c97ac92bd5f54e3754a7ba95184 100644
--- a/src/main/java/ca/spottedleaf/concurrentutil/set/LinkedSortedSet.java
+++ b/src/main/java/ca/spottedleaf/concurrentutil/set/LinkedSortedSet.java
@@ -8,8 +8,8 @@ public final class LinkedSortedSet<E> implements Iterable<E> {
public final Comparator<? super E> comparator;
- protected Link<E> head;
- protected Link<E> tail;
+ private Link<E> head;
+ private Link<E> tail;
public LinkedSortedSet() {
this((Comparator)Comparator.naturalOrder());
@@ -257,8 +257,6 @@ public final class LinkedSortedSet<E> implements Iterable<E> {
private Link<E> prev;
private Link<E> next;
- private Link() {}
-
private Link(final E element) {
this.element = element;
}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/set/LinkedUnsortedList.java b/src/main/java/ca/spottedleaf/concurrentutil/set/LinkedUnsortedList.java
new file mode 100644
index 0000000000000000000000000000000000000000..bd8eb4f25d1dee00fbf9c05c14b0d94c5c641a55
--- /dev/null
+++ b/src/main/java/ca/spottedleaf/concurrentutil/set/LinkedUnsortedList.java
@@ -0,0 +1,204 @@
+package ca.spottedleaf.concurrentutil.set;
+
+import java.util.Iterator;
+import java.util.NoSuchElementException;
+import java.util.Objects;
+
+public final class LinkedUnsortedList<E> implements Iterable<E> {
+
+ private Link<E> head;
+ private Link<E> tail;
+
+ public LinkedUnsortedList() {}
+
+ public void clear() {
+ this.head = this.tail = null;
+ }
+
+ public boolean isEmpty() {
+ return this.head == null;
+ }
+
+ public E first() {
+ final Link<E> head = this.head;
+ return head == null ? null : head.element;
+ }
+
+ public E last() {
+ final Link<E> tail = this.tail;
+ return tail == null ? null : tail.element;
+ }
+
+ public boolean containsFirst(final E element) {
+ for (Link<E> curr = this.head; curr != null; curr = curr.next) {
+ if (Objects.equals(element, curr.element)) {
+ return true;
+ }
+ }
+ return false;
+ }
+
+ public boolean containsLast(final E element) {
+ for (Link<E> curr = this.tail; curr != null; curr = curr.prev) {
+ if (Objects.equals(element, curr.element)) {
+ return true;
+ }
+ }
+ return false;
+ }
+
+ private void removeNode(final Link<E> node) {
+ final Link<E> prev = node.prev;
+ final Link<E> next = node.next;
+
+ // help GC
+ node.element = null;
+ node.prev = null;
+ node.next = null;
+
+ if (prev == null) {
+ this.head = next;
+ } else {
+ prev.next = next;
+ }
+
+ if (next == null) {
+ this.tail = prev;
+ } else {
+ next.prev = prev;
+ }
+ }
+
+ public boolean remove(final Link<E> link) {
+ if (link.element == null) {
+ return false;
+ }
+
+ this.removeNode(link);
+ return true;
+ }
+
+ public boolean removeFirst(final E element) {
+ for (Link<E> curr = this.head; curr != null; curr = curr.next) {
+ if (Objects.equals(element, curr.element)) {
+ this.removeNode(curr);
+ return true;
+ }
+ }
+ return false;
+ }
+
+ public boolean removeLast(final E element) {
+ for (Link<E> curr = this.tail; curr != null; curr = curr.prev) {
+ if (Objects.equals(element, curr.element)) {
+ this.removeNode(curr);
+ return true;
+ }
+ }
+ return false;
+ }
+
+ @Override
+ public Iterator<E> iterator() {
+ return new Iterator<>() {
+ private Link<E> next = LinkedUnsortedList.this.head;
+
+ @Override
+ public boolean hasNext() {
+ return this.next != null;
+ }
+
+ @Override
+ public E next() {
+ final Link<E> next = this.next;
+ if (next == null) {
+ throw new NoSuchElementException();
+ }
+ this.next = next.next;
+ return next.element;
+ }
+ };
+ }
+
+ public E pollFirst() {
+ final Link<E> head = this.head;
+ if (head == null) {
+ return null;
+ }
+
+ final E ret = head.element;
+ final Link<E> next = head.next;
+
+ // unlink head
+ this.head = next;
+ if (next == null) {
+ this.tail = null;
+ } else {
+ next.prev = null;
+ }
+
+ // help GC
+ head.element = null;
+ head.next = null;
+
+ return ret;
+ }
+
+ public E pollLast() {
+ final Link<E> tail = this.tail;
+ if (tail == null) {
+ return null;
+ }
+
+ final E ret = tail.element;
+ final Link<E> prev = tail.prev;
+
+ // unlink tail
+ this.tail = prev;
+ if (prev == null) {
+ this.head = null;
+ } else {
+ prev.next = null;
+ }
+
+ // help GC
+ tail.element = null;
+ tail.prev = null;
+
+ return ret;
+ }
+
+ public Link<E> addLast(final E element) {
+ final Link<E> curr = this.tail;
+ if (curr != null) {
+ return this.tail = new Link<>(element, curr, null);
+ } else {
+ return this.head = this.tail = new Link<>(element);
+ }
+ }
+
+ public Link<E> addFirst(final E element) {
+ final Link<E> curr = this.head;
+ if (curr != null) {
+ return this.head = new Link<>(element, null, curr);
+ } else {
+ return this.head = this.tail = new Link<>(element);
+ }
+ }
+
+ public static final class Link<E> {
+ private E element;
+ private Link<E> prev;
+ private Link<E> next;
+
+ private Link(final E element) {
+ this.element = element;
+ }
+
+ private Link(final E element, final Link<E> prev, final Link<E> next) {
+ this.element = element;
+ this.prev = prev;
+ this.next = next;
+ }
+ }
+}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/util/ArrayUtil.java b/src/main/java/ca/spottedleaf/concurrentutil/util/ArrayUtil.java
deleted file mode 100644
index ebb1ab06165addb173fea4d295001fe37f4e79d3..0000000000000000000000000000000000000000
--- a/src/main/java/ca/spottedleaf/concurrentutil/util/ArrayUtil.java
+++ /dev/null
@@ -1,816 +0,0 @@
-package ca.spottedleaf.concurrentutil.util;
-
-import java.lang.invoke.VarHandle;
-
-public final class ArrayUtil {
-
- public static final VarHandle BOOLEAN_ARRAY_HANDLE = ConcurrentUtil.getArrayHandle(boolean[].class);
-
- public static final VarHandle BYTE_ARRAY_HANDLE = ConcurrentUtil.getArrayHandle(byte[].class);
-
- public static final VarHandle SHORT_ARRAY_HANDLE = ConcurrentUtil.getArrayHandle(short[].class);
-
- public static final VarHandle INT_ARRAY_HANDLE = ConcurrentUtil.getArrayHandle(int[].class);
-
- public static final VarHandle LONG_ARRAY_HANDLE = ConcurrentUtil.getArrayHandle(long[].class);
-
- public static final VarHandle OBJECT_ARRAY_HANDLE = ConcurrentUtil.getArrayHandle(Object[].class);
-
- private ArrayUtil() {
- throw new RuntimeException();
- }
-
- /* byte array */
-
- public static byte getPlain(final byte[] array, final int index) {
- return (byte)BYTE_ARRAY_HANDLE.get(array, index);
- }
-
- public static byte getOpaque(final byte[] array, final int index) {
- return (byte)BYTE_ARRAY_HANDLE.getOpaque(array, index);
- }
-
- public static byte getAcquire(final byte[] array, final int index) {
- return (byte)BYTE_ARRAY_HANDLE.getAcquire(array, index);
- }
-
- public static byte getVolatile(final byte[] array, final int index) {
- return (byte)BYTE_ARRAY_HANDLE.getVolatile(array, index);
- }
-
- public static void setPlain(final byte[] array, final int index, final byte value) {
- BYTE_ARRAY_HANDLE.set(array, index, value);
- }
-
- public static void setOpaque(final byte[] array, final int index, final byte value) {
- BYTE_ARRAY_HANDLE.setOpaque(array, index, value);
- }
-
- public static void setRelease(final byte[] array, final int index, final byte value) {
- BYTE_ARRAY_HANDLE.setRelease(array, index, value);
- }
-
- public static void setVolatile(final byte[] array, final int index, final byte value) {
- BYTE_ARRAY_HANDLE.setVolatile(array, index, value);
- }
-
- public static void setVolatileContended(final byte[] array, final int index, final byte param) {
- int failures = 0;
-
- for (byte curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, param))) {
- return;
- }
- }
- }
-
- public static byte compareAndExchangeVolatile(final byte[] array, final int index, final byte expect, final byte update) {
- return (byte)BYTE_ARRAY_HANDLE.compareAndExchange(array, index, expect, update);
- }
-
- public static byte getAndAddVolatile(final byte[] array, final int index, final byte param) {
- return (byte)BYTE_ARRAY_HANDLE.getAndAdd(array, index, param);
- }
-
- public static byte getAndAndVolatile(final byte[] array, final int index, final byte param) {
- return (byte)BYTE_ARRAY_HANDLE.getAndBitwiseAnd(array, index, param);
- }
-
- public static byte getAndOrVolatile(final byte[] array, final int index, final byte param) {
- return (byte)BYTE_ARRAY_HANDLE.getAndBitwiseOr(array, index, param);
- }
-
- public static byte getAndXorVolatile(final byte[] array, final int index, final byte param) {
- return (byte)BYTE_ARRAY_HANDLE.getAndBitwiseXor(array, index, param);
- }
-
- public static byte getAndSetVolatile(final byte[] array, final int index, final byte param) {
- return (byte)BYTE_ARRAY_HANDLE.getAndSet(array, index, param);
- }
-
- public static byte compareAndExchangeVolatileContended(final byte[] array, final int index, final byte expect, final byte update) {
- return (byte)BYTE_ARRAY_HANDLE.compareAndExchange(array, index, expect, update);
- }
-
- public static byte getAndAddVolatileContended(final byte[] array, final int index, final byte param) {
- int failures = 0;
-
- for (byte curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (byte) (curr + param)))) {
- return curr;
- }
- }
- }
-
- public static byte getAndAndVolatileContended(final byte[] array, final int index, final byte param) {
- int failures = 0;
-
- for (byte curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (byte) (curr & param)))) {
- return curr;
- }
- }
- }
-
- public static byte getAndOrVolatileContended(final byte[] array, final int index, final byte param) {
- int failures = 0;
-
- for (byte curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (byte) (curr | param)))) {
- return curr;
- }
- }
- }
-
- public static byte getAndXorVolatileContended(final byte[] array, final int index, final byte param) {
- int failures = 0;
-
- for (byte curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (byte) (curr ^ param)))) {
- return curr;
- }
- }
- }
-
- public static byte getAndSetVolatileContended(final byte[] array, final int index, final byte param) {
- int failures = 0;
-
- for (byte curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, param))) {
- return curr;
- }
- }
- }
-
- /* short array */
-
- public static short getPlain(final short[] array, final int index) {
- return (short)SHORT_ARRAY_HANDLE.get(array, index);
- }
-
- public static short getOpaque(final short[] array, final int index) {
- return (short)SHORT_ARRAY_HANDLE.getOpaque(array, index);
- }
-
- public static short getAcquire(final short[] array, final int index) {
- return (short)SHORT_ARRAY_HANDLE.getAcquire(array, index);
- }
-
- public static short getVolatile(final short[] array, final int index) {
- return (short)SHORT_ARRAY_HANDLE.getVolatile(array, index);
- }
-
- public static void setPlain(final short[] array, final int index, final short value) {
- SHORT_ARRAY_HANDLE.set(array, index, value);
- }
-
- public static void setOpaque(final short[] array, final int index, final short value) {
- SHORT_ARRAY_HANDLE.setOpaque(array, index, value);
- }
-
- public static void setRelease(final short[] array, final int index, final short value) {
- SHORT_ARRAY_HANDLE.setRelease(array, index, value);
- }
-
- public static void setVolatile(final short[] array, final int index, final short value) {
- SHORT_ARRAY_HANDLE.setVolatile(array, index, value);
- }
-
- public static void setVolatileContended(final short[] array, final int index, final short param) {
- int failures = 0;
-
- for (short curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, param))) {
- return;
- }
- }
- }
-
- public static short compareAndExchangeVolatile(final short[] array, final int index, final short expect, final short update) {
- return (short)SHORT_ARRAY_HANDLE.compareAndExchange(array, index, expect, update);
- }
-
- public static short getAndAddVolatile(final short[] array, final int index, final short param) {
- return (short)SHORT_ARRAY_HANDLE.getAndAdd(array, index, param);
- }
-
- public static short getAndAndVolatile(final short[] array, final int index, final short param) {
- return (short)SHORT_ARRAY_HANDLE.getAndBitwiseAnd(array, index, param);
- }
-
- public static short getAndOrVolatile(final short[] array, final int index, final short param) {
- return (short)SHORT_ARRAY_HANDLE.getAndBitwiseOr(array, index, param);
- }
-
- public static short getAndXorVolatile(final short[] array, final int index, final short param) {
- return (short)SHORT_ARRAY_HANDLE.getAndBitwiseXor(array, index, param);
- }
-
- public static short getAndSetVolatile(final short[] array, final int index, final short param) {
- return (short)SHORT_ARRAY_HANDLE.getAndSet(array, index, param);
- }
-
- public static short compareAndExchangeVolatileContended(final short[] array, final int index, final short expect, final short update) {
- return (short)SHORT_ARRAY_HANDLE.compareAndExchange(array, index, expect, update);
- }
-
- public static short getAndAddVolatileContended(final short[] array, final int index, final short param) {
- int failures = 0;
-
- for (short curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (short) (curr + param)))) {
- return curr;
- }
- }
- }
-
- public static short getAndAndVolatileContended(final short[] array, final int index, final short param) {
- int failures = 0;
-
- for (short curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (short) (curr & param)))) {
- return curr;
- }
- }
- }
-
- public static short getAndOrVolatileContended(final short[] array, final int index, final short param) {
- int failures = 0;
-
- for (short curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (short) (curr | param)))) {
- return curr;
- }
- }
- }
-
- public static short getAndXorVolatileContended(final short[] array, final int index, final short param) {
- int failures = 0;
-
- for (short curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (short) (curr ^ param)))) {
- return curr;
- }
- }
- }
-
- public static short getAndSetVolatileContended(final short[] array, final int index, final short param) {
- int failures = 0;
-
- for (short curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, param))) {
- return curr;
- }
- }
- }
-
- /* int array */
-
- public static int getPlain(final int[] array, final int index) {
- return (int)INT_ARRAY_HANDLE.get(array, index);
- }
-
- public static int getOpaque(final int[] array, final int index) {
- return (int)INT_ARRAY_HANDLE.getOpaque(array, index);
- }
-
- public static int getAcquire(final int[] array, final int index) {
- return (int)INT_ARRAY_HANDLE.getAcquire(array, index);
- }
-
- public static int getVolatile(final int[] array, final int index) {
- return (int)INT_ARRAY_HANDLE.getVolatile(array, index);
- }
-
- public static void setPlain(final int[] array, final int index, final int value) {
- INT_ARRAY_HANDLE.set(array, index, value);
- }
-
- public static void setOpaque(final int[] array, final int index, final int value) {
- INT_ARRAY_HANDLE.setOpaque(array, index, value);
- }
-
- public static void setRelease(final int[] array, final int index, final int value) {
- INT_ARRAY_HANDLE.setRelease(array, index, value);
- }
-
- public static void setVolatile(final int[] array, final int index, final int value) {
- INT_ARRAY_HANDLE.setVolatile(array, index, value);
- }
-
- public static void setVolatileContended(final int[] array, final int index, final int param) {
- int failures = 0;
-
- for (int curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, param))) {
- return;
- }
- }
- }
-
- public static int compareAndExchangeVolatile(final int[] array, final int index, final int expect, final int update) {
- return (int)INT_ARRAY_HANDLE.compareAndExchange(array, index, expect, update);
- }
-
- public static int getAndAddVolatile(final int[] array, final int index, final int param) {
- return (int)INT_ARRAY_HANDLE.getAndAdd(array, index, param);
- }
-
- public static int getAndAndVolatile(final int[] array, final int index, final int param) {
- return (int)INT_ARRAY_HANDLE.getAndBitwiseAnd(array, index, param);
- }
-
- public static int getAndOrVolatile(final int[] array, final int index, final int param) {
- return (int)INT_ARRAY_HANDLE.getAndBitwiseOr(array, index, param);
- }
-
- public static int getAndXorVolatile(final int[] array, final int index, final int param) {
- return (int)INT_ARRAY_HANDLE.getAndBitwiseXor(array, index, param);
- }
-
- public static int getAndSetVolatile(final int[] array, final int index, final int param) {
- return (int)INT_ARRAY_HANDLE.getAndSet(array, index, param);
- }
-
- public static int compareAndExchangeVolatileContended(final int[] array, final int index, final int expect, final int update) {
- return (int)INT_ARRAY_HANDLE.compareAndExchange(array, index, expect, update);
- }
-
- public static int getAndAddVolatileContended(final int[] array, final int index, final int param) {
- int failures = 0;
-
- for (int curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (int) (curr + param)))) {
- return curr;
- }
- }
- }
-
- public static int getAndAndVolatileContended(final int[] array, final int index, final int param) {
- int failures = 0;
-
- for (int curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (int) (curr & param)))) {
- return curr;
- }
- }
- }
-
- public static int getAndOrVolatileContended(final int[] array, final int index, final int param) {
- int failures = 0;
-
- for (int curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (int) (curr | param)))) {
- return curr;
- }
- }
- }
-
- public static int getAndXorVolatileContended(final int[] array, final int index, final int param) {
- int failures = 0;
-
- for (int curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (int) (curr ^ param)))) {
- return curr;
- }
- }
- }
-
- public static int getAndSetVolatileContended(final int[] array, final int index, final int param) {
- int failures = 0;
-
- for (int curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, param))) {
- return curr;
- }
- }
- }
-
- /* long array */
-
- public static long getPlain(final long[] array, final int index) {
- return (long)LONG_ARRAY_HANDLE.get(array, index);
- }
-
- public static long getOpaque(final long[] array, final int index) {
- return (long)LONG_ARRAY_HANDLE.getOpaque(array, index);
- }
-
- public static long getAcquire(final long[] array, final int index) {
- return (long)LONG_ARRAY_HANDLE.getAcquire(array, index);
- }
-
- public static long getVolatile(final long[] array, final int index) {
- return (long)LONG_ARRAY_HANDLE.getVolatile(array, index);
- }
-
- public static void setPlain(final long[] array, final int index, final long value) {
- LONG_ARRAY_HANDLE.set(array, index, value);
- }
-
- public static void setOpaque(final long[] array, final int index, final long value) {
- LONG_ARRAY_HANDLE.setOpaque(array, index, value);
- }
-
- public static void setRelease(final long[] array, final int index, final long value) {
- LONG_ARRAY_HANDLE.setRelease(array, index, value);
- }
-
- public static void setVolatile(final long[] array, final int index, final long value) {
- LONG_ARRAY_HANDLE.setVolatile(array, index, value);
- }
-
- public static void setVolatileContended(final long[] array, final int index, final long param) {
- int failures = 0;
-
- for (long curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, param))) {
- return;
- }
- }
- }
-
- public static long compareAndExchangeVolatile(final long[] array, final int index, final long expect, final long update) {
- return (long)LONG_ARRAY_HANDLE.compareAndExchange(array, index, expect, update);
- }
-
- public static long getAndAddVolatile(final long[] array, final int index, final long param) {
- return (long)LONG_ARRAY_HANDLE.getAndAdd(array, index, param);
- }
-
- public static long getAndAndVolatile(final long[] array, final int index, final long param) {
- return (long)LONG_ARRAY_HANDLE.getAndBitwiseAnd(array, index, param);
- }
-
- public static long getAndOrVolatile(final long[] array, final int index, final long param) {
- return (long)LONG_ARRAY_HANDLE.getAndBitwiseOr(array, index, param);
- }
-
- public static long getAndXorVolatile(final long[] array, final int index, final long param) {
- return (long)LONG_ARRAY_HANDLE.getAndBitwiseXor(array, index, param);
- }
-
- public static long getAndSetVolatile(final long[] array, final int index, final long param) {
- return (long)LONG_ARRAY_HANDLE.getAndSet(array, index, param);
- }
-
- public static long compareAndExchangeVolatileContended(final long[] array, final int index, final long expect, final long update) {
- return (long)LONG_ARRAY_HANDLE.compareAndExchange(array, index, expect, update);
- }
-
- public static long getAndAddVolatileContended(final long[] array, final int index, final long param) {
- int failures = 0;
-
- for (long curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (long) (curr + param)))) {
- return curr;
- }
- }
- }
-
- public static long getAndAndVolatileContended(final long[] array, final int index, final long param) {
- int failures = 0;
-
- for (long curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (long) (curr & param)))) {
- return curr;
- }
- }
- }
-
- public static long getAndOrVolatileContended(final long[] array, final int index, final long param) {
- int failures = 0;
-
- for (long curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (long) (curr | param)))) {
- return curr;
- }
- }
- }
-
- public static long getAndXorVolatileContended(final long[] array, final int index, final long param) {
- int failures = 0;
-
- for (long curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (long) (curr ^ param)))) {
- return curr;
- }
- }
- }
-
- public static long getAndSetVolatileContended(final long[] array, final int index, final long param) {
- int failures = 0;
-
- for (long curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, param))) {
- return curr;
- }
- }
- }
-
- /* boolean array */
-
- public static boolean getPlain(final boolean[] array, final int index) {
- return (boolean)BOOLEAN_ARRAY_HANDLE.get(array, index);
- }
-
- public static boolean getOpaque(final boolean[] array, final int index) {
- return (boolean)BOOLEAN_ARRAY_HANDLE.getOpaque(array, index);
- }
-
- public static boolean getAcquire(final boolean[] array, final int index) {
- return (boolean)BOOLEAN_ARRAY_HANDLE.getAcquire(array, index);
- }
-
- public static boolean getVolatile(final boolean[] array, final int index) {
- return (boolean)BOOLEAN_ARRAY_HANDLE.getVolatile(array, index);
- }
-
- public static void setPlain(final boolean[] array, final int index, final boolean value) {
- BOOLEAN_ARRAY_HANDLE.set(array, index, value);
- }
-
- public static void setOpaque(final boolean[] array, final int index, final boolean value) {
- BOOLEAN_ARRAY_HANDLE.setOpaque(array, index, value);
- }
-
- public static void setRelease(final boolean[] array, final int index, final boolean value) {
- BOOLEAN_ARRAY_HANDLE.setRelease(array, index, value);
- }
-
- public static void setVolatile(final boolean[] array, final int index, final boolean value) {
- BOOLEAN_ARRAY_HANDLE.setVolatile(array, index, value);
- }
-
- public static void setVolatileContended(final boolean[] array, final int index, final boolean param) {
- int failures = 0;
-
- for (boolean curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, param))) {
- return;
- }
- }
- }
-
- public static boolean compareAndExchangeVolatile(final boolean[] array, final int index, final boolean expect, final boolean update) {
- return (boolean)BOOLEAN_ARRAY_HANDLE.compareAndExchange(array, index, expect, update);
- }
-
- public static boolean getAndOrVolatile(final boolean[] array, final int index, final boolean param) {
- return (boolean)BOOLEAN_ARRAY_HANDLE.getAndBitwiseOr(array, index, param);
- }
-
- public static boolean getAndXorVolatile(final boolean[] array, final int index, final boolean param) {
- return (boolean)BOOLEAN_ARRAY_HANDLE.getAndBitwiseXor(array, index, param);
- }
-
- public static boolean getAndSetVolatile(final boolean[] array, final int index, final boolean param) {
- return (boolean)BOOLEAN_ARRAY_HANDLE.getAndSet(array, index, param);
- }
-
- public static boolean compareAndExchangeVolatileContended(final boolean[] array, final int index, final boolean expect, final boolean update) {
- return (boolean)BOOLEAN_ARRAY_HANDLE.compareAndExchange(array, index, expect, update);
- }
-
- public static boolean getAndAndVolatileContended(final boolean[] array, final int index, final boolean param) {
- int failures = 0;
-
- for (boolean curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (boolean) (curr & param)))) {
- return curr;
- }
- }
- }
-
- public static boolean getAndOrVolatileContended(final boolean[] array, final int index, final boolean param) {
- int failures = 0;
-
- for (boolean curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (boolean) (curr | param)))) {
- return curr;
- }
- }
- }
-
- public static boolean getAndXorVolatileContended(final boolean[] array, final int index, final boolean param) {
- int failures = 0;
-
- for (boolean curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, (boolean) (curr ^ param)))) {
- return curr;
- }
- }
- }
-
- public static boolean getAndSetVolatileContended(final boolean[] array, final int index, final boolean param) {
- int failures = 0;
-
- for (boolean curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, param))) {
- return curr;
- }
- }
- }
-
- @SuppressWarnings("unchecked")
- public static <T> T getPlain(final T[] array, final int index) {
- final Object ret = OBJECT_ARRAY_HANDLE.get((Object[])array, index);
- return (T)ret;
- }
-
- @SuppressWarnings("unchecked")
- public static <T> T getOpaque(final T[] array, final int index) {
- final Object ret = OBJECT_ARRAY_HANDLE.getOpaque((Object[])array, index);
- return (T)ret;
- }
-
- @SuppressWarnings("unchecked")
- public static <T> T getAcquire(final T[] array, final int index) {
- final Object ret = OBJECT_ARRAY_HANDLE.getAcquire((Object[])array, index);
- return (T)ret;
- }
-
- @SuppressWarnings("unchecked")
- public static <T> T getVolatile(final T[] array, final int index) {
- final Object ret = OBJECT_ARRAY_HANDLE.getVolatile((Object[])array, index);
- return (T)ret;
- }
-
- public static <T> void setPlain(final T[] array, final int index, final T value) {
- OBJECT_ARRAY_HANDLE.set((Object[])array, index, (Object)value);
- }
-
- public static <T> void setOpaque(final T[] array, final int index, final T value) {
- OBJECT_ARRAY_HANDLE.setOpaque((Object[])array, index, (Object)value);
- }
-
- public static <T> void setRelease(final T[] array, final int index, final T value) {
- OBJECT_ARRAY_HANDLE.setRelease((Object[])array, index, (Object)value);
- }
-
- public static <T> void setVolatile(final T[] array, final int index, final T value) {
- OBJECT_ARRAY_HANDLE.setVolatile((Object[])array, index, (Object)value);
- }
-
- public static <T> void setVolatileContended(final T[] array, final int index, final T param) {
- int failures = 0;
-
- for (T curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, param))) {
- return;
- }
- }
- }
-
- @SuppressWarnings("unchecked")
- public static <T> T compareAndExchangeVolatile(final T[] array, final int index, final T expect, final T update) {
- final Object ret = OBJECT_ARRAY_HANDLE.compareAndExchange((Object[])array, index, (Object)expect, (Object)update);
- return (T)ret;
- }
-
- @SuppressWarnings("unchecked")
- public static <T> T getAndSetVolatile(final T[] array, final int index, final T param) {
- final Object ret = BYTE_ARRAY_HANDLE.getAndSet((Object[])array, index, (Object)param);
- return (T)ret;
- }
-
- @SuppressWarnings("unchecked")
- public static <T> T compareAndExchangeVolatileContended(final T[] array, final int index, final T expect, final T update) {
- final Object ret = OBJECT_ARRAY_HANDLE.compareAndExchange((Object[])array, index, (Object)expect, (Object)update);
- return (T)ret;
- }
-
- public static <T> T getAndSetVolatileContended(final T[] array, final int index, final T param) {
- int failures = 0;
-
- for (T curr = getVolatile(array, index);;++failures) {
- for (int i = 0; i < failures; ++i) {
- ConcurrentUtil.backoff();
- }
-
- if (curr == (curr = compareAndExchangeVolatileContended(array, index, curr, param))) {
- return curr;
- }
- }
- }
-}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/util/IntegerUtil.java b/src/main/java/ca/spottedleaf/concurrentutil/util/IntegerUtil.java
index 77699c5fa9681f9ec7aa05cbb50eb60828e193ab..9d7b9b8158cd01d12adbd7896ff77bee9828e101 100644
--- a/src/main/java/ca/spottedleaf/concurrentutil/util/IntegerUtil.java
+++ b/src/main/java/ca/spottedleaf/concurrentutil/util/IntegerUtil.java
@@ -170,6 +170,26 @@ public final class IntegerUtil {
return (mask ^ val) - mask; // if val < 0, then (0 ^ val) - 0 else (-1 ^ val) + 1
}
+ // https://lemire.me/blog/2019/02/08/faster-remainders-when-the-divisor-is-a-constant-beating-compilers-and-libdivide
+ /**
+ *
+ * Usage:
+ * <pre>
+ * {@code
+ * static final long mult = getSimpleMultiplier(divisor, bits);
+ * long x = ...;
+ * long magic = x * mult;
+ * long divQ = magic >>> bits;
+ * long divR = ((magic & ((1 << bits) - 1)) * divisor) >>> bits;
+ * }
+ * </pre>
+ *
+ * @param bits The number of bits of precision for the returned result
+ */
+ public static long getUnsignedDivisorMagic(final long divisor, final int bits) {
+ return (((1L << bits) - 1L) / divisor) + 1;
+ }
+
private IntegerUtil() {
throw new RuntimeException();
}
diff --git a/src/main/java/ca/spottedleaf/concurrentutil/util/Priority.java b/src/main/java/ca/spottedleaf/concurrentutil/util/Priority.java
new file mode 100644
index 0000000000000000000000000000000000000000..2919bbaa07b70f182438c3be8f9ebbe0649809b6
--- /dev/null
+++ b/src/main/java/ca/spottedleaf/concurrentutil/util/Priority.java
@@ -0,0 +1,145 @@
+package ca.spottedleaf.concurrentutil.util;
+
+public enum Priority {
+
+ /**
+ * Priority value indicating the task has completed or is being completed.
+ * This priority cannot be used to schedule tasks.
+ */
+ COMPLETING(-1),
+
+ /**
+ * Absolute highest priority, should only be used for when a task is blocking a time-critical thread.
+ */
+ BLOCKING(),
+
+ /**
+ * Should only be used for urgent but not time-critical tasks.
+ */
+ HIGHEST(),
+
+ /**
+ * Two priorities above normal.
+ */
+ HIGHER(),
+
+ /**
+ * One priority above normal.
+ */
+ HIGH(),
+
+ /**
+ * Default priority.
+ */
+ NORMAL(),
+
+ /**
+ * One priority below normal.
+ */
+ LOW(),
+
+ /**
+ * Two priorities below normal.
+ */
+ LOWER(),
+
+ /**
+ * Use for tasks that should eventually execute, but are not needed to.
+ */
+ LOWEST(),
+
+ /**
+ * Use for tasks that can be delayed indefinitely.
+ */
+ IDLE();
+
+ // returns whether the priority can be scheduled
+ public static boolean isValidPriority(final Priority priority) {
+ return priority != null && priority != priority.COMPLETING;
+ }
+
+ // returns the higher priority of the two
+ public static Priority max(final Priority p1, final Priority p2) {
+ return p1.isHigherOrEqualPriority(p2) ? p1 : p2;
+ }
+
+ // returns the lower priroity of the two
+ public static Priority min(final Priority p1, final Priority p2) {
+ return p1.isLowerOrEqualPriority(p2) ? p1 : p2;
+ }
+
+ public boolean isHigherOrEqualPriority(final Priority than) {
+ return this.priority <= than.priority;
+ }
+
+ public boolean isHigherPriority(final Priority than) {
+ return this.priority < than.priority;
+ }
+
+ public boolean isLowerOrEqualPriority(final Priority than) {
+ return this.priority >= than.priority;
+ }
+
+ public boolean isLowerPriority(final Priority than) {
+ return this.priority > than.priority;
+ }
+
+ public boolean isHigherOrEqualPriority(final int than) {
+ return this.priority <= than;
+ }
+
+ public boolean isHigherPriority(final int than) {
+ return this.priority < than;
+ }
+
+ public boolean isLowerOrEqualPriority(final int than) {
+ return this.priority >= than;
+ }
+
+ public boolean isLowerPriority(final int than) {
+ return this.priority > than;
+ }
+
+ public static boolean isHigherOrEqualPriority(final int priority, final int than) {
+ return priority <= than;
+ }
+
+ public static boolean isHigherPriority(final int priority, final int than) {
+ return priority < than;
+ }
+
+ public static boolean isLowerOrEqualPriority(final int priority, final int than) {
+ return priority >= than;
+ }
+
+ public static boolean isLowerPriority(final int priority, final int than) {
+ return priority > than;
+ }
+
+ static final Priority[] PRIORITIES = Priority.values();
+
+ /** includes special priorities */
+ public static final int TOTAL_PRIORITIES = PRIORITIES.length;
+
+ public static final int TOTAL_SCHEDULABLE_PRIORITIES = TOTAL_PRIORITIES - 1;
+
+ public static Priority getPriority(final int priority) {
+ return PRIORITIES[priority + 1];
+ }
+
+ private static int priorityCounter;
+
+ private static int nextCounter() {
+ return priorityCounter++;
+ }
+
+ public final int priority;
+
+ private Priority() {
+ this(nextCounter());
+ }
+
+ private Priority(final int priority) {
+ this.priority = priority;
+ }
+}
\ No newline at end of file
Reference in a new issue View git blame Copy permalink