raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 10

Avoid allocation on entity lookup

Browse source
This commit is contained in:
Johan Andrén 2020-06-11 15:58:33 +02:00
parent 55311cf914
commit 1e811eebbc
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
akka-cluster-sharding/src/main/scala/akka/cluster/sharding/Shard.scala
View file

@ -246,7 +246,7 @@ private[akka] object Shard {
final class Entities(log: LoggingAdapter, rememberingEntities: Boolean, verboseDebug: Boolean) {
private val entities: java.util.Map[EntityId, EntityState] = new util.HashMap[EntityId, EntityState]()
// needed to look up entity by reg when a Passivating is received
// needed to look up entity by ref when a Passivating is received
private val byRef = new util.HashMap[ActorRef, EntityId]()
// optimization to not have to go through all entities to find batched writes
private val remembering = new util.HashSet[EntityId]()
@ -302,9 +302,11 @@ private[akka] object Shard {
case _ => OptionVal.None
}
def entityState(id: EntityId): EntityState = {
OptionVal(entities.get(id)).getOrElse(NoState)
}
def entityState(id: EntityId): EntityState =
entities.get(id) match {
case null => NoState
case state => state
}
def entityId(ref: ActorRef): OptionVal[EntityId] = OptionVal(byRef.get(ref))