fixed bugs with StartEntity

akkadotnet · Jan 4, 2024 · b431ebc · b431ebc
1 parent b8514fb
commit b431ebc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/contrib/cluster/Akka.Cluster.Sharding/ClusterSharding.cs b/src/contrib/cluster/Akka.Cluster.Sharding/ClusterSharding.cs
@@ -71,7 +71,6 @@ public ExtractorAdapter(IMessageExtractor underlying)
         {
             return message switch
             {
-                StartEntity se => se.EntityId,
                 ShardingEnvelope se => se.EntityId,
                 _ => _underlying.EntityId(message)
             };

diff --git a/src/contrib/cluster/Akka.Cluster.Sharding/ShardRegion.cs b/src/contrib/cluster/Akka.Cluster.Sharding/ShardRegion.cs
@@ -609,6 +609,9 @@ protected override bool Receive(object message)
                 case RestartShard restart:
                     DeliverRestartShard(restart, Sender);
                     return true;
+                case StartEntity se:
+                    DeliverMessage(se.EntityId, message, Sender);
+                    return true;
                 default:
                     var entityId = _messageExtractor.EntityId(message);
                     if (entityId is null)