Moving to ValueFactory overloads of ConcurrentDictionary (#2852)

Azure · Nov 10, 2021 · 8646349 · 8646349
1 parent 5001696
commit 8646349
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/Microsoft.Azure.Cosmos/src/Routing/GatewayAddressCache.cs b/Microsoft.Azure.Cosmos/src/Routing/GatewayAddressCache.cs
@@ -554,7 +554,8 @@ internal Tuple<PartitionKeyRangeIdentity, PartitionAddressInformation> ToPartiti
                        addressInfo.PhysicalUri);
 
                     HashSet<PartitionKeyRangeIdentity> pkRangeIdSet = this.serverPartitionAddressToPkRangeIdMap.GetOrAdd(
-                        new ServerKey(new Uri(addressInfo.PhysicalUri)), new HashSet<PartitionKeyRangeIdentity>());
+                        new ServerKey(new Uri(addressInfo.PhysicalUri)),
+                        (_) => new HashSet<PartitionKeyRangeIdentity>());
                     lock (pkRangeIdSet)
                     {
                         pkRangeIdSet.Add(partitionKeyRangeIdentity);

diff --git a/Microsoft.Azure.Cosmos/src/Routing/GlobalPartitionEndpointManagerCore.cs b/Microsoft.Azure.Cosmos/src/Routing/GlobalPartitionEndpointManagerCore.cs
@@ -134,7 +134,7 @@ public override bool TryMarkEndpointUnavailableForPartitionKeyRange(
 
             PartitionKeyRangeFailoverInfo partionFailover = this.PartitionKeyRangeToLocation.Value.GetOrAdd(
                 partitionKeyRange,
-                new PartitionKeyRangeFailoverInfo(failedLocation));
+                (_) => new PartitionKeyRangeFailoverInfo(failedLocation));
 
             // Will return true if it was able to update to a new region
             if (partionFailover.TryMoveNextLocation(