Merge pull request #126 from Azure/fix-112

fix issue #112 : AzureData.replace throwing partition key error
Azure · May 24, 2019 · 4d3ed1e · 4d3ed1e
2 parents ed6a4a1 + d6fcadf
commit 4d3ed1e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/AzureData/Source/DocumentClient.swift b/AzureData/Source/DocumentClient.swift
@@ -298,11 +298,11 @@ class DocumentClient {
 
     // replace
     func replace<T: Document> (_ document: T, inCollection collectionId: String, inDatabase databaseId: String, callback: @escaping (Response<T>) -> ()) {
-        return self.replace(DocumentContainer(document), at: .document(databaseId: databaseId, collectionId: collectionId, id: document.id)) { callback($0.map { $0.document }) }
+        return self.replace(DocumentContainer(document), at: .document(databaseId: databaseId, collectionId: collectionId, id: document.id), additionalHeaders: HttpHeaders.msDocumentdbPartitionKey(document)) { callback($0.map { $0.document }) }
     }
 
     func replace<T: Document> (_ document: T, in collection: DocumentCollection, callback: @escaping (Response<T>) -> ()) {
-        return self.replace(DocumentContainer(document), at: .child(.document, in: collection, id: document.id)) { callback($0.map { $0.document }) }
+        return self.replace(DocumentContainer(document), at: .child(.document, in: collection, id: document.id), additionalHeaders: HttpHeaders.msDocumentdbPartitionKey(document)) { callback($0.map { $0.document }) }
     }
 
     // query