sharding PassivateIdle fix (#3808)

akkadotnet · Jul 26, 2019 · 285cf5f · 285cf5f
1 parent dfcd558
commit 285cf5f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/contrib/cluster/Akka.Cluster.Sharding/DDataShard.cs b/src/contrib/cluster/Akka.Cluster.Sharding/DDataShard.cs
@@ -114,6 +114,11 @@ internal sealed class DDataShard : ActorBase, IShard, IWithUnboundedStash
         public void EntityTerminated(IActorRef tref) => this.BaseEntityTerminated(tref);
         public void DeliverTo(string id, object message, object payload, IActorRef sender) => this.BaseDeliverTo(id, message, payload, sender);
 
+        protected override void PostStop()
+        {
+            PassivateIdleTask?.Cancel();
+            base.PostStop();
+        }
 
         protected override bool Receive(object message) => WaitingForState(ImmutableHashSet<int>.Empty)(message);
 

diff --git a/src/contrib/cluster/Akka.Cluster.Sharding/PersistentShard.cs b/src/contrib/cluster/Akka.Cluster.Sharding/PersistentShard.cs
@@ -84,6 +84,12 @@ internal sealed class PersistentShard : PersistentActor, IShard
 
         public override string PersistenceId { get; }
 
+        protected override void PostStop()
+        {
+            PassivateIdleTask?.Cancel();
+            base.PostStop();
+        }
+
         protected override bool ReceiveCommand(object message)
         {
             switch (message)