Added check for connection policy

Author: NaluTripician

Microsoft.Azure.Cosmos/src/DocumentClient.cs

@@ -6821,7 +6821,14 @@ private async Task InitializeGatewayConfigurationReaderAsync()
             this.accountServiceConfiguration = new CosmosAccountServiceConfiguration(accountReader.InitializeReaderAsync);
 
             await this.accountServiceConfiguration.InitializeAsync();
-            AccountProperties accountProperties = this.accountServiceConfiguration.AccountProperties;
+            AccountProperties accountProperties = this.accountServiceConfiguration.AccountProperties;
+            this.ConnectionPolicy.EnablePartitionLevelFailover = accountProperties.EnablePartitionLevelFailover;
+            this.PartitionKeyRangeLocation = this.ConnectionPolicy.EnablePartitionLevelFailover || this.ConnectionPolicy.EnablePartitionLevelCircuitBreaker
+                ? new GlobalPartitionEndpointManagerCore(
+                    this.GlobalEndpointManager,
+                    this.ConnectionPolicy.EnablePartitionLevelFailover,
+                    this.ConnectionPolicy.EnablePartitionLevelCircuitBreaker)
+                : GlobalPartitionEndpointManagerNoOp.Instance;
             this.UseMultipleWriteLocations = this.ConnectionPolicy.UseMultipleWriteLocations && accountProperties.EnableMultipleWriteLocations;
 
             this.GlobalEndpointManager.InitializeAccountPropertiesAndStartBackgroundRefresh(accountProperties);

Microsoft.Azure.Cosmos/src/Resource/Settings/AccountProperties.cs

@@ -244,6 +244,9 @@ internal long ProvisionedDocumentStorageInMB
         [JsonProperty(PropertyName = Constants.Properties.EnableMultipleWriteLocations)]
         internal bool EnableMultipleWriteLocations { get; set; }
 
+        [JsonProperty(PropertyName = Constants.Properties.EnablePartitionLevelFailover)]
+        internal bool EnablePartitionLevelFailover { get; set; }
+
         private IDictionary<string, object> QueryStringToDictConverter()
         {
             if (!string.IsNullOrEmpty(this.QueryEngineConfigurationString))

Microsoft.Azure.Cosmos/src/Routing/GlobalEndpointManager.cs

@@ -554,6 +554,7 @@ public virtual void InitializeAccountPropertiesAndStartBackgroundRefresh(Account
             {
                 return;
             }
+            this.connectionPolicy.EnablePartitionLevelFailover = databaseAccount.EnablePartitionLevelFailover;
             GlobalEndpointManager.ParseThinClientLocationsFromAdditionalProperties(databaseAccount);
 
             this.locationCache.OnDatabaseAccountRead(databaseAccount);