Code changes to address review comments.

Azure · Dec 29, 2023 · 2f427e3 · 2f427e3
1 parent b507ed2
commit 2f427e3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/Microsoft.Azure.Cosmos/src/MetadataRequestThrottleRetryPolicy.cs b/Microsoft.Azure.Cosmos/src/MetadataRequestThrottleRetryPolicy.cs
@@ -146,7 +146,7 @@ public void OnBeforeSendRequest(DocumentServiceRequest request)
         }
 
         /// <summary>
-        /// Marks an endpoint unavailable in the global endpoint manager, for any future read requests.
+        /// Increments the location index when a service unavailable exception ocurrs, for any future read requests.
         /// </summary>
         /// <returns>A boolean flag indicating if the operation was successful.</returns>
         private bool IncrementRetryIndexOnServiceUnavailableForMetadataRead()

diff --git a/Microsoft.Azure.Cosmos/src/Routing/PartitionKeyRangeCache.cs b/Microsoft.Azure.Cosmos/src/Routing/PartitionKeyRangeCache.cs
@@ -314,6 +314,11 @@ private async Task<DocumentServiceResponse> ExecutePartitionKeyRangeReadChangeFe
                             childTrace.AddDatum("Exception Message", ex.Message);
                             throw;
                         }
+                        catch (CosmosException ce)
+                        {
+                            childTrace.AddDatum("Exception Message", ce.Message);
+                            throw;
+                        }
                     }
                 }
             }