Keep assignments got from shard manager in updateAssignments

devsisters · Jul 22, 2024 · 3ff4eb5 · 3ff4eb5
1 parent f341cf2
commit 3ff4eb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/entities/src/main/scala/com/devsisters/shardcake/Sharding.scala b/entities/src/main/scala/com/devsisters/shardcake/Sharding.scala
@@ -139,7 +139,7 @@ class Sharding private (
       Metrics.shards
         .set(assignmentsOpt.count { case (_, podOpt) => podOpt.contains(address) })
         .when(fromShardManager) *>
-      (if (fromShardManager) shardAssignments.update(map => if (map.isEmpty) assignments else map)
+      (if (fromShardManager) shardAssignments.set(assignments)
        else
          shardAssignments.update(map =>
            // we keep self assignments (we don't override them with the new assignments