1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
package me.bush.illnamethislater
import org.apache.logging.log4j.LogManager
import org.apache.logging.log4j.Logger
import kotlin.reflect.KClass
/**
* [A simple event dispatcher](http://github.com/therealbush/eventbus-kotlin)
*
* @author bush
* @since 1.0.0
*/
class EventBus(private val logger: Logger = LogManager.getLogger()) {
private val listeners = hashMapOf<KClass<*>, ListenerList>()
private val subscribers = mutableSetOf<Any>()
/**
* blah blah annotation properties override listener properties
*/
infix fun subscribe(subscriber: Any): Boolean {
return if (subscriber in subscribers) false
else runCatching {
subscriber::class.listeners.forEach {
register(it.handleCall(subscriber), subscriber)
}
true
}.getOrElse {
logger.error("Unable to register listeners for subscriber $subscriber", it)
false
}
}
/**
* Registers a listener (which may not belong to any subscriber) to this [EventBus]. If no object
* is given, a key will be returned which can be used in [unsubscribe] to remove the listener.
*
* The
*/
fun register(listener: Listener, subscriber: Any = Any()): Any {
listeners.getOrPut(listener.type) { ListenerList() }
.add(listener.also { it.subscriber = subscriber })
subscribers += subscriber
return subscriber
}
/**
*
*/
// doc
infix fun unsubscribe(subscriber: Any) = subscribers.remove(subscriber).apply {
if (this) listeners.entries.removeIf {
it.value.removeFrom(subscriber)
it.value.isEmpty
}
}
/**
* Posts an event.
*/
// doc
infix fun post(event: Any) = listeners[event::class]?.post(event)
/**
* Logs the subscriber count, total listener count, and listener count
* for every event type with at least one subscriber to [logger].
* Per-event counts are sorted from greatest to least listeners.
* ```
* Subscribers: 5
* Listeners: 8 sequential, 4 parallel
* SomeInnerClass: 4, 2
* OtherEvent: 3, 1
* String: 1, 1
*/
fun debugInfo() {
logger.info(StringBuilder().apply {
append("\nSubscribers: ${subscribers.size}")
val sequential = listeners.values.sumOf { it.sequential.size }
val parallel = listeners.values.sumOf { it.parallel.size }
append("\nListeners: $sequential sequential, $parallel parallel")
listeners.entries.sortedByDescending { it.value.sequential.size + it.value.parallel.size }.forEach {
append("\n${it.key.simpleName}: ${it.value.sequential.size}, ${it.value.parallel.size}")
}
}.toString())
}
}
|
