fix: add key for fix update method to the database

Signed-off-by: Otavio Santana <otaviopolianasantana@gmail.com>
eclipse · May 19, 2023 · 700b295 · 700b295
1 parent 3319edd
commit 700b295
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/...a/org/eclipse/jnosql/databases/arangodb/communication/DefaultArangoDBDocumentManager.java b/...a/org/eclipse/jnosql/databases/arangodb/communication/DefaultArangoDBDocumentManager.java
@@ -73,13 +73,18 @@ public DocumentEntity insert(DocumentEntity entity) throws NullPointerException
     public DocumentEntity update(DocumentEntity entity) {
         String collectionName = entity.name();
         checkCollection(collectionName);
+        String id = entity.find(ID, String.class)
+                .orElseThrow(() -> new IllegalArgumentException("The document does not provide" +
+                        " the _id column"));
+        feedKey(entity, id);
         BaseDocument baseDocument = ArangoDBUtil.getBaseDocument(entity);
         DocumentUpdateEntity<BaseDocument> arandoDocument = arangoDB.db(DbName.of(database))
                 .collection(collectionName).updateDocument(baseDocument.getKey(), baseDocument);
         updateEntity(entity, arandoDocument.getKey(), arandoDocument.getId(), arandoDocument.getRev());
         return entity;
     }
 
+
     @Override
     public Iterable<DocumentEntity> update(Iterable<DocumentEntity> entities) {
         Objects.requireNonNull(entities, "entities is required");
@@ -192,6 +197,17 @@ private void updateEntity(DocumentEntity entity, String key, String id, String r
         entity.add(Document.of(REV, rev));
     }
 
+    private static void feedKey(DocumentEntity entity, String id) {
+        if (entity.find(KEY).isEmpty()) {
+            String[] values = id.split("/");
+            if (values.length == 2) {
+                entity.add(KEY, values[1]);
+            } else {
+                entity.add(KEY, values[0]);
+            }
+        }
+    }
+
     ArangoDB getArangoDB() {
         return arangoDB;
     }