将 EventManager 中的 synchronized 替换为 ReentrantLock (#4562)
This commit is contained in:
@@ -22,6 +22,7 @@ import org.jackhuang.hmcl.util.SimpleMultimap;
|
|||||||
import java.lang.ref.WeakReference;
|
import java.lang.ref.WeakReference;
|
||||||
import java.util.EnumMap;
|
import java.util.EnumMap;
|
||||||
import java.util.concurrent.CopyOnWriteArraySet;
|
import java.util.concurrent.CopyOnWriteArraySet;
|
||||||
|
import java.util.concurrent.locks.ReentrantLock;
|
||||||
import java.util.function.Consumer;
|
import java.util.function.Consumer;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -30,6 +31,7 @@ import java.util.function.Consumer;
|
|||||||
*/
|
*/
|
||||||
public final class EventManager<T extends Event> {
|
public final class EventManager<T extends Event> {
|
||||||
|
|
||||||
|
private final ReentrantLock lock = new ReentrantLock();
|
||||||
private final SimpleMultimap<EventPriority, Consumer<T>, CopyOnWriteArraySet<Consumer<T>>> handlers
|
private final SimpleMultimap<EventPriority, Consumer<T>, CopyOnWriteArraySet<Consumer<T>>> handlers
|
||||||
= new SimpleMultimap<>(() -> new EnumMap<>(EventPriority.class), CopyOnWriteArraySet::new);
|
= new SimpleMultimap<>(() -> new EnumMap<>(EventPriority.class), CopyOnWriteArraySet::new);
|
||||||
|
|
||||||
@@ -47,9 +49,13 @@ public final class EventManager<T extends Event> {
|
|||||||
register(consumer, EventPriority.NORMAL);
|
register(consumer, EventPriority.NORMAL);
|
||||||
}
|
}
|
||||||
|
|
||||||
public synchronized void register(Consumer<T> consumer, EventPriority priority) {
|
public void register(Consumer<T> consumer, EventPriority priority) {
|
||||||
if (!handlers.get(priority).contains(consumer))
|
lock.lock();
|
||||||
handlers.put(priority, consumer);
|
try {
|
||||||
|
handlers.get(priority).add(consumer);
|
||||||
|
} finally {
|
||||||
|
lock.unlock();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public void register(Runnable runnable) {
|
public void register(Runnable runnable) {
|
||||||
@@ -60,23 +66,30 @@ public final class EventManager<T extends Event> {
|
|||||||
register(t -> runnable.run(), priority);
|
register(t -> runnable.run(), priority);
|
||||||
}
|
}
|
||||||
|
|
||||||
public synchronized Event.Result fireEvent(T event) {
|
public Event.Result fireEvent(T event) {
|
||||||
|
lock.lock();
|
||||||
|
try {
|
||||||
for (EventPriority priority : EventPriority.values()) {
|
for (EventPriority priority : EventPriority.values()) {
|
||||||
for (Consumer<T> handler : handlers.get(priority))
|
for (Consumer<T> handler : handlers.get(priority))
|
||||||
handler.accept(event);
|
handler.accept(event);
|
||||||
}
|
}
|
||||||
|
} finally {
|
||||||
if (event.hasResult())
|
lock.unlock();
|
||||||
return event.getResult();
|
|
||||||
else
|
|
||||||
return Event.Result.DEFAULT;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public synchronized void unregister(Consumer<T> consumer) {
|
return event.hasResult() ? event.getResult() : Event.Result.DEFAULT;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void unregister(Consumer<T> consumer) {
|
||||||
|
lock.lock();
|
||||||
|
try {
|
||||||
handlers.removeValue(consumer);
|
handlers.removeValue(consumer);
|
||||||
|
} finally {
|
||||||
|
lock.unlock();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private class WeakListener implements Consumer<T> {
|
private final class WeakListener implements Consumer<T> {
|
||||||
private final WeakReference<Consumer<T>> ref;
|
private final WeakReference<Consumer<T>> ref;
|
||||||
|
|
||||||
public WeakListener(Consumer<T> listener) {
|
public WeakListener(Consumer<T> listener) {
|
||||||
|
|||||||
Reference in New Issue
Block a user