diff --git a/src/DocumentDBRepository.cs b/src/DocumentDBRepository.cs index 23bbd61..fa20c53 100644 --- a/src/DocumentDBRepository.cs +++ b/src/DocumentDBRepository.cs @@ -23,7 +23,8 @@ public static async Task GetItemAsync(string id, string category) try { Document document = - await client.ReadDocumentAsync(UriFactory.CreateDocumentUri(DatabaseId, CollectionId, id)); + await client.ReadDocumentAsync(UriFactory.CreateDocumentUri(DatabaseId, CollectionId, id), + new RequestOptions { PartitionKey = new PartitionKey(category) }); return (T)(dynamic)document; } catch (DocumentClientException e) @@ -68,7 +69,8 @@ public static async Task UpdateItemAsync(string id, T item) public static async Task DeleteItemAsync(string id, string category) { - await client.DeleteDocumentAsync(UriFactory.CreateDocumentUri(DatabaseId, CollectionId, id)); + await client.DeleteDocumentAsync(UriFactory.CreateDocumentUri(DatabaseId, CollectionId, id), + new RequestOptions { PartitionKey = new PartitionKey(category) }); } public static void Initialize()