UpdateIAmAlive for SingleMembership fixed to not create invalid subdo…

…cuments without required properties
OrleansContrib · Mar 12, 2024 · ec538dd · ec538dd
1 parent cd9d50c
commit ec538dd
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/Orleans.Providers.MongoDB/Membership/Store/Single/SingleMembershipCollection.cs b/Orleans.Providers.MongoDB/Membership/Store/Single/SingleMembershipCollection.cs
@@ -92,8 +92,11 @@ public async Task<MembershipTableData> ReadRow(string deploymentId, SiloAddress
 
         public async Task UpdateIAmAlive(string deploymentId, SiloAddress address, DateTime iAmAliveTime)
         {
-            await Collection.UpdateOneAsync(x => x.DeploymentId == deploymentId,
-                Update
+            var filter = Builders<DeploymentDocument>.Filter.And(
+                Builders<DeploymentDocument>.Filter.Eq(x => x.DeploymentId, deploymentId),
+                Builders<DeploymentDocument>.Filter.Exists($"Members.{BuildKey(address)}", true));
+
+            await Collection.UpdateOneAsync(filter, Update
                     .Set($"Members.{BuildKey(address)}.IAmAliveTime", LogFormatter.PrintDate(iAmAliveTime)));
         }