Repo-Mirror/Leaf
9
0
Fork 0
mirror of https://github.com/Winds-Studio/Leaf.git synced 2025-12-19 15:09:25 +00:00
Code Issues Releases Activity

fix unpark race

Browse Source
This commit is contained in:
hayanesuru
2025-07-16 15:24:59 +09:00
parent cdd379f424
commit cecd8d751a
1 changed files with 22 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
leaf-server/src/main/java/org/dreeam/leaf/async/FixedThreadExecutor.java
View File

@@ -5,10 +5,12 @@ import org.dreeam.leaf.util.queue.MpmcQueue;
import java.util.concurrent.Callable;
import java.util.concurrent.FutureTask;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.locks.LockSupport;
public final class FixedThreadExecutor {
private final Thread[] workers;
private final Thread[] threads;
private final Worker[] workers;
public final MpmcQueue<Runnable> channel;
private static volatile boolean SHUTDOWN = false;
@@ -16,15 +18,17 @@ public final class FixedThreadExecutor {
if (numThreads <= 0) {
throw new IllegalArgumentException();
}
this.workers = new Thread[numThreads];
this.threads = new Thread[numThreads];
this.workers = new Worker[numThreads];
this.channel = new MpmcQueue<>(Runnable.class, queue);
for (int i = 0; i < numThreads; i++) {
workers[i] = Thread.ofPlatform()
workers[i] = new Worker(channel, new AtomicBoolean(false));
threads[i] = Thread.ofPlatform()
.uncaughtExceptionHandler(Util::onThreadException)
.daemon(false)
.priority(Thread.NORM_PRIORITY)
.name(prefix + " - " + i)
.start(new Worker(channel));
.start(workers[i]);
}
}
@@ -41,15 +45,17 @@ public final class FixedThreadExecutor {
}
public void unpack() {
int size = Math.min(Math.max(1, channel.length()), workers.length);
int size = Math.min(Math.max(1, channel.length()), threads.length);
for (int i = 0; i < size; i++) {
LockSupport.unpark(workers[i]);
if (workers[i].parked.get()) {
LockSupport.unpark(threads[i]);
}
}
}
public void shutdown() {
SHUTDOWN = true;
for (final Thread worker : workers) {
for (final Thread worker : threads) {
LockSupport.unpark(worker);
}
}
@@ -57,7 +63,7 @@ public final class FixedThreadExecutor {
public void join(long timeoutMillis) throws InterruptedException {
final long startTime = System.currentTimeMillis();
for (final Thread worker : workers) {
for (final Thread worker : threads) {
final long remaining = timeoutMillis - System.currentTimeMillis() + startTime;
if (remaining <= 0) {
return;
@@ -69,19 +75,24 @@ public final class FixedThreadExecutor {
}
}
private record Worker(MpmcQueue<Runnable> channel) implements Runnable {
private record Worker(MpmcQueue<Runnable> channel, AtomicBoolean parked) implements Runnable {
@Override
public void run() {
while (true) {
final Runnable task = channel.recv();
if (task != null) {
parked.set(false);
task.run();
} else if (SHUTDOWN) {
break;
} else if (channel.isEmpty()) {
Thread.yield();
if (parked.compareAndSet(false, true)) {
if (channel.isEmpty()) {
LockSupport.park();
} else {
parked.set(false);
}
}
}
}