Fix address logging

Azure · Mar 15, 2022 · 6f28f2f · 6f28f2f
1 parent 2f2b687
commit 6f28f2f
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/Microsoft.Azure.Cosmos/src/Routing/GatewayAddressCache.cs b/Microsoft.Azure.Cosmos/src/Routing/GatewayAddressCache.cs
@@ -192,11 +192,15 @@ public async Task<PartitionAddressInformation> TryGetAddressesAsync(
                 {
                     addresses = await this.serverPartitionAddressCache.GetAsync(
                         key: partitionKeyRangeIdentity,
-                        singleValueInitFunc: (staleAddresses) => this.GetAddressesForRangeIdAsync(
-                            request,
-                            partitionKeyRangeIdentity.CollectionRid,
-                            partitionKeyRangeIdentity.PartitionKeyRangeId,
-                            forceRefresh: forceRefreshPartitionAddresses),
+                        singleValueInitFunc: (currentCachedValue) =>
+                        { 
+                            staleAddressInfo = currentCachedValue;
+                            return this.GetAddressesForRangeIdAsync(
+                                request,
+                                partitionKeyRangeIdentity.CollectionRid,
+                                partitionKeyRangeIdentity.PartitionKeyRangeId,
+                                forceRefresh: forceRefreshPartitionAddresses);
+                        },
                         forceRefresh: (_) => true);
 
                     if (staleAddressInfo != null)