Merge pull request #35 from erikpaperik/bugfix/enable-cross-aprtition…

…-query Set EnableCrossPartitionQuery=true
imranmomin · Feb 11, 2019 · f7f6fc0 · f7f6fc0
2 parents 6825038 + 758284f
commit f7f6fc0
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 6 deletions.
diff --git a/Hangfire.AzureDocumentDB/DocumentDbConnection.cs b/Hangfire.AzureDocumentDB/DocumentDbConnection.cs
@@ -214,7 +214,11 @@ public override List<string> GetRangeFromSet(string key, int startingFrom, int e
         {
             if (key == null) throw new ArgumentNullException(nameof(key));
 
-            FeedOptions feedOptions = new FeedOptions { MaxItemCount = endingAt + 1 };
+            FeedOptions feedOptions = new FeedOptions { 
+              EnableCrossPartitionQuery = true,
+              MaxItemCount = endingAt + 1 
+            };
+
             endingAt += 1 - startingFrom;
 
             return Storage.Client.CreateDocumentQuery<Set>(Storage.CollectionUri, feedOptions)
@@ -486,7 +490,10 @@ public override List<string> GetRangeFromList(string key, int startingFrom, int
         {
             if (key == null) throw new ArgumentNullException(nameof(key));
 
-            FeedOptions feedOptions = new FeedOptions { MaxItemCount = endingAt + 1 };
+            FeedOptions feedOptions = new FeedOptions { 
+              EnableCrossPartitionQuery = true,
+              MaxItemCount = endingAt + 1 
+            };
             endingAt += 1 - startingFrom;
 
             return Storage.Client.CreateDocumentQuery<List>(Storage.CollectionUri, feedOptions)

diff --git a/Hangfire.AzureDocumentDB/DocumentDbMonitoringApi.cs b/Hangfire.AzureDocumentDB/DocumentDbMonitoringApi.cs
@@ -119,6 +119,7 @@ public StatisticsDto GetStatistics()
 
                     FeedOptions feedOptions = new FeedOptions
                     {
+                        EnableCrossPartitionQuery = true,
                         MaxItemCount = 1000,
                     };
 
@@ -290,7 +291,10 @@ public JobList<DeletedJobDto> DeletedJobs(int from, int count)
         private JobList<T> GetJobsOnState<T>(string stateName, int from, int count, Func<State, Common.Job, T> selector)
         {
             List<KeyValuePair<string, T>> jobs = new List<KeyValuePair<string, T>>();
-            FeedOptions feedOptions = new FeedOptions { MaxItemCount = from + count };
+            FeedOptions feedOptions = new FeedOptions { 
+              EnableCrossPartitionQuery = true,
+              MaxItemCount = from + count 
+            };
 
             List<Documents.Job> filterJobs = storage.Client.CreateDocumentQuery<Documents.Job>(storage.CollectionUri, feedOptions)
                  .Where(j => j.DocumentType == DocumentTypes.Job && j.StateName == stateName)
@@ -334,7 +338,10 @@ private JobList<T> GetJobsOnQueue<T>(string queryText, string queue, int from, i
                 }
             };
 
-            FeedOptions feedOptions = new FeedOptions { MaxItemCount = from + count };
+            FeedOptions feedOptions = new FeedOptions { 
+              EnableCrossPartitionQuery = true,
+              MaxItemCount = from + count 
+            };
 
             List<Documents.Queue> queues = storage.Client.CreateDocumentQuery<Documents.Queue>(storage.CollectionUri, sql, feedOptions)
                  .ToQueryResult()

diff --git a/Hangfire.AzureDocumentDB/Queue/JobQueueMonitoringApi.cs b/Hangfire.AzureDocumentDB/Queue/JobQueueMonitoringApi.cs
@@ -65,7 +65,10 @@ public int GetEnqueuedCount(string queue)
 
         public IEnumerable<string> GetEnqueuedJobIds(string queue, int from, int perPage)
         {
-            FeedOptions feedOptions = new FeedOptions { MaxItemCount = from + perPage };
+            FeedOptions feedOptions = new FeedOptions { 
+                EnableCrossPartitionQuery = true,
+                MaxItemCount = from + perPage 
+            };
 
             return storage.Client.CreateDocumentQuery<Documents.Queue>(storage.CollectionUri, feedOptions)
                 .Where(q => q.DocumentType == DocumentTypes.Queue && q.Name == queue && q.FetchedAt.IsDefined() == false)
@@ -77,7 +80,10 @@ public IEnumerable<string> GetEnqueuedJobIds(string queue, int from, int perPage
 
         public IEnumerable<string> GetFetchedJobIds(string queue, int from, int perPage)
         {
-            FeedOptions feedOptions = new FeedOptions { MaxItemCount = from + perPage };
+            FeedOptions feedOptions = new FeedOptions { 
+                EnableCrossPartitionQuery = true,
+                MaxItemCount = from + perPage 
+            };
 
             return storage.Client.CreateDocumentQuery<Documents.Queue>(storage.CollectionUri, feedOptions)
                 .Where(q => q.DocumentType == DocumentTypes.Queue && q.Name == queue && q.FetchedAt.IsDefined())