Minecraft/mc-core
Archived
0
Fork 0
Code Activity

EventBus unregister

Browse Source
This commit is contained in:
DmitriyMX
2019-01-12 19:41:22 +03:00
parent 3659095851
commit 71c1699206
2 changed files with 49 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
core/src/main/java/mc/core/eventbus/EventBus.java
View File

@@ -17,16 +17,20 @@ public class EventBus {
private Queue<Event> eventQueue;
private Map<Class<? extends Event>, List<Pair<Object, Method>>> subscribes = new HashMap<>();
@SuppressWarnings("unchecked")
public void registerSubscribes(Object subscriberObject) {
Stream.of(subscriberObject.getClass().getDeclaredMethods())
private Stream<Method> getMethods(Object subscriberObject) {
return Stream.of(subscriberObject.getClass().getDeclaredMethods())
.filter(method -> method.isAnnotationPresent(Subscriber.class))
.filter(method -> method.getReturnType().equals(Void.TYPE))
.filter(method -> method.getParameterCount() == 1)
.filter(method -> Event.class.isAssignableFrom(method.getParameterTypes()[0]))
.filter(method -> Event.class.isAssignableFrom(method.getParameterTypes()[0]));
}
@SuppressWarnings("unchecked")
public void registerSubscribes(Object subscriberObject) {
getMethods(subscriberObject)
.forEach(method -> {
Class<? extends Event> type = (Class<? extends Event>) method.getParameterTypes()[0];
List<Pair<Object, Method>> pairs;
final Class<? extends Event> type = (Class<? extends Event>) method.getParameterTypes()[0];
final List<Pair<Object, Method>> pairs;
if (subscribes.containsKey(type)) {
pairs = subscribes.get(type);
} else {
@@ -36,4 +40,20 @@ public class EventBus {
pairs.add(new Pair<>(subscriberObject, method));
});
}
@SuppressWarnings("unchecked")
public void unregisterSubscribes(Object subscriberObject) {
getMethods(subscriberObject)
.forEach(method -> {
final Class<? extends Event> type = (Class<? extends Event>) method.getParameterTypes()[0];
if (subscribes.containsKey(type)) {
final List<Pair<Object, Method>> pairs = subscribes.get(type);
pairs.removeIf(pair -> pair.getKey() == subscriberObject);
if (pairs.isEmpty()) {
subscribes.remove(type);
}
}
});
}
}