highperfocused/amethyst
1
0
Fork 0
mirror of https://github.com/vitorpamplona/amethyst.git synced 2025-11-10 07:37:12 +01:00
Code Issues Packages Projects Releases Wiki Activity

Improves the debug of the memory use per event kind

Browse Source
This commit is contained in:
Vitor Pamplona
2024-05-24 19:26:39 -04:00
parent 2c2e4a1694
commit cd008d78a0
2 changed files with 49 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
commons/src/main/java/com/vitorpamplona/amethyst/commons/data/LargeCache.kt
View File

@@ -136,6 +136,15 @@ class LargeCache<K, V> {
return runner.results
}
fun <R> sumByGroup(
groupMap: BiNotNullMapper<K, V, R>,
sumOf: BiNotNullMapper<K, V, Long>,
): Map<R, Long> {
val runner = BiSumByGroupCollector(groupMap, sumOf)
innerForEach(runner)
return runner.results
}
fun count(consumer: BiFilter<K, V>): Int {
val runner = BiCountIfCollector(consumer)
innerForEach(runner)
@@ -356,6 +365,24 @@ class BiCountByGroupCollector<K, V, R>(val mapper: BiNotNullMapper<K, V, R>) : B
}
}
class BiSumByGroupCollector<K, V, R>(val mapper: BiNotNullMapper<K, V, R>, val sumOf: BiNotNullMapper<K, V, Long>) : BiConsumer<K, V> {
var results = HashMap<R, Long>()
override fun accept(
k: K,
v: V,
) {
val group = mapper.map(k, v)
val sum = results[group]
if (sum == null) {
results[group] = sumOf.map(k, v)
} else {
results[group] = sum + sumOf.map(k, v)
}
}
}
class BiCountIfCollector<K, V>(val filter: BiFilter<K, V>) : BiConsumer<K, V> {
var count = 0