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

Hardened EOSEAccountFast against concurrent access so callers no longer iterate over live mutable maps

Browse Source
This commit is contained in:
davotoula
2025-11-05 18:55:58 +01:00
parent c07202944d
commit a82d6565fa
1 changed files with 22 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
amethyst/src/main/java/com/vitorpamplona/amethyst/service/relays/EOSE.kt
View File

@@ -125,36 +125,46 @@ class EOSEAccountFast<T : Any>(
cacheSize: Int = 20,
) {
private val users: LruCache<T, EOSERelayList> = LruCache<T, EOSERelayList>(cacheSize)
private val lock = Any()
fun addOrUpdate(
user: T,
relayUrl: NormalizedRelayUrl,
time: Long,
) {
val relayList = users[user]
if (relayList == null) {
val newList = EOSERelayList()
users.put(user, newList)
synchronized(lock) {
val relayList = users[user]
if (relayList == null) {
val newList = EOSERelayList()
users.put(user, newList)
newList.addOrUpdate(relayUrl, time)
} else {
relayList.addOrUpdate(relayUrl, time)
newList.addOrUpdate(relayUrl, time)
} else {
relayList.addOrUpdate(relayUrl, time)
}
}
}
fun removeEveryoneBut(list: Set<T>) {
users.snapshot().forEach {
if (it.key !in list) {
users.remove(it.key)
synchronized(lock) {
users.snapshot().forEach {
if (it.key !in list) {
users.remove(it.key)
}
}
}
}
fun removeDataFor(user: T) {
users.remove(user)
synchronized(lock) {
users.remove(user)
}
}
fun since(key: T) = users[key]?.relayList
fun since(key: T): SincePerRelayMap? =
synchronized(lock) {
users[key]?.relayList?.toMutableMap()
}
fun newEose(
user: T,