blob: 6b9b0bf7d09962ebe9eaf5b3d4264feafd4335f8 (
plain)
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
86
87
88
89
90
91
92
93
94
95
96
97
98
|
package me.bush.illnamethislater
import kotlin.reflect.KClass
/**
* A simple event dispatcher
*
* [Information and examples](https://github.com/therealbush/eventbus-kotlin#tododothething)
*
* @author bush
* @since 1.0.0
*/
class EventBus(private val config: Config = Config()) {
private val listeners = hashMapOf<KClass<*>, ListenerGroup>()
private val subscribers = mutableSetOf<Any>()
/**
* doc
*
* [Information and examples](https://github.com/therealbush/eventbus-kotlin#tododothething)
*/
fun subscribe(subscriber: Any): Boolean {
return if (subscriber in subscribers) false
else runCatching {
subscriber::class.listeners.forEach { member ->
register(member.handleCall(subscriber).also {
it.subscriber = subscriber
})
}
subscribers.add(subscriber)
true
}.getOrElse {
config.logger.error("Unable to register listeners for subscriber $subscriber", it)
false
}
}
/**
* doc
*
* [Information and examples](https://github.com/therealbush/eventbus-kotlin#tododothething)
*/
fun unsubscribe(subscriber: Any): Boolean {
return subscribers.remove(subscriber).also { contains ->
if (contains) listeners.entries.removeIf {
it.value.unsubscribe(subscriber)
it.value.sequential.isEmpty() && it.value.parallel.isEmpty()
}
}
}
/**
* 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.
*
* [Information and examples](https://github.com/therealbush/eventbus-kotlin#tododothething)
*/
fun register(listener: Listener) = listener.also {
listeners.computeIfAbsent(it.type) { type -> ListenerGroup(type, config) }.register(it)
}
/**
* doc
*
* [Information and examples](https://github.com/therealbush/eventbus-kotlin#tododothething)
*/
fun unregister(listener: Listener) = listener.also {
listeners[it.type]?.unregister(it)
}
/**
* Posts an event. doc
*
* [Information and examples](https://github.com/therealbush/eventbus-kotlin#tododothething)
*/
fun post(event: Any) = listeners[event::class]?.post(event) ?: false
/**
* Logs the subscriber count, total listener count, and listener count for every event type with at
* least one subscriber to [Config.logger]. Per-event counts are sorted from greatest to least listeners.
*
* [Information and examples](https://github.com/therealbush/eventbus-kotlin#tododothething)
* ```
* Subscribers: 5
* Listeners: 8 sequential, 4 parallel
* BushIsSoCool: 4, 2
* OtherEvent: 3, 1
* String: 1, 1
*/
fun debugInfo() {
config.logger.info("Subscribers: ${subscribers.size}")
val sequential = listeners.values.sumOf { it.sequential.size }
val parallel = listeners.values.sumOf { it.parallel.size }
config.logger.info("Listeners: $sequential sequential, $parallel parallel")
listeners.values.sortedByDescending { it.sequential.size + it.parallel.size }.forEach { it.debugInfo() }
}
}
|
