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

Merge pull request #26962 from akka/wip-26957-ClusterShardingFailureSpec-patriknw

Browse source 
Deliver buffered messages after passivation, #26957
This commit is contained in:
Patrik Nordwall 2019-05-20 11:06:00 +02:00 • committed by GitHub
commit d9b6f633a0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 27 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -430,11 +430,20 @@ private[akka] class Shard(
// EntityStopped handler
def passivateCompleted(event: EntityStopped): Unit = {
log.debug("Entity stopped after passivation [{}]", event.entityId)
val hasBufferedMessages = messageBuffers.getOrEmpty(event.entityId).nonEmpty
state = state.copy(state.entities - event.entityId)
if (hasBufferedMessages) {
log.debug(
"Entity stopped after passivation [{}], but will be started again due to buffered messages.",
event.entityId)
processChange(EntityStarted(event.entityId))(sendMsgBuffer)
} else {
log.debug("Entity stopped after passivation [{}]", event.entityId)
messageBuffers.remove(event.entityId)
}
}
// EntityStarted handler
def sendMsgBuffer(event: EntityStarted): Unit = {
//Get the buffered messages and remove the buffer

8
akka-cluster-sharding/src/main/scala/akka/cluster/sharding/ShardCoordinator.scala
View file

@ -1171,8 +1171,14 @@ class DDataShardCoordinator(
unstashAll()
}
private def stashGetShardHomeRequest(sender: ActorRef, request: GetShardHome): Unit =
private def stashGetShardHomeRequest(sender: ActorRef, request: GetShardHome): Unit = {
log.debug(
"GetShardHome [{}] request from [{}] stashed, because waiting for initial state or update of state. " +
"It will be handled afterwards.",
request.shard,
sender)
getShardHomeRequests += (sender -> request)
}
private def unstashGetShardHomeRequests(): Unit = {
getShardHomeRequests.foreach {

12
akka-cluster-sharding/src/multi-jvm/scala/akka/cluster/sharding/ClusterShardingFailureSpec.scala
View file

@ -238,10 +238,16 @@ abstract class ClusterShardingFailureSpec(config: ClusterShardingFailureSpecConf
//Test the Shard passivate works after a journal failure
shard2.tell(Passivate(PoisonPill), entity21)
awaitCond({
awaitAssert {
// Note that the order between this Get message to 21 and the above Passivate to 21 is undefined.
// If this Get arrives first the reply will be Value("21", 3) and then it is retried by the
// awaitAssert.
// Also note that there is no timeout parameter on below expectMsg because messages should not
// be lost here. They should be buffered and delivered also after Passivate completed.
region ! Get("21")
expectMsgType[Value] == Value("21", 0)
}, message = "Passivating did not reset Value down to 0")
// counter reset to 0 when started again
expectMsg(Value("21", 0))
}
region ! Add("21", 1)