Pass cancellation token to the resolver (microsoft#349)

* Pass cancellation token to the resolver func * pass cancellation token to RegisterServiceNotificationFilterAsync * service Uri can be added to registrationCache before the RegisterServiceNotificationFilterAsync call
LastTribunal · Mar 20, 2023 · 2901088 · 2901088
1 parent 4c39943
commit 2901088
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/Microsoft.ServiceFabric.Services/Client/ServicePartitionResolver.cs b/src/Microsoft.ServiceFabric.Services/Client/ServicePartitionResolver.cs
@@ -570,7 +570,7 @@ private async Task<ResolvedServicePartition> ResolveHelperAsync(
                         client,
                         previousRsp,
                         totaltime.GetRemainingTime(),
-                        CancellationToken.None);
+                        cancellationToken);
                 }
                 catch (Exception ex)
                 {
@@ -625,10 +625,10 @@ private async Task<ResolvedServicePartition> ResolveHelperAsync(
                                 if (added)
                                 {
                                     ServiceNotificationFilterDescription filter = new ServiceNotificationFilterDescription(
-                                       name: serviceUri,
-                                       matchNamePrefix: true,
-                                       matchPrimaryChangeOnly: false);
-                                    await client.ServiceManager.RegisterServiceNotificationFilterAsync(filter, totaltime.GetRemainingTime(), CancellationToken.None);
+                                        name: serviceUri,
+                                        matchNamePrefix: true,
+                                        matchPrimaryChangeOnly: false);
+                                    await client.ServiceManager.RegisterServiceNotificationFilterAsync(filter, totaltime.GetRemainingTime(), cancellationToken);
                                 }
                             }
                         }