Fix for Issue #416 and #417, updated document fixed

stargate · May 9, 2023 · 70720c2 · 70720c2
1 parent 5d5d678
commit 70720c2
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/...in/java/io/stargate/sgv2/jsonapi/service/operation/model/impl/ReadAndUpdateOperation.java b/...in/java/io/stargate/sgv2/jsonapi/service/operation/model/impl/ReadAndUpdateOperation.java
@@ -175,10 +175,11 @@ private Uni<UpdatedDocument> processUpdate(
                 }
               }
 
-              // otherwise shred
-              JsonNode updatedDocument = documentUpdaterResponse.document();
               final WritableShreddedDocument writableShreddedDocument =
-                  shredder().shred(updatedDocument, readDocument.txnId());
+                  shredder().shred(documentUpdaterResponse.document(), readDocument.txnId());
+
+              // Have to do this because shredder adds _id field to the document if it doesn't exist
+              JsonNode updatedDocument = writableShreddedDocument.docJsonNode();
 
               // update the document
               return updatedDocument(queryExecutor, writableShreddedDocument)