feat: add id as field

Signed-off-by: Otavio Santana <otaviopolianasantana@gmail.com>
eclipse · May 19, 2023 · b3f1867 · b3f1867
1 parent 68c1099
commit b3f1867
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/...a/org/eclipse/jnosql/databases/orientdb/communication/DefaultOrientDBDocumentManager.java b/...a/org/eclipse/jnosql/databases/orientdb/communication/DefaultOrientDBDocumentManager.java
@@ -20,6 +20,7 @@
 import com.orientechnologies.orient.core.db.ODatabaseSession;
 import com.orientechnologies.orient.core.fetch.OFetchHelper;
 import com.orientechnologies.orient.core.id.ORecordId;
+import com.orientechnologies.orient.core.record.ORecord;
 import com.orientechnologies.orient.core.record.impl.ODocument;
 import com.orientechnologies.orient.core.sql.executor.OResult;
 import com.orientechnologies.orient.core.sql.executor.OResultSet;
@@ -33,6 +34,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.Objects;
+import java.util.Optional;
 import java.util.stream.Stream;
 import java.util.stream.StreamSupport;
 
@@ -103,6 +105,12 @@ public Iterable<DocumentEntity> insert(Iterable<DocumentEntity> entities, Durati
     @Override
     public DocumentEntity update(DocumentEntity entity) {
         requireNonNull(entity, "Entity is required");
+        Optional<Document> rid = entity.find(RID_FIELD);
+        Optional<Document> id = entity.find("_id");
+        try (ODatabaseSession tx = pool.acquire()) {
+            ORecord document = tx.load((ORecord) rid.get());
+            updateEntity(entity, document);
+        }
         return insert(entity);
     }
 

diff --git a/.../src/main/java/org/eclipse/jnosql/databases/orientdb/communication/OrientDBConverter.java b/.../src/main/java/org/eclipse/jnosql/databases/orientdb/communication/OrientDBConverter.java
@@ -41,6 +41,8 @@ final class OrientDBConverter {
     static final String RID_FIELD = "@rid";
     static final String VERSION_FIELD = "@version";
 
+    static final String ID_FIELD = "_id";
+
     private OrientDBConverter() {
     }
 
@@ -70,6 +72,7 @@ static DocumentEntity convert(OResult row) {
         long clusterPosition = identity.getClusterPosition();
         entity.add(VERSION_FIELD, element.getVersion());
         entity.add(RID_FIELD, "#" + clusterId + ":" + clusterPosition);
+        entity.add(ID_FIELD,  "#" + clusterId + ":" + clusterPosition);
         return entity;
     }